1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
# 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 concurrent.futures
import os
import subprocess
import time
import dns.query
import dns.update
def rndc_loop(test_state, server):
rndc = os.getenv("RNDC")
port = os.getenv("CONTROLPORT")
cmdline = [
rndc,
"-c",
"../common/rndc.conf",
"-p",
port,
"-s",
server,
"reload",
]
while not test_state["finished"]:
subprocess.run(cmdline, check=False)
time.sleep(1)
def update_zone(test_state, zone, named_port):
server = "10.53.0.2"
for i in range(1000):
if test_state["finished"]:
return
update = dns.update.UpdateMessage(zone)
update.add(f"dynamic-{i}.{zone}", 300, "TXT", f"txt-{i}")
try:
response = dns.query.udp(update, server, 10, named_port)
assert response.rcode() == dns.rcode.NOERROR
except dns.exception.Timeout:
print(f"error: query timeout for {zone}")
print(f"Update of {server} zone {zone} successful")
# If the test has run to completion without named crashing, it has succeeded.
def test_update_stress(named_port):
test_state = {"finished": False}
with concurrent.futures.ThreadPoolExecutor() as executor:
executor.submit(rndc_loop, test_state, "10.53.0.3")
updaters = []
for i in range(5):
zone = f"zone00000{i}.example."
updaters.append(executor.submit(update_zone, test_state, zone, named_port))
# All the update_zone() tasks are expected to complete within 5
# minutes. If they do not, we cannot assert immediately as that will
# cause the ThreadPoolExecutor context manager to wait indefinitely;
# instead, we first signal all tasks that it is time to exit and only
# check whether any task failed to finish within 5 minutes outside of
# the ThreadPoolExecutor context manager.
unfinished_tasks = concurrent.futures.wait(updaters, timeout=300).not_done
test_state["finished"] = True
assert not unfinished_tasks
|
