when(ctx.getTransportPool()).thenReturn(poolSpy);

        // Mock transports
        SmbTransportImpl trans1 = mock(SmbTransportImpl.class);
        SmbTransportImpl trans2 = mock(SmbTransportImpl.class);

        // trans2 fails first (lower fail count, tried first), trans1 succeeds
        when(trans2.unwrap(SmbTransportImpl.class)).thenReturn(trans2);

    private final ConcurrentLinkedQueue<SmbTransportImpl> connections = new ConcurrentLinkedQueue<>();
    private final ConcurrentLinkedQueue<SmbTransportImpl> nonPooledConnections = new ConcurrentLinkedQueue<>();
    private final ConcurrentLinkedQueue<SmbTransportImpl> toRemove = new ConcurrentLinkedQueue<>();
    final Map<String, Integer> failCounts = new ConcurrentHashMap<>();

        try (SmbSessionImpl session = this.treeConnection.getSession(); SmbTransportImpl transport = session.getTransport()) {
            return transport.getNegotiateResponse().getSendBufferSize();
        }
    }

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

    void signingModes() throws Exception {
        // Enforced via constructor flag -> optional false, enforced true regardless of negotiation
        SmbTransportImpl enforced = new SmbTransportImpl(ctx, address, 445, null, 0, true);
        assertFalse(enforced.isSigningOptional());
        assertTrue(enforced.isSigningEnforced());

            // Multi-channel transports should use signing consistently with the main session
            boolean forceSigning = context.getConfig().isSigningEnforced();

            SmbTransportImpl transport = new SmbTransportImpl(context, serverUniAddress, port, localAddress, localPort, forceSigning);

            // Ensure the transport is connected and ready
            transport.ensureConnected();

     */
    public boolean isPossiblyDfs() throws SmbException {
        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
     */

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

    private CIFSContext mockContext;
    private SmbTransportImpl mockTransport;
    private Configuration mockConfig;
    private Credentials mockCredentials;
    private CredentialsInternal mockCredentialsInternal;

    @BeforeEach
    public void setUp() {
        mockContext = Mockito.mock(CIFSContext.class);
        mockTransport = Mockito.mock(SmbTransportImpl.class);
        mockConfig = Mockito.mock(Configuration.class);

    private CIFSContext context;

    @Mock
    private Configuration config;

    @Mock
    private Address address;

    @Mock
    private Smb2NegotiateResponse negotiateResponse;

    private SmbTransportImpl transport;

    @BeforeEach
    void setUp() throws Exception {
        when(context.getConfig()).thenReturn(config);
        when(config.isSigningEnforced()).thenReturn(false);

            return null;
        }
        final SmbTransportImpl transport = getReferralTransport(tf, getDcReferrals(tf, domain));
        if (transport == null && log.isDebugEnabled()) {
            log.debug(String.format("Failed to connect to domain controller for %s", domain));
        }
        return transport;
    }