return delegate.getSmbSession(tf, targetHost, targetDomain);
    }

    @Override
    public boolean isSMB2() throws SmbException {
        return delegate.isSMB2();
    }

    @Override
    public int getInflightRequests() {
        return delegate.getInflightRequests();
    }

    @Override
    public CIFSContext getContext() {

            // and send it as a separate request instead
            String svc = null;
            final int t = this.tid;
            request.setTid(t);

            if (!transport.isSMB2()) {
                final ServerMessageBlock req = (ServerMessageBlock) request;
                svc = this.service;
                if (svc == null) {

        lenient().when(mockTreeHandle.getConfig()).thenReturn(mockConfig);
    }

    @Test
    void testWriteSingleByte() throws IOException, CIFSException {
        // Given
        when(mockTreeHandle.isSMB2()).thenReturn(true);
        when(mockTreeHandle.getSendBufferSize()).thenReturn(65536);
        when(mockFileHandle.isValid()).thenReturn(false, true); // First false to trigger ensureOpen, then true

    private SmbPipeInputStream newStreamWithInit(boolean smb2) throws CIFSException {
        when(handle.getPipe()).thenReturn(pipe);
        // init(th) behavior inside SmbFileInputStream constructor
        when(tree.isSMB2()).thenReturn(smb2);
        when(tree.getReceiveBufferSize()).thenReturn(4096);
        when(tree.getMaximumBufferSize()).thenReturn(65535);
        when(tree.hasCapability(anyInt())).thenReturn(false);

        SmbPipeHandleImpl handle = newHandleWithBasicStubs(0, "\\\\pipe\\\\foo");
        when(pipe.ensureTreeConnected()).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true); // simplifies stream init
        when(tree.getReceiveBufferSize()).thenReturn(4096);

        // Act: first calls construct streams, second calls return cached

    @Mock
    Configuration mockConfig;

    private SmbFileInputStream newStream() throws SmbException {
        // Constructor that avoids external I/O; stays in-memory
        when(mockTree.isSMB2()).thenReturn(true); // default SMB2 for simple happy-path
        return new SmbFileInputStream(mockFile, mockTree, mockHandle);
    }

    @BeforeEach
    void setUp() throws Exception {

    private void setupSmb2(NotifyResponse resp, byte[] fileId) throws Exception {
        when(handle.isValid()).thenReturn(true);
        when(handle.getTree()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);
        when(tree.getConfig()).thenReturn(mock(Configuration.class));
        when(handle.getFileId()).thenReturn(fileId);

            try {
                entries = doMsrpcShareEnum(tc, locator, transport.getRemoteAddress());
            } catch (final IOException ioe) {
                if (th.isSMB2()) {
                    throw ioe;
                }
                log.debug("doMsrpcShareEnum failed", ioe);
                entries = doNetShareEnum(th);
            }

    void testIsOpenAndIsStale() throws CIFSException {
        when(pipe.ensureTreeConnected()).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);

        // SMB2 branch in ensureOpen uses openUnshared(String, ...)
        when(pipe.openUnshared(anyString(), anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(fileHandle);

            throw new SmbException("Pipe handle already closed");
        }

        if (!isOpen()) {
            try (SmbTreeHandleImpl th = ensureTreeConnected()) {

                if (th.isSMB2()) {
                    this.handle = this.pipe.openUnshared(this.uncPath, 0, this.access, this.sharing, SmbConstants.ATTR_NORMAL, 0);
                    return this.handle.acquire();
                }