assertEquals(tree1, tree2);
        assertFalse(tree1.equals(tree3));
    }

    // Test case for the unwrap method
    @Test
    void testUnwrap() {
        SmbTreeImpl tree = new SmbTreeImpl(session, "SHARE", "A:");
        SmbTreeImpl unwrapped = tree.unwrap(SmbTreeImpl.class);
        assertEquals(tree, unwrapped);
        // Test unwrapping to a non-assignable type should fail

    }

    @Test
    @DisplayName("close() calls handle.close when open and handle.release otherwise; releases tree")
    void testCloseBehavior() throws CIFSException {
        when(pipe.ensureTreeConnected()).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);
        when(pipe.openUnshared(anyString(), anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(fileHandle);

    }

    @Test
    @DisplayName("acquire calls tree.acquire on first usage, release calls tree.release at zero")
    void acquire_release_lifecycle() {
        SmbTreeConnection c = newConn();
        SmbTreeImpl tree = mock(SmbTreeImpl.class);
        when(tree.acquire(false)).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);

        setTree(c, tree);

        // Act: first acquire triggers tree.acquire()

        SmbPipeInputStream stream = newStreamWithMinimalStubs(true);

        when(handle.ensureOpen()).thenReturn(fd);
        when(fd.getTree()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);
        when(tree.getConfig()).thenReturn(config);
        when(config.getTransactionBufferSize()).thenReturn(65535);

    void unwrap_and_close_interactions() {
        // Arrange
        when(tree.unwrap(SmbTreeInternal.class)).thenReturn(tree);

        // Act
        SmbTree unwrapped = tree.unwrap(SmbTreeInternal.class);
        tree.close();

        // Assert
        assertNotNull(unwrapped);
        verify(tree).unwrap(SmbTreeInternal.class);
        verify(tree).close();
    }

 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package jcifs.internal;

/**
 * Interface for SMB Tree Connect response messages.
 * Handles the server's response to a tree connect request, providing tree ID,
 * service type, and DFS information for the connected share.
 *
 * @author mbechler
 */

        // Setup the tree handle properly
        when(pipe.ensureTreeConnected()).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);

        // Setup for ensureOpen
        when(tree.isSMB2()).thenReturn(false);
        when(tree.hasCapability(SmbConstants.CAP_NT_SMBS)).thenReturn(true);

    tree.put("tree", 3);
    assertThat(tree.keySet()).containsExactly("tree", "google").inOrder();
    assertThat(tree.get("google")).containsExactly(6, 2).inOrder();

    TreeMultimap<String, Integer> copy = TreeMultimap.create(tree);
    assertEquals(tree, copy);
    assertThat(copy.keySet()).containsExactly("google", "tree").inOrder();
    assertThat(copy.get("google")).containsExactly(2, 6).inOrder();

        TestIterator it = TestIterator.create(tree, parent, wildcard, null, attrs, initial, pages);

        // Assert
        assertSame(tree, it.getTreeHandle(), "tree handle should be same instance returned by acquire");
        assertEquals(attrs, it.getSearchAttributes());
        assertEquals(wildcard, it.getWildcard());
        assertSame(parent, it.getParent());
        verify(tree, times(1)).acquire();
    }

    @Test

            log.debug("Switching tree");
            if (t != null) {
                log.debug("Acquired tree on switch " + t);
                t.acquire();
                this.treeAcquired = true;
            } else {
                this.treeAcquired = false;
            }

            this.tree = t;
            if ((old != null) && wasAcquired) {
                // release