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);

    @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 {

        this.signingEnforced = signingEnforced;
        setFlags2(config.getFlags2());

        if (config.getMinimumVersion().isSMB2()) {
            this.dialects = new String[] { "SMB 2.???", "SMB 2.002" };
        } else if (config.getMaximumVersion().isSMB2()) {
            this.dialects = new String[] { "NT LM 0.12", "SMB 2.???", "SMB 2.002" };
        } else {
            this.dialects = new String[] { "NT LM 0.12" };

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

        this.openFlags = openFlags;
        this.access = access;
        this.sharing = sharing;

        try (SmbTreeHandleInternal th = file.ensureTreeConnected()) {
            this.smb2 = th.isSMB2();
            if (file.getType() != SmbConstants.TYPE_NAMED_PIPE) {
                try (SmbFileHandle h = ensureOpen()) {}
                this.openFlags &= ~(SmbConstants.O_CREAT | SmbConstants.O_TRUNC);
            }

        assertTrue(tree.isInDomainDfs());
    }

    // Test case for treeConnect method with SMB1
    @Test
    void testTreeConnectSmb1() throws CIFSException, IOException {
        when(transport.isSMB2()).thenReturn(false);
        when(transport.getContext()).thenReturn(context);
        when(session.getTargetHost()).thenReturn("localhost");

        // Mock negotiate response for SMB1

        }
    }

    /**
     * @return whether this tree handle uses SMB2
     */
    @Override
    public boolean isSMB2() {
        try (SmbSessionImpl session = this.treeConnection.getSession(); SmbTransportImpl transport = session.getTransport()) {
            return transport.isSMB2();
        } catch (final SmbException e) {
            log.debug("Failed to connect for determining SMB2 support", e);

    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);

            final SmbTreeHandleImpl sh, final SmbTreeHandleImpl dh) throws SmbException {

        if (sh.isSMB2() && dh.isSMB2() && sh.isSameTree(dh)) {
            try {
                serverSideCopy(src, dest, sh, dh, false);
                return;
            } catch (final SmbUnsupportedOperationException e) {

            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();
                }