verify(transport).ensureConnected();
    }

    // Happy path: getDfsReferrals returns referral, verify arguments captured
    @Test
    @DisplayName("getDfsReferrals returns data and receives correct args")
    void getDfsReferrals_happy() throws Exception {
        when(transport.getDfsReferrals(any(), any(), any(), any(), anyInt())).thenReturn(referral);

            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;
                do {
                    final String domain = dr.server.toLowerCase();

        when(mockTransportPool.getSmbTransport(any(), anyString(), anyInt(), anyBoolean(), anyBoolean())).thenReturn(mockTransport);
        when(mockTransport.unwrap(any())).thenReturn(mockTransport);
        when(mockTransport.getDfsReferrals(any(), anyString(), anyString(), anyString(), anyInt())).thenReturn(null);
        assertNull(dfsResolver.getDc(mockContext, "anyDomain"));
    }

    // Test for resolve method
    @Test

        case NtStatus.NT_STATUS_PATH_NOT_COVERED:
            if (req.auth == null) {
                throw new SmbException(resp.errorCode, null);
            }

            final DfsReferral dr = getDfsReferrals(req.auth, req.path, 1);
            if (dr == null) {
                throw new SmbException(resp.errorCode, null);
            }

            SmbFile.dfs.insert(req.path, dr);
            throw dr;

            }
        }

        return result;
    }

    @Override
    public DfsReferralData getDfsReferrals(CIFSContext ctx, String name, String targetHost, String targetDomain, int rn)
            throws CIFSException {
        return delegate.getDfsReferrals(ctx, name, targetHost, targetDomain, rn);
    }

    @Override
    public boolean isSigningOptional() throws SmbException {

        DfsReferral referral = new DfsReferral();
        referral.server = "dc1.domain.com";
        referral.next = referral;
        lenient().when(smbTransport.getDfsReferrals(auth, "\\domain.com", 1)).thenReturn(referral);

        UniAddress dcAddress = mock(UniAddress.class);
        lenient().when(UniAddress.getByName("dc1.domain.com")).thenReturn(dcAddress);

     * @param rn the referral number
     * @return dfs referral
     * @throws SmbException if an SMB error occurs
     * @throws CIFSException if a CIFS error occurs
     */
    DfsReferralData getDfsReferrals(CIFSContext ctx, String name, String targetHost, String targetDomain, int rn) throws CIFSException;

    /**
     * Checks if message signing is supported but not mandatory.
     *

                final SmbTransportInternal trans = dc != null ? dc.unwrap(SmbTransportInternal.class) : null;
                if (trans != null) {
                    // get domain referral
                    initial = trans.getDfsReferrals(tf.withAnonymousCredentials(), "", trans.getRemoteHostName(), authDomain, 0);
                }
                if (initial != null) {

    }

    @Test
    @DisplayName("DFS referrals: invalid double-slash prefix triggers exception")
    void dfsReferrals_invalidPath() {
        CIFSException ex = assertThrows(CIFSException.class, () -> transport.getDfsReferrals(ctx, "\\\\server\\share", null, null, 1));
        assertTrue(ex.getMessage().contains("double slash"));
    }

    @Nested
    @MockitoSettings(strictness = Strictness.LENIENT)

        DfsReferralData dr = null;
        if (!getContext().getConfig().isDfsDisabled()) {
            try {
                dr = getDfsReferrals(getContext(), path, req.getServer(), req.getDomain(), 0);
            } catch (final CIFSException e) {
                throw new SmbException("Failed to get DFS referral", e);
            }
        }