}

    // 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

     */
    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;

    /**

    /**
     * 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

        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

        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 transport.getNegotiateResponse().getSendBufferSize();
        }
    }

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

        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() {

        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();

    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);