mirror of
https://github.com/isc-projects/bind9.git
synced 2026-04-29 09:59:08 -04:00
9549c8885a
Make sure the queries and responses are logged at the DEBUG level, which
may provide useful information in case of failing tests.
This doesn't seem to significantly increase the overall artifacts size.
Previously, pytest.log.txt files from all system tests would take around
3 MB, with this change, it's around 8 MB).
(cherry picked from commit 56fec9ba04)
52 lines
1.5 KiB
Python
52 lines
1.5 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", log_response=False)
|
|
|
|
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", log_response=False)
|
|
|
|
assert res.flags & dns.flags.TC, "TC flag was not set"
|