Tom Krizek
1283d5263b
It is better to use consistent file names to avoid issue with sorting
etc.
Using underscore in filenames as opposed to dash was chosen because it
seems more common in pytest/python to use underscore for filenames.
Also rename the bin/tests/system/timeouts/tests-tcp.py file to
bin/tests/system/timeouts/tests_tcp_timeouts.py to avoid pytest name
collision (there can't be two files named tests_tcp.py).
(cherry picked from commit f6409ee6ac)
109 lines
2.7 KiB
Python
Executable File
109 lines
2.7 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
# 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.
|
|
|
|
from datetime import datetime
|
|
|
|
import pytest
|
|
|
|
import generic
|
|
import pytest_custom_markers
|
|
|
|
pytestmark = pytest_custom_markers.have_json_c
|
|
requests = pytest.importorskip("requests")
|
|
|
|
|
|
# JSON helper functions
|
|
def fetch_zones_json(statsip, statsport):
|
|
|
|
r = requests.get(
|
|
"http://{}:{}/json/v1/zones".format(statsip, statsport), timeout=600
|
|
)
|
|
assert r.status_code == 200
|
|
|
|
data = r.json()
|
|
return data["views"]["_default"]["zones"]
|
|
|
|
|
|
def fetch_traffic_json(statsip, statsport):
|
|
|
|
r = requests.get(
|
|
"http://{}:{}/json/v1/traffic".format(statsip, statsport), timeout=600
|
|
)
|
|
assert r.status_code == 200
|
|
|
|
data = r.json()
|
|
|
|
return data["traffic"]
|
|
|
|
|
|
def load_timers_json(zone, primary=True):
|
|
|
|
name = zone["name"]
|
|
|
|
# Check if the primary zone timer exists
|
|
assert "loaded" in zone
|
|
loaded = datetime.strptime(zone["loaded"], generic.fmt)
|
|
|
|
if primary:
|
|
# Check if the secondary zone timers does not exist
|
|
assert "expires" not in zone
|
|
assert "refresh" not in zone
|
|
expires = None
|
|
refresh = None
|
|
else:
|
|
assert "expires" in zone
|
|
assert "refresh" in zone
|
|
expires = datetime.strptime(zone["expires"], generic.fmt)
|
|
refresh = datetime.strptime(zone["refresh"], generic.fmt)
|
|
|
|
return (name, loaded, expires, refresh)
|
|
|
|
|
|
def load_zone_json(zone):
|
|
name = zone["name"]
|
|
|
|
return name
|
|
|
|
|
|
def test_zone_timers_primary_json(statsport):
|
|
generic.test_zone_timers_primary(
|
|
fetch_zones_json,
|
|
load_timers_json,
|
|
statsip="10.53.0.1",
|
|
statsport=statsport,
|
|
zonedir="ns1",
|
|
)
|
|
|
|
|
|
def test_zone_timers_secondary_json(statsport):
|
|
generic.test_zone_timers_secondary(
|
|
fetch_zones_json,
|
|
load_timers_json,
|
|
statsip="10.53.0.3",
|
|
statsport=statsport,
|
|
zonedir="ns3",
|
|
)
|
|
|
|
|
|
def test_zone_with_many_keys_json(statsport):
|
|
generic.test_zone_with_many_keys(
|
|
fetch_zones_json, load_zone_json, statsip="10.53.0.2", statsport=statsport
|
|
)
|
|
|
|
|
|
def test_traffic_json(named_port, statsport):
|
|
generic_dnspython = pytest.importorskip("generic_dnspython")
|
|
generic_dnspython.test_traffic(
|
|
fetch_traffic_json, statsip="10.53.0.2", statsport=statsport, port=named_port
|
|
)
|