// Re-establish channel binding
            manager.performChannelBinding(channel);

            // Update channel with new transport
            channel.setTransport(newTransport);
            channel.setState(ChannelState.ESTABLISHED);

            // Clear failover state on success
            failoverStates.remove(channel.getChannelId());

    /**
     * Get the transport for this channel
     *
     * @return SMB transport
     */
    public SmbTransport getTransport() {
        return transport;
    }

    /**
     * Set the transport for this channel
     *
     * @param transport SMB transport
     */
    public void setTransport(SmbTransport transport) {
        this.transport = transport;
    }

    /**
     * Get local network interface

                
                // Re-establish channel binding
                manager.performChannelBinding(channel);
                
                // Update channel with new transport
                channel.setTransport(newTransport);
                channel.setState(ChannelState.ESTABLISHED);
                
                // Clear failover state on success
                failoverStates.remove(channel.getChannelId());

            return transport.getNegotiateResponse().getSendBufferSize();
        }
    }

    @Override
    public int getReceiveBufferSize() throws SmbException {
        try (SmbSessionImpl session = this.treeConnection.getSession(); SmbTransportImpl transport = session.getTransport()) {

    }

    // Happy path: transport is returned and verified
    @Test
    @DisplayName("getTransport returns mocked transport")
    void getTransport_happy() {
        when(session.getTransport()).thenReturn(transport);
        assertSame(transport, session.getTransport());
        verify(session).getTransport();
    }

    // Happy path: connect to logon share completes without exception
    @Test

        try (SmbSessionImpl session = getSession()) {
            if (session == null) {
                return;
            }
            try (SmbTransportImpl transport = session.getTransport()) {
                synchronized (transport) {
                    final SmbTreeImpl t = getTreeInternal();
                    if (t != null) {
                        try {

        if (this.connectionState.get() == 2) {
            // we are connected, so we know
            return isDfs();
        }
        try (SmbTransportImpl transport = this.session.getTransport()) {
            return transport.getNegotiateResponse().isDFSSupported();
        }
    }

    /**
     * @return the session this tree is connected in
     */
    public SmbSessionImpl getSession() {

        if (channelGroup != null) {
            for (ChannelInfo channel : channelGroup.getChannels()) {
                try {
                    if (channel.getTransport() != null) {
                        channel.getTransport().disconnect(true);
                    }
                } catch (Exception e) {
                    log.warn("Error closing channel {}: {}", channel.getChannelId(), e.getMessage());

        return channels.values().stream().filter(c -> c.getTransport() == transport).findFirst().orElse(null);
    }

    /**
     * Remove a channel
     *
     * @param channel channel to remove
     */
    public void removeChannel(ChannelInfo channel) {
        channels.remove(channel.getChannelId());
        try {
            if (channel.getTransport() != null) {
                channel.getTransport().close();

        verify(enc, times(1)).decryptMessage(eq(new byte[] { 5 }));
    }

    @Test
    @DisplayName("getTransport returns acquired transport instance")
    void testGetTransportAcquire() {
        SmbSessionImpl session = newSession();
        clearInvocations(transport);
        SmbTransportImpl t = session.getTransport();
        assertSame(transport, t);
        verify(transport, times(1)).acquire();
    }

    @Test