mirror of https://github.com/ansible/ansible.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.5 KiB
Python
68 lines
1.5 KiB
Python
"""Monkey patch os._exit when running under coverage so we don't lose coverage data in forks, such as with `pytest --boxed`. PYTEST_DONT_REWRITE"""
|
|
from __future__ import annotations
|
|
|
|
|
|
def pytest_configure():
|
|
"""Configure this pytest plugin."""
|
|
try:
|
|
if pytest_configure.executed:
|
|
return
|
|
except AttributeError:
|
|
pytest_configure.executed = True
|
|
|
|
try:
|
|
import coverage
|
|
except ImportError:
|
|
coverage = None
|
|
|
|
try:
|
|
coverage.Coverage
|
|
except AttributeError:
|
|
coverage = None
|
|
|
|
if not coverage:
|
|
return
|
|
|
|
import gc
|
|
import os
|
|
|
|
coverage_instances = []
|
|
|
|
for obj in gc.get_objects():
|
|
if isinstance(obj, coverage.Coverage):
|
|
coverage_instances.append(obj)
|
|
|
|
if not coverage_instances:
|
|
coverage_config = os.environ.get('COVERAGE_CONF')
|
|
|
|
if not coverage_config:
|
|
return
|
|
|
|
coverage_output = os.environ.get('COVERAGE_FILE')
|
|
|
|
if not coverage_output:
|
|
return
|
|
|
|
cov = coverage.Coverage(config_file=coverage_config)
|
|
coverage_instances.append(cov)
|
|
else:
|
|
cov = None
|
|
|
|
# noinspection PyProtectedMember
|
|
os_exit = os._exit # pylint: disable=protected-access
|
|
|
|
def coverage_exit(*args, **kwargs):
|
|
for instance in coverage_instances:
|
|
instance.stop()
|
|
instance.save()
|
|
|
|
os_exit(*args, **kwargs)
|
|
|
|
os._exit = coverage_exit # pylint: disable=protected-access
|
|
|
|
if cov:
|
|
cov.start()
|
|
|
|
|
|
pytest_configure()
|