verify(session).treeConnectLogon();
    }

    // Happy path: getSmbTree returns a tree and receives correct args
    @Test
    @DisplayName("getSmbTree returns tree and captures arguments")
    void getSmbTree_happy() {
        when(session.getSmbTree(anyString(), anyString())).thenReturn(tree);

        String share = "share";

        // 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
        session.release();

        assertArrayEquals(key, session.getSessionKey());
    }

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

                          Skipping DC interrogation.""");
            }
        } else {
            final SmbSession ssn = trans.getSmbSession(NtlmPasswordAuthentication.DEFAULT);
            ssn.getSmbTree(LOGON_SHARE, null).treeConnect(null, null);
        }
        return new NtlmChallenge(trans.server.encryptionKey, dc);
    }

    /**

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

     *
     * @param share the share name to connect to
     * @param service the service type for the connection
     * @return tree instance
     */
    SmbTree getSmbTree(String share, String service);

    /**
     * Initiate reauthentication
     *
     * @throws CIFSException if reauthentication fails
     */
    void reauthenticate() throws CIFSException;

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

            ri++;
        }
    }

    DfsReferral getDfsReferrals(final NtlmPasswordAuthentication auth, final String path, int rn) throws SmbException {
        final SmbTree ipc = getSmbSession(auth).getSmbTree("IPC$", null);
        final Trans2GetDfsReferralResponse resp = new Trans2GetDfsReferralResponse();
        ipc.send(new Trans2GetDfsReferral(path), resp);

        if (resp.numReferrals == 0) {

                            // Test concurrent tree operations (should be thread-safe with CopyOnWriteArrayList)
                            long opStart = System.nanoTime();
                            session.getSmbTree(shareName, null);
                            long opEnd = System.nanoTime();

                            totalTime.addAndGet(opEnd - opStart);
                            treeOperations.incrementAndGet();

        when(internal.getSmbSession(eq(ctx), eq("server.test"), isNull())).thenReturn(session);
        when(session.unwrap(SmbSessionInternal.class)).thenReturn(session);
        when(session.getSmbTree(eq("IPC$"), isNull())).thenReturn(tree);
        when(tree.unwrap(SmbTreeInternal.class)).thenReturn(tree);

        doReturn(mockTransport).when(poolSpy).getSmbTransport(eq(ctx), eq(address), anyInt(), eq(false), anyBoolean());