// Act
        CIFSContext result = smbTransport.getContext();

        // Assert
        assertNotNull(result, "Context should not be null");
        assertEquals(cifsContext, result, "Returned context should be the mocked context");
        verify(smbTransport).getContext(); // Verify that the method was called
    }

    @Test
    void testUnwrap() {
        // Arrange

      serverSocketChannel.configureBlocking(true);
      serverSocketChannel.socket().bind(address);
    }

    @Override public int getLocalPort() {
      return 1; // A white lie. There is no local port.
    }

    @Override public SocketAddress getLocalSocketAddress() {
      return endpoint;
    }

    @Override public Socket accept() throws IOException {
      try {

  ): RealConnection {
    val result =
      RealConnection.newTestConnection(
        taskRunner = taskRunner,
        connectionPool = pool,
        route = route,
        socket = Socket(),
        idleAtNs = idleAtNanos,
      )
    result.withLock { pool.put(result) }
    return result
  }

  fun newConnectionPool(
    taskRunner: TaskRunner = this.taskRunner,

  fun upgradeToSocket(): Socket {
    call.timeoutEarlyExit()
    (codec.carrier as RealConnection).useAsSocket()

    eventListener.requestBodyStart(call)

    return object : Socket {
      override fun cancel() {
        ******@****.***()
      }

      override val sink = RequestBodySink(codec.socket.sink, -1L)

        }

        @Test
        @DisplayName("Should decode output parameters correctly")
        void testDecodeOut() throws NdrException {
            // Given: Close handle message with mocked return value
            samr.SamrCloseHandle message = new samr.SamrCloseHandle(mockPolicyHandle);
            when(mockNdrBuffer.dec_ndr_long()).thenReturn(0);

            // When: Decoding output

    private SmbTree tree;

    @BeforeEach
    void resetMocks() {
        Mockito.reset(session, transport, tree);
    }

    // Happy path: isInUse returns the mocked state
    @ParameterizedTest
    @DisplayName("isInUse reflects current mocked state")
    @ValueSource(booleans = { true, false })
    void isInUse_variants(boolean inUse) {
        when(session.isInUse()).thenReturn(inUse);

    }

    @Mock
    SessionSetupHandler mocked;

    @Test
    @DisplayName("Interactions: Mockito mock has no interactions by default")
    void testMockitoNoInteractions() {
        // Intent: ensure we can mock the interface and verify no unexpected calls
        // Act & Assert
        assertNotNull(mocked);
        verifyNoInteractions(mocked);
    }

    void testEncode() throws NdrException {
        // Test the encode method
        long testValue = 987654321098765L;
        NdrHyper ndrHyper = new NdrHyper(testValue);

        // Call the encode method with the mocked NdrBuffer
        ndrHyper.encode(mockNdrBuffer);

        // Verify that enc_ndr_hyper was called exactly once with the correct value
        verify(mockNdrBuffer, times(1)).enc_ndr_hyper(testValue);

    /** Maximum size for RDMA read/write operations */
    protected int maxReadWriteSize;

    /**
     * Create new RDMA connection
     *
     * @param remote remote socket address
     * @param local local socket address, may be null
     */
    public RdmaConnection(InetSocketAddress remote, InetSocketAddress local) {
        this.remoteAddress = remote;
        this.localAddress = local;

import okio.Sink

/**
 * A sink that sleeps [periodDelayNanos] every [bytesPerPeriod] bytes. Unlike [okio.Throttler],
 * this permits any interval to be used.
 */
internal class ThrottledSink(
  private val socket: MockWebServerSocket,
  private val delegate: Sink,
  private val bytesPerPeriod: Long,
  private val periodDelayNanos: Long,
) : Sink by delegate {
  private var bytesWrittenSinceLastDelay = 0L

  override fun write(