From ab081c68d0059744cc568fd78f7e84f3c4f915a6 Mon Sep 17 00:00:00 2001
From: briandorsey <none@none>
Date: Mon, 30 Mar 2009 00:10:24 +0200
Subject: [PATCH] [svn r63427] added figleaf plugin to pylib

--HG--
branch : trunk
---
 py/test/plugin/pytest_figleaf.py | 48 ++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 py/test/plugin/pytest_figleaf.py

diff --git a/py/test/plugin/pytest_figleaf.py b/py/test/plugin/pytest_figleaf.py
new file mode 100644
index 000000000..8f914061c
--- /dev/null
+++ b/py/test/plugin/pytest_figleaf.py
@@ -0,0 +1,48 @@
+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, {})
+
+