# Copyright (C) Internet Systems Consortium, Inc. ("ISC")
#
# SPDX-License-Identifier: MPL-2.0
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0.  If a copy of the MPL was not distributed with this
# file, you can obtain one at https://mozilla.org/MPL/2.0/.
#
# See the COPYRIGHT file distributed with this work for additional
# information regarding copyright ownership.

import dns.flags
import dns.message
import dns.rcode
import pytest

import isctest

pytestmark = pytest.mark.extra_artifacts(
    [
        "ns2/example.bk",
        "ns3/example.bk",
    ]
)


@pytest.mark.parametrize("ns", [2, 3])
def test_dialup_zone_transfer(named_port, servers, ns):
    msg = dns.message.make_query("example.", "SOA")
    # Drop the RD flag from the query
    msg.flags &= ~dns.flags.RD
    ns1response = isctest.query.tcp(msg, "10.53.0.1")
    with servers[f"ns{ns}"].watch_log_from_start(timeout=90) as watcher:
        watcher.wait_for_line(
            isctest.transfer.transfer_message(
                "example", f"10.53.0.{ns-1}", "Transfer status: success", named_port
            )
        )
    response = isctest.query.tcp(msg, f"10.53.0.{ns}")
    if response.rcode() != dns.rcode.SERVFAIL:
        assert response.answer == ns1response.answer
        assert response.authority == ns1response.authority
