when(transportPool.getSmbTransport(context, address, DEFAULT_PORT, false)).thenReturn(transport);

            // When
            SmbTransport result = transportPool.getSmbTransport(context, address, DEFAULT_PORT, false);

            // Then
            assertNotNull(result);
            assertEquals(transport, result);
            verify(transportPool).getSmbTransport(context, address, DEFAULT_PORT, false);

        // Return mocked transports
        doReturn(trans1).when(poolSpy).getSmbTransport(eq(ctx), eq(addr1), anyInt(), anyBoolean(), anyBoolean());
        doReturn(trans2).when(poolSpy).getSmbTransport(eq(ctx), eq(addr2), anyInt(), anyBoolean(), anyBoolean());

        // When: Get transport by name
        SmbTransportImpl result = poolSpy.getSmbTransport(ctx, "test.server", 445, false, false);

        transport.server = serverData;

        smbtStatic = mockStatic(SmbTransport.class);
        smbtStatic.when(() -> SmbTransport.getSmbTransport(addr, 445, inet, 0, null)).thenReturn(transport);
        smbtStatic.when(() -> SmbTransport.getSmbTransport(addr, 445)).thenReturn(transport);
    }

    @AfterEach
    void tearDown() {
        if (smbtStatic != null) {
            smbtStatic.close();
        }

    @Override
    public SmbTransportImpl getSmbTransport(final CIFSContext tc, final Address address, final int port, final boolean nonPooled) {
        return getSmbTransport(tc, address, port, tc.getConfig().getLocalAddr(), tc.getConfig().getLocalPort(), null, nonPooled);
    }

    @Override
    public SmbTransportImpl getSmbTransport(final CIFSContext tc, final Address address, final int port, final boolean nonPooled,

     * @return a connected transport
     * @throws UnknownHostException if the host cannot be resolved
     * @throws IOException if an I/O error occurs
     */
    SmbTransport getSmbTransport(CIFSContext tf, String name, int port, boolean exclusive, boolean forceSigning)
            throws UnknownHostException, IOException;

    /**
     * Get transport connection
     *
     * @param tc

        if (_domains != null) {
            return _domains.map;
        }
        try {
            final UniAddress addr = UniAddress.getByName(auth.domain, true);
            final SmbTransport trans = SmbTransport.getSmbTransport(addr, 0);
            final CacheEntry entry = new CacheEntry(Dfs.TTL * 10L);

            DfsReferral dr = trans.getDfsReferrals(auth, "", 0);
            if (dr != null) {
                final DfsReferral start = dr;

    static int dc_list_counter;

    private static NtlmChallenge interrogate(final NbtAddress addr) throws SmbException {
        final UniAddress dc = new UniAddress(addr);
        final SmbTransport trans = SmbTransport.getSmbTransport(dc, 0);
        if (USERNAME == null) {
            trans.connect();
            if (LogStream.level >= 3) {
                SmbTransport.log.println("""

    static LogStream log = LogStream.getInstance();
    static HashMap dfsRoots = null;

    static synchronized SmbTransport getSmbTransport(final UniAddress address, final int port) {
        return getSmbTransport(address, port, LADDR, LPORT, null);
    }

    static synchronized SmbTransport getSmbTransport(final UniAddress address, final int port, final InetAddress localAddr,
            final int localPort, final String hostName) {

    }

    @Test
    void testGetDc_Success() throws IOException, SmbAuthException {
        lenient().when(UniAddress.getByName("domain.com", true)).thenReturn(uniAddress);
        lenient().when(SmbTransport.getSmbTransport(uniAddress, 0)).thenReturn(smbTransport);

        DfsReferral referral = new DfsReferral();
        referral.server = "dc1.domain.com";
        referral.next = referral;

        when(mockConfig.isDfsDisabled()).thenReturn(false);
        when(mockContext.getTransportPool()).thenReturn(mockTransportPool);
        when(mockTransportPool.getSmbTransport(any(), anyString(), anyInt(), anyBoolean(), anyBoolean())).thenReturn(mockTransport);
        when(mockTransport.unwrap(any())).thenReturn(mockTransport);