fa2adf3b7b
* Update pytest plugins. * Fix update-bundled sanity test. * Remove old validate-modules comment. * Fix ansible-test plugin loading * Update code coverage comments. * Fix Makefile ansible-test reference. * Remove incorrect path from lint output. * Update ansible-test unit tests. * Make ansible-test's own unit tests singular.
59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
"""Monkey patch os._exit when running under coverage so we don't lose coverage data in forks, such as with `pytest --boxed`."""
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
|
|
def pytest_configure():
|
|
"""Configure this pytest plugin."""
|
|
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()
|