import jcifs.internal.smb2.nego.Smb2NegotiateResponse;
import jcifs.internal.smb2.tree.Smb2TreeConnectRequest;
import jcifs.internal.smb2.tree.Smb2TreeDisconnectRequest;

class SmbTreeImpl implements SmbTreeInternal {

    private static final Logger log = LoggerFactory.getLogger(SmbTreeImpl.class);

    private static AtomicLong TREE_CONN_COUNTER = new AtomicLong();

    /*
     * 0 - not connected
     * 1 - connecting
     * 2 - connected

        SmbTreeConnection c1 = newConn();
        SmbTreeConnection c2 = newConn();
        SmbTreeImpl shared = mock(SmbTreeImpl.class);
        when(shared.acquire(false)).thenReturn(shared);

        setTree(c1, shared);
        setTree(c2, shared);
        assertTrue(c1.isSame(c2));

        SmbTreeImpl other = mock(SmbTreeImpl.class);
        when(other.acquire(false)).thenReturn(other);
        setTree(c2, other);

                    try (SmbSessionInternal smbSession = trans.getSmbSession(this.ctx, host, targetDomain).unwrap(SmbSessionInternal.class);
                            SmbTreeImpl uct = smbSession.getSmbTree(share, null).unwrap(SmbTreeImpl.class);
                            SmbTreeImpl ct = connectTree(loc, host, share, trans, uct, dr)) {

                        if (dr != null) {
                            ct.setTreeReferral(dr);

    @Test
    void testMatches() {
        SmbTreeImpl tree = new SmbTreeImpl(session, "SHARE", "A:");
        assertTrue(tree.matches("share", "A:"));
        assertFalse(tree.matches("othershare", "A:"));
    }

    // Test case for the equals method
    @Test
    void testEquals() {
        SmbTreeImpl tree1 = new SmbTreeImpl(session, "SHARE", "A:");
        SmbTreeImpl tree2 = new SmbTreeImpl(session, "SHARE", "A:");

        }
        return this.sessionKey;
    }

    @Override
    public SmbTreeImpl getSmbTree(String share, String service) {
        if (share == null) {
            share = "IPC$";
        }

        synchronized (this.trees) {
            for (SmbTreeImpl t : this.trees) {
                if (t.matches(share, service)) {
                    return t.acquire();
                }

        // Given
        SmbSessionImpl session = new SmbSessionImpl(mockContext, "testhost", "testdomain", mockTransport);

        // Add some trees
        SmbTreeImpl tree1 = session.getSmbTree("share1", null);
        SmbTreeImpl tree2 = session.getSmbTree("share2", null);
        assertNotNull(tree1);
        assertNotNull(tree2);

        // When - Release session to trigger cleanup

    }

    @Test
    @DisplayName("getSmbTree returns same instance for same share/service")
    void testGetSmbTreeReuses() {
        SmbSessionImpl session = newSession();
        SmbTreeImpl t1 = session.getSmbTree("IPC$", null);
        SmbTreeImpl t2 = session.getSmbTree("ipc$", null); // case-insensitive match
        assertSame(t1, t2, "Expected same tree instance to be reused");
    }

    @ParameterizedTest

        // Session starts with usage count > 0
        assertTrue(session.isInUse(), "Session should be in use initially");

        // Create and release trees
        SmbTreeImpl tree = session.getSmbTree("share1", null);
        assertNotNull(tree, "Tree should be created");
        tree.release();

        // Release the session (back to initial state)
        session.release();

        }

        try (SmbSessionImpl sess = getSmbSession(ctx, targetHost, targetDomain);
                SmbTransportImpl transport = sess.getTransport();
                SmbTreeImpl ipc = sess.getSmbTree("IPC$", null)) {

            final DfsReferralRequestBuffer dfsReq = new DfsReferralRequestBuffer(path, 3);
            DfsReferralResponseBuffer dfsResp;
            if (isSMB2()) {