mirror of
https://github.com/isc-projects/bind9.git
synced 2026-02-27 20:11:12 -05:00
107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
############################################################################
|
|
# Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
#
|
|
# 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 os
|
|
import pytest
|
|
|
|
|
|
def pytest_configure(config):
|
|
config.addinivalue_line(
|
|
"markers", "requests: mark tests that need requests to function"
|
|
)
|
|
config.addinivalue_line(
|
|
"markers", "json: mark tests that need json to function"
|
|
)
|
|
config.addinivalue_line(
|
|
"markers", "xml: mark tests that need xml.etree to function"
|
|
)
|
|
config.addinivalue_line(
|
|
"markers", "dnspython: mark tests that need dnspython to function"
|
|
)
|
|
|
|
|
|
def pytest_collection_modifyitems(config, items):
|
|
# pylint: disable=unused-argument,unused-import,too-many-branches
|
|
# pylint: disable=import-outside-toplevel
|
|
# Test for requests module
|
|
skip_requests = pytest.mark.skip(
|
|
reason="need requests module to run")
|
|
try:
|
|
import requests # noqa: F401
|
|
except ModuleNotFoundError:
|
|
for item in items:
|
|
if "requests" in item.keywords:
|
|
item.add_marker(skip_requests)
|
|
# Test for json module
|
|
skip_json = pytest.mark.skip(
|
|
reason="need json module to run")
|
|
try:
|
|
import json # noqa: F401
|
|
except ModuleNotFoundError:
|
|
for item in items:
|
|
if "json" in item.keywords:
|
|
item.add_marker(skip_json)
|
|
# Test for xml module
|
|
skip_xml = pytest.mark.skip(
|
|
reason="need xml module to run")
|
|
try:
|
|
import xml.etree.ElementTree # noqa: F401
|
|
except ModuleNotFoundError:
|
|
for item in items:
|
|
if "xml" in item.keywords:
|
|
item.add_marker(skip_xml)
|
|
# Test if JSON statistics channel was enabled
|
|
no_jsonstats = pytest.mark.skip(
|
|
reason="need JSON statistics to be enabled")
|
|
if os.getenv("HAVEJSONSTATS") is None:
|
|
for item in items:
|
|
if "json" in item.keywords:
|
|
item.add_marker(no_jsonstats)
|
|
# Test if XML statistics channel was enabled
|
|
no_xmlstats = pytest.mark.skip(
|
|
reason="need XML statistics to be enabled")
|
|
if os.getenv("HAVEXMLSTATS") is None:
|
|
for item in items:
|
|
if "xml" in item.keywords:
|
|
item.add_marker(no_xmlstats)
|
|
# Test for dnspython module
|
|
skip_dnspython = pytest.mark.skip(
|
|
reason="need dnspython module to run")
|
|
try:
|
|
import dns.query # noqa: F401
|
|
except ModuleNotFoundError:
|
|
for item in items:
|
|
if "dnspython" in item.keywords:
|
|
item.add_marker(skip_dnspython)
|
|
|
|
|
|
@pytest.fixture
|
|
def statsport(request):
|
|
# pylint: disable=unused-argument
|
|
env_port = os.getenv("EXTRAPORT1")
|
|
if port is None:
|
|
env_port = 5301
|
|
else:
|
|
env_port = int(env_port)
|
|
|
|
return env_port
|
|
|
|
|
|
@pytest.fixture
|
|
def port(request):
|
|
# pylint: disable=unused-argument
|
|
env_port = os.getenv("PORT")
|
|
if port is None:
|
|
env_port = 5300
|
|
else:
|
|
env_port = int(env_port)
|
|
|
|
return env_port
|