52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
# 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 itertools
|
|
|
|
import isctest
|
|
import pytest
|
|
|
|
import dns.message
|
|
|
|
# Everything from getting a big answer to creating an RR set with thousands
|
|
# of records takes minutes of CPU and real time with dnspython < 2.0.0.
|
|
pytest.importorskip("dns", minversion="2.0.0")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"name,limit",
|
|
[
|
|
("1000", 1000),
|
|
("2000", 2000),
|
|
("3000", 3000),
|
|
("4000", 4000),
|
|
("a-maximum-rrset", 4091),
|
|
],
|
|
)
|
|
def test_limits(name, limit):
|
|
msg_query = dns.message.make_query(f"{name}.example.", "A")
|
|
res = isctest.query.tcp(msg_query, "10.53.0.1")
|
|
|
|
iplist = [
|
|
f"10.0.{x}.{y}"
|
|
for x, y in itertools.islice(itertools.product(range(256), repeat=2), limit)
|
|
]
|
|
|
|
msg_rrset = [dns.rrset.from_text_list(f"{name}.example.", "5M", "IN", "A", iplist)]
|
|
|
|
assert res.answer == msg_rrset
|
|
|
|
|
|
def test_limit_exceeded():
|
|
msg_query = dns.message.make_query("5000.example.", "A")
|
|
res = isctest.query.tcp(msg_query, "10.53.0.1")
|
|
|
|
assert res.flags & dns.flags.TC, "TC flag was not set"
|