mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-11-29 16:20:35 +00:00
324 lines
12 KiB
Python
324 lines
12 KiB
Python
"""Coverage plugin for pytest."""
|
|
import os
|
|
import warnings
|
|
|
|
import pytest
|
|
import argparse
|
|
|
|
from coverage.misc import CoverageException
|
|
|
|
from . import embed
|
|
from . import engine
|
|
from . import compat
|
|
|
|
|
|
class CoverageError(Exception):
|
|
"""Indicates that our coverage is too low"""
|
|
|
|
|
|
def validate_report(arg):
|
|
file_choices = ['annotate', 'html', 'xml']
|
|
term_choices = ['term', 'term-missing']
|
|
term_modifier_choices = ['skip-covered']
|
|
all_choices = term_choices + file_choices
|
|
values = arg.split(":", 1)
|
|
report_type = values[0]
|
|
if report_type not in all_choices + ['']:
|
|
msg = 'invalid choice: "{}" (choose from "{}")'.format(arg, all_choices)
|
|
raise argparse.ArgumentTypeError(msg)
|
|
|
|
if len(values) == 1:
|
|
return report_type, None
|
|
|
|
report_modifier = values[1]
|
|
if report_type in term_choices and report_modifier in term_modifier_choices:
|
|
return report_type, report_modifier
|
|
|
|
if report_type not in file_choices:
|
|
msg = 'output specifier not supported for: "{}" (choose from "{}")'.format(arg,
|
|
file_choices)
|
|
raise argparse.ArgumentTypeError(msg)
|
|
|
|
return values
|
|
|
|
|
|
class StoreReport(argparse.Action):
|
|
def __call__(self, parser, namespace, values, option_string=None):
|
|
report_type, file = values
|
|
namespace.cov_report[report_type] = file
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
"""Add options to control coverage."""
|
|
|
|
group = parser.getgroup(
|
|
'cov', 'coverage reporting with distributed testing support')
|
|
group.addoption('--cov', action='append', default=[], metavar='path',
|
|
nargs='?', const=True, dest='cov_source',
|
|
help='measure coverage for filesystem path '
|
|
'(multi-allowed)')
|
|
group.addoption('--cov-report', action=StoreReport, default={},
|
|
metavar='type', type=validate_report,
|
|
help='type of report to generate: term, term-missing, '
|
|
'annotate, html, xml (multi-allowed). '
|
|
'term, term-missing may be followed by ":skip-covered". '
|
|
'annotate, html and xml may be followed by ":DEST" '
|
|
'where DEST specifies the output location. '
|
|
'Use --cov-report= to not generate any output.')
|
|
group.addoption('--cov-config', action='store', default='.coveragerc',
|
|
metavar='path',
|
|
help='config file for coverage, default: .coveragerc')
|
|
group.addoption('--no-cov-on-fail', action='store_true', default=False,
|
|
help='do not report coverage if test run fails, '
|
|
'default: False')
|
|
group.addoption('--no-cov', action='store_true', default=False,
|
|
help='Disable coverage report completely (useful for debuggers) '
|
|
'default: False')
|
|
group.addoption('--cov-fail-under', action='store', metavar='MIN', type=int,
|
|
help='Fail if the total coverage is less than MIN.')
|
|
group.addoption('--cov-append', action='store_true', default=False,
|
|
help='do not delete coverage but append to current, '
|
|
'default: False')
|
|
group.addoption('--cov-branch', action='store_true', default=None,
|
|
help='Enable branch coverage.')
|
|
|
|
|
|
def _prepare_cov_source(cov_source):
|
|
"""
|
|
Prepare cov_source so that:
|
|
|
|
--cov --cov=foobar is equivalent to --cov (cov_source=None)
|
|
--cov=foo --cov=bar is equivalent to cov_source=['foo', 'bar']
|
|
"""
|
|
return None if True in cov_source else [path for path in cov_source if path is not True]
|
|
|
|
|
|
@pytest.mark.tryfirst
|
|
def pytest_load_initial_conftests(early_config, parser, args):
|
|
if early_config.known_args_namespace.cov_source:
|
|
plugin = CovPlugin(early_config.known_args_namespace, early_config.pluginmanager)
|
|
early_config.pluginmanager.register(plugin, '_cov')
|
|
|
|
|
|
class CovPlugin(object):
|
|
"""Use coverage package to produce code coverage reports.
|
|
|
|
Delegates all work to a particular implementation based on whether
|
|
this test process is centralised, a distributed master or a
|
|
distributed slave.
|
|
"""
|
|
|
|
def __init__(self, options, pluginmanager, start=True):
|
|
"""Creates a coverage pytest plugin.
|
|
|
|
We read the rc file that coverage uses to get the data file
|
|
name. This is needed since we give coverage through it's API
|
|
the data file name.
|
|
"""
|
|
|
|
# Our implementation is unknown at this time.
|
|
self.pid = None
|
|
self.cov = None
|
|
self.cov_controller = None
|
|
self.cov_report = compat.StringIO()
|
|
self.cov_total = None
|
|
self.failed = False
|
|
self._started = False
|
|
self._disabled = False
|
|
self.options = options
|
|
|
|
is_dist = (getattr(options, 'numprocesses', False) or
|
|
getattr(options, 'distload', False) or
|
|
getattr(options, 'dist', 'no') != 'no')
|
|
if getattr(options, 'no_cov', False):
|
|
self._disabled = True
|
|
return
|
|
|
|
if not self.options.cov_report:
|
|
self.options.cov_report = ['term']
|
|
elif len(self.options.cov_report) == 1 and '' in self.options.cov_report:
|
|
self.options.cov_report = {}
|
|
self.options.cov_source = _prepare_cov_source(self.options.cov_source)
|
|
|
|
if is_dist and start:
|
|
self.start(engine.DistMaster)
|
|
elif start:
|
|
self.start(engine.Central)
|
|
|
|
# slave is started in pytest hook
|
|
|
|
def start(self, controller_cls, config=None, nodeid=None):
|
|
|
|
if config is None:
|
|
# fake config option for engine
|
|
class Config(object):
|
|
option = self.options
|
|
|
|
config = Config()
|
|
|
|
self.cov_controller = controller_cls(
|
|
self.options.cov_source,
|
|
self.options.cov_report,
|
|
self.options.cov_config,
|
|
self.options.cov_append,
|
|
self.options.cov_branch,
|
|
config,
|
|
nodeid
|
|
)
|
|
self.cov_controller.start()
|
|
self._started = True
|
|
cov_config = self.cov_controller.cov.config
|
|
if self.options.cov_fail_under is None and hasattr(cov_config, 'fail_under'):
|
|
self.options.cov_fail_under = cov_config.fail_under
|
|
|
|
def _is_slave(self, session):
|
|
return hasattr(session.config, 'slaveinput')
|
|
|
|
def pytest_sessionstart(self, session):
|
|
"""At session start determine our implementation and delegate to it."""
|
|
|
|
if self.options.no_cov:
|
|
# Coverage can be disabled because it does not cooperate with debuggers well.
|
|
self._disabled = True
|
|
return
|
|
|
|
self.pid = os.getpid()
|
|
if self._is_slave(session):
|
|
nodeid = session.config.slaveinput.get('slaveid',
|
|
getattr(session, 'nodeid'))
|
|
self.start(engine.DistSlave, session.config, nodeid)
|
|
elif not self._started:
|
|
self.start(engine.Central)
|
|
|
|
def pytest_configure_node(self, node):
|
|
"""Delegate to our implementation.
|
|
|
|
Mark this hook as optional in case xdist is not installed.
|
|
"""
|
|
self.cov_controller.configure_node(node)
|
|
pytest_configure_node.optionalhook = True
|
|
|
|
def pytest_testnodedown(self, node, error):
|
|
"""Delegate to our implementation.
|
|
|
|
Mark this hook as optional in case xdist is not installed.
|
|
"""
|
|
self.cov_controller.testnodedown(node, error)
|
|
pytest_testnodedown.optionalhook = True
|
|
|
|
def _should_report(self):
|
|
return not (self.failed and self.options.no_cov_on_fail)
|
|
|
|
def _failed_cov_total(self):
|
|
cov_fail_under = self.options.cov_fail_under
|
|
return cov_fail_under is not None and self.cov_total < cov_fail_under
|
|
|
|
# we need to wrap pytest_runtestloop. by the time pytest_sessionfinish
|
|
# runs, it's too late to set testsfailed
|
|
@compat.hookwrapper
|
|
def pytest_runtestloop(self, session):
|
|
yield
|
|
|
|
if self._disabled:
|
|
return
|
|
|
|
compat_session = compat.SessionWrapper(session)
|
|
|
|
self.failed = bool(compat_session.testsfailed)
|
|
if self.cov_controller is not None:
|
|
self.cov_controller.finish()
|
|
|
|
if not self._is_slave(session) and self._should_report():
|
|
try:
|
|
self.cov_total = self.cov_controller.summary(self.cov_report)
|
|
except CoverageException as exc:
|
|
message = 'Failed to generate report: %s\n' % exc
|
|
session.config.pluginmanager.getplugin("terminalreporter").write(
|
|
'WARNING: %s\n' % message, red=True, bold=True)
|
|
if pytest.__version__ >= '3.8':
|
|
warnings.warn(pytest.PytestWarning(message))
|
|
else:
|
|
session.config.warn(code='COV-2', message=message)
|
|
self.cov_total = 0
|
|
assert self.cov_total is not None, 'Test coverage should never be `None`'
|
|
if self._failed_cov_total():
|
|
# make sure we get the EXIT_TESTSFAILED exit code
|
|
compat_session.testsfailed += 1
|
|
|
|
def pytest_terminal_summary(self, terminalreporter):
|
|
if self._disabled:
|
|
message = 'Coverage disabled via --no-cov switch!'
|
|
terminalreporter.write('WARNING: %s\n' % message, red=True, bold=True)
|
|
if pytest.__version__ >= '3.8':
|
|
warnings.warn(pytest.PytestWarning(message))
|
|
else:
|
|
terminalreporter.config.warn(code='COV-1', message=message)
|
|
return
|
|
if self.cov_controller is None:
|
|
return
|
|
|
|
if self.cov_total is None:
|
|
# we shouldn't report, or report generation failed (error raised above)
|
|
return
|
|
|
|
terminalreporter.write('\n' + self.cov_report.getvalue() + '\n')
|
|
|
|
if self.options.cov_fail_under is not None and self.options.cov_fail_under > 0:
|
|
if self.cov_total < self.options.cov_fail_under:
|
|
markup = {'red': True, 'bold': True}
|
|
message = (
|
|
'FAIL Required test coverage of %d%% not '
|
|
'reached. Total coverage: %.2f%%\n'
|
|
% (self.options.cov_fail_under, self.cov_total)
|
|
)
|
|
else:
|
|
markup = {'green': True}
|
|
message = (
|
|
'Required test coverage of %d%% '
|
|
'reached. Total coverage: %.2f%%\n'
|
|
% (self.options.cov_fail_under, self.cov_total)
|
|
)
|
|
terminalreporter.write(message, **markup)
|
|
|
|
def pytest_runtest_setup(self, item):
|
|
if os.getpid() != self.pid:
|
|
# test is run in another process than session, run
|
|
# coverage manually
|
|
self.cov = embed.init()
|
|
|
|
def pytest_runtest_teardown(self, item):
|
|
if self.cov is not None:
|
|
embed.cleanup(self.cov)
|
|
self.cov = None
|
|
|
|
@compat.hookwrapper
|
|
def pytest_runtest_call(self, item):
|
|
if (item.get_closest_marker('no_cover')
|
|
or 'no_cover' in getattr(item, 'fixturenames', ())):
|
|
self.cov_controller.pause()
|
|
yield
|
|
self.cov_controller.resume()
|
|
else:
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def no_cover():
|
|
pass
|
|
|
|
|
|
@pytest.fixture
|
|
def cov(request):
|
|
"""A pytest fixture to provide access to the underlying coverage object."""
|
|
|
|
# Check with hasplugin to avoid getplugin exception in older pytest.
|
|
if request.config.pluginmanager.hasplugin('_cov'):
|
|
plugin = request.config.pluginmanager.getplugin('_cov')
|
|
if plugin.cov_controller:
|
|
return plugin.cov_controller.cov
|
|
return None
|
|
|
|
|
|
def pytest_configure(config):
|
|
config.addinivalue_line("markers", "no_cover: disable coverage for this test.")
|