connectionState = 2; // connected
            } catch (final SmbException se) {
                treeDisconnect(true);
                connectionState = 0;
                throw se;
            }
        }
    }

    void treeDisconnect(final boolean inError) {
        synchronized (session.transport()) {

            if (connectionState != 2) { // not-connected

        tree.treeConnect(null, null);

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

    // Test case for treeDisconnect method
    @Test
    void testTreeDisconnect() throws CIFSException, IOException {
        when(transport.isSMB2()).thenReturn(true);
        when(transport.getContext()).thenReturn(context);