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

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

    }

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

        when(mockConfig.getDfsTtl()).thenReturn(300L);

        // Mock transport to throw IOException - simulating connection failure
        when(mockTransportPool.getSmbTransport(any(CIFSContext.class), anyString(), anyInt(), anyBoolean(), anyBoolean()))
                .thenThrow(new IOException("Connection failed"));

        // Instantiate the class under test