75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
import py
|
|
|
|
class FigleafPlugin:
|
|
def pytest_addoption(self, parser):
|
|
group = parser.addgroup('figleaf options')
|
|
group.addoption('-F', action='store_true', default=False,
|
|
dest = 'figleaf',
|
|
help=('trace coverage with figleaf and write HTML '
|
|
'for files below the current working dir'))
|
|
group.addoption('--figleaf-data', action='store', default='.figleaf',
|
|
dest='figleafdata',
|
|
help='path coverage tracing file.')
|
|
group.addoption('--figleaf-html', action='store', default='html',
|
|
dest='figleafhtml',
|
|
help='path to the coverage html dir.')
|
|
|
|
def pytest_configure(self, config):
|
|
if config.getvalue('figleaf'):
|
|
try:
|
|
import figleaf
|
|
import figleaf.annotate_html
|
|
except ImportError:
|
|
raise config.Error('Could not import figleaf module')
|
|
self.figleaf = figleaf
|
|
self.figleaf.start()
|
|
|
|
def pytest_terminal_summary(self, terminalreporter):
|
|
if hasattr(self, 'figleaf'):
|
|
config = terminalreporter.config
|
|
datafile = py.path.local(config.getvalue('figleafdata'))
|
|
tw = terminalreporter._tw
|
|
tw.sep('-', 'figleaf')
|
|
tw.line('Writing figleaf data to %s' % (datafile))
|
|
self.figleaf.stop()
|
|
self.figleaf.write_coverage(str(datafile))
|
|
coverage = self.get_coverage(datafile, config)
|
|
|
|
reportdir = py.path.local(config.getvalue('figleafhtml'))
|
|
tw.line('Writing figleaf html to file://%s' % (reportdir))
|
|
self.figleaf.annotate_html.prepare_reportdir(str(reportdir))
|
|
exclude = []
|
|
self.figleaf.annotate_html.report_as_html(coverage,
|
|
str(reportdir), exclude, {})
|
|
|
|
def get_coverage(self, datafile, config):
|
|
# basepath = config.topdir
|
|
basepath = py.path.local()
|
|
data = self.figleaf.read_coverage(str(datafile))
|
|
d = {}
|
|
coverage = self.figleaf.combine_coverage(d, data)
|
|
for path in coverage.keys():
|
|
if not py.path.local(path).relto(basepath):
|
|
del coverage[path]
|
|
return coverage
|
|
|
|
|
|
def test_generic(plugintester):
|
|
plugintester.apicheck(FigleafPlugin)
|
|
|
|
def test_functional(testdir):
|
|
py.test.importorskip("figleaf")
|
|
testdir.plugins.append('figleaf')
|
|
testdir.makepyfile("""
|
|
def f():
|
|
x = 42
|
|
def test_whatever():
|
|
pass
|
|
""")
|
|
result = testdir.runpytest('-F')
|
|
assert result.ret == 0
|
|
assert result.stdout.fnmatch_lines([
|
|
'*figleaf html*'
|
|
])
|
|
#print result.stdout.str()
|