blob: 1b865fb522b8386d13a11ad3b31851d01521f913 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import ipaddress
# These are IPs of a.ns.nic.cz
AUTHORITATIVE_SERVERS = [ipaddress.IPv4Address("194.0.12.1"),
ipaddress.IPv6Address("2001:678:f::1")]
def pytest_addoption(parser):
parser.addoption("--forwarder", action="append", help="IP of forwarder to test")
def pytest_generate_tests(metafunc):
if 'forwarder' in metafunc.fixturenames:
forwarder = metafunc.config.option.forwarder
metafunc.parametrize("forwarder", [ipaddress.ip_address(f) for f in forwarder], ids=str)
if 'tcp' in metafunc.fixturenames:
metafunc.parametrize("tcp", [False, True], ids=lambda x: "TCP" if x else "UDP")
if 'server' in metafunc.fixturenames:
metafunc.parametrize("server", AUTHORITATIVE_SERVERS, ids=str)
|