ChannelInfo channel = channelManager.selectChannel(message);
                return (SmbTransportImpl) channel.getTransport();
            } catch (Exception e) {
                log.debug("Multi-channel selection failed, using primary transport", e);
            }
        }
        return getTransport();
    }

    @Override
    public String toString() {

     */
    byte[] getSessionKey() throws CIFSException;

    /**
     * Returns the SMB transport associated with this session.
     *
     * @return the transport for this session
     */
    SmbTransport getTransport();

    /**
     * Connect to the logon share
     *
     * @throws SmbException if the connection fails
     */
    void treeConnectLogon() throws SmbException;

    /**

        when(treeConnection.acquire()).thenReturn(treeConnection);
        when(treeConnection.getSession()).thenReturn(session);
        when(session.getTransport()).thenReturn(transport);
        handle = new SmbTreeHandleImpl(resourceLoc, treeConnection);
    }

    @Test
    @DisplayName("Constructor acquires tree connection; getConfig/isConnected delegate")

    }

    @Test
    void testConstructor() {
        assertEquals("test-channel", channelInfo.getChannelId());
        assertEquals(mockTransport, channelInfo.getTransport());
        assertEquals(localInterface, channelInfo.getLocalInterface());
        assertEquals(remoteInterface, channelInfo.getRemoteInterface());
        assertEquals(ChannelState.DISCONNECTED, channelInfo.getState());

        channel.setState(ChannelState.ESTABLISHED);

        // Verify channel properties
        assertEquals("test-channel", channel.getChannelId());
        assertEquals(mockTransport, channel.getTransport());
        assertEquals(localNic, channel.getLocalInterface());
        assertEquals(remoteNic, channel.getRemoteInterface());
        assertEquals(ChannelState.ESTABLISHED, channel.getState());

    void setUp() throws Exception {
        MockitoAnnotations.openMocks(this);
        when(session.getConfig()).thenReturn(config);
        when(session.getContext()).thenReturn(context);
        when(session.getTransport()).thenReturn(transport);
        when(session.acquire()).thenReturn(session);
        when(config.isTraceResourceUsage()).thenReturn(false);
        when(context.getConfig()).thenReturn(config);

        try (SmbTreeHandleImpl th = treeConn.connectHost(locator, locator.getServerWithDfs());
                SmbSessionImpl session = th.getSession();
                SmbTransportImpl transport = session.getTransport()) {
            try {
                entries = doMsrpcShareEnum(tc, locator, transport.getRemoteAddress());
            } catch (final IOException ioe) {
                if (th.isSMB2()) {

        when(tree.isConnected()).thenReturn(true);
        when(tree.getSession()).thenReturn(session);
        when(tree.acquire(false)).thenReturn(tree);
        when(session.getTransport()).thenReturn(transport);
        when(transport.isDisconnected()).thenReturn(false);
        when(transport.getRemoteHostName()).thenReturn("host");
        setTree(c, tree);

    
    // Write some data
    try (OutputStream os = file.getOutputStream()) {
        os.write("test data".getBytes());
    }
    
    // Simulate network disconnection
    file.getTree().getSession().getTransport().disconnect();
    
    // Try to read - should trigger reconnection
    try (InputStream is = file.getInputStream()) {
        byte[] buffer = new byte[9];
        is.read(buffer);

        }

        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;