assertTrue(tree.isConnected());
        assertEquals("A:", tree.getService());
    }

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