49 lines
1.9 KiB
Python
49 lines
1.9 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')
|
|
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'):
|
|
data_file = terminalreporter.config.getvalue('figleafdata')
|
|
data_file = py.path.local(data_file)
|
|
tw = terminalreporter._tw
|
|
tw.sep('-', 'figleaf')
|
|
tw.line('Writing figleaf data to %s' % (data_file))
|
|
self.figleaf.stop()
|
|
self.figleaf.write_coverage(str(data_file))
|
|
data = self.figleaf.read_coverage(str(data_file))
|
|
d = {}
|
|
coverage = self.figleaf.combine_coverage(d, data)
|
|
# TODO exclude pylib
|
|
exclude = []
|
|
|
|
reportdir = terminalreporter.config.getvalue('figleafhtml')
|
|
reportdir = py.path.local(reportdir)
|
|
tw.line('Writing figleaf html to file://%s' % (reportdir))
|
|
self.figleaf.annotate_html.prepare_reportdir(str(reportdir))
|
|
self.figleaf.annotate_html.report_as_html(coverage,
|
|
str(reportdir), exclude, {})
|
|
|
|
|