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
78
|
from urllib import parse
import pytest
from packaging import version
DEFAULT_NETBOX_VERSIONS = "4.0"
def pytest_addoption(parser):
"""Hook on the pytest option parser setup.
Add some extra options to the parser.
"""
parser.addoption(
"--netbox-versions",
action="store",
default=DEFAULT_NETBOX_VERSIONS,
help=(
"The versions of netbox to run integration tests against, as a"
" comma-separated list. Default: %s" % DEFAULT_NETBOX_VERSIONS
),
)
parser.addoption(
"--no-cleanup",
dest="cleanup",
action="store_false",
help=(
"Skip any cleanup steps after the pytest session finishes. Any containers"
" created will be left running and the docker-compose files used to"
" create them will be left on disk."
),
)
parser.addoption(
"--url-override",
dest="url_override",
action="store",
help=(
"Overrides the URL to run tests to. This allows for testing to the same"
" containers for separate runs."
),
)
def pytest_configure(config):
"""Hook that runs after test collection is completed.
Here we can modify items in the collected tests or parser args.
"""
# verify the netbox versions parse correctly and split them
config.option.netbox_versions = [
version.Version(version_string)
for version_string in config.option.netbox_versions.split(",")
]
if "no:docker" in config.option.plugins and config.option.url_override:
url_parse = parse.urlparse(config.option.url_override)
class DockerServicesMock:
def __init__(self, ports):
self.ports = ports
def wait_until_responsive(self, *args, **kwargs):
return None
def port_for(self, *args):
return self.ports
class Plugin:
@pytest.fixture(scope="session")
def docker_ip(self):
return "127.0.0.1"
@pytest.fixture(scope="session")
def docker_services(self):
return DockerServicesMock(url_parse.port)
config.pluginmanager.register(Plugin())
|
