testRequest.setOverrideTimeout(timeout);
            assertEquals(timeout, testRequest.getOverrideTimeout());
        }

        @Test
        @DisplayName("setOverrideTimeout should handle null value")
        void testSetOverrideTimeoutNull() {
            testRequest.setOverrideTimeout(5000);
            testRequest.setOverrideTimeout(null);
            assertNull(testRequest.getOverrideTimeout());

                + ",maxParameterCount=" + maxParameterCount + ",maxDataCount=" + maxDataCount + ",maxSetupCount=" + (int) maxSetupCount
                + ",flags=0x" + Hexdump.toHexString(flags, 2) + ",timeout=" + timeout + ",parameterCount=" + parameterCount
                + ",parameterOffset=" + parameterOffset + ",parameterDisplacement=" + parameterDisplacement + ",dataCount=" + dataCount

        }
        return this.nextNameTrnId;
    }

    void ensureOpen(final int timeout) throws IOException {
        this.closeTimeout = 0;
        if (this.transportContext.getConfig().getNetbiosSoTimeout() != 0) {
            this.closeTimeout = Math.max(this.transportContext.getConfig().getNetbiosSoTimeout(), timeout);
        }
        // If socket is still good, the new closeTimeout will

  public boolean isTerminated() {
    synchronized (lock) {
      return shutdown && runningTasks == 0;
    }
  }

  @Override
  public boolean awaitTermination(long timeout, TimeUnit unit) throws InterruptedException {
    long nanos = unit.toNanos(timeout);
    synchronized (lock) {
      while (true) {
        if (shutdown && runningTasks == 0) {
          return true;
        } else if (nanos <= 0) {
          return false;

     * @param path the file path
     * @param type the handle type
     * @param timeout the timeout in milliseconds
     * @param leaseKey the associated lease key (can be null)
     * @return the handle GUID
     */
    public HandleGuid requestDurableHandle(String path, HandleType type, long timeout, Smb2LeaseKey leaseKey) {
        HandleGuid guid = new HandleGuid();

import java.util.concurrent.CountDownLatch
import java.util.concurrent.atomic.AtomicInteger
import mockwebserver3.junit5.StartStop
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.Timeout

@Timeout(30)
class CustomDispatcherTest {
  @StartStop private val mockWebServer = MockWebServer()

  @Test
  fun simpleDispatch() {
    val requestsMade = mutableListOf<RecordedRequest>()

  /**
   * @since 28.0
   */
  @Override
  public final void awaitRunning(Duration timeout) throws TimeoutException {
    Service.super.awaitRunning(timeout);
  }

  /**
   * @since 15.0
   */
  @Override
  public final void awaitRunning(long timeout, TimeUnit unit) throws TimeoutException {
    delegate.awaitRunning(timeout, unit);
  }

  /**
   * @since 15.0
   */
  @Override

     * @param priority The priority.
     */
    public void setPriority(final int priority) {
        this.priority = priority;
    }

    /**
     * Sets the request timeout.
     * @param bulkRequestTimeout The request timeout.
     */
    public void setRequestTimeout(final String bulkRequestTimeout) {
        requestTimeout = bulkRequestTimeout;
    }

    /**
     * Sets the request cache size.

     */
    public int getHeartbeatFailures() {
        return heartbeatFailures;
    }

    /**
     * Checks if this registration has expired based on the timeout.
     *
     * @param timeoutMs the timeout in milliseconds
     * @return true if the registration is expired
     */
    public boolean isExpired(long timeoutMs) {
        return System.currentTimeMillis() - lastHeartbeat > timeoutMs;

      override fun connect(
        endpoint: SocketAddress?,
        timeout: Int,
      ) {
        val requested = (endpoint as InetSocketAddress)
        val inetSocketAddress = hostMapping[requested.address] ?: defaultAddress ?: requested
        Platform.get().log("Socket connection to: $inetSocketAddress was: $requested")
        super.connect(inetSocketAddress, timeout)
      }
    }