Index: scripts/slave/unittests/expect_tests/cover.py |
diff --git a/scripts/slave/unittests/expect_tests/cover.py b/scripts/slave/unittests/expect_tests/cover.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..fa321557e6f143d0f1f5fc4752aeb3993cc27e74 |
--- /dev/null |
+++ b/scripts/slave/unittests/expect_tests/cover.py |
@@ -0,0 +1,88 @@ |
+# Copyright 2014 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+import contextlib |
+ |
+from cStringIO import StringIO |
+ |
+try: |
+ import coverage |
+ coverage_version = coverage.__version__ |
+except ImportError: |
+ coverage = None |
+ coverage_version = None |
+ |
+from distutils.version import StrictVersion |
+ |
+if (not coverage or |
+ #not hasattr(coverage.collector, 'CTracer') or |
agable
2014/04/03 00:38:00
Remove commented out lines
|
+ #not coverage.collector.CTracer or |
+ StrictVersion(coverage_version) < StrictVersion('3.7.1')): |
+ raise ImportError( |
+ 'This test requires natively-installed python-coverage (>=3.7.1). ' |
+ 'Current is %s.' % coverage_version) |
+ |
+ |
+@contextlib.contextmanager |
+def _cover(**kwargs): |
+ c = coverage.coverage(**kwargs) |
+ c._warn_no_data = False # pylint: disable=protected-access |
+ c.start() |
+ try: |
+ yield |
+ finally: |
+ c.stop() |
+ c.save() |
+ |
+ |
+@contextlib.contextmanager |
+def _nop_cover(): |
+ yield |
+ |
+ |
+class CoverageContext(object): |
+ def __init__(self, includes, omits, enabled=True): |
+ self.opts = None |
+ self.cov = None |
+ self.enabled = enabled |
+ |
+ if enabled: |
+ self.opts = { |
+ 'include': includes, |
+ 'omit': omits, |
+ 'data_suffix': True |
+ } |
+ self.cov = coverage.coverage(**self.opts) |
+ self.cov.erase() |
+ |
+ def cleanup(self): |
+ if self.enabled: |
+ self.cov.combine() |
+ |
+ def report(self, verbose): |
+ fail = False |
+ |
+ if self.enabled: |
+ outf = StringIO() |
+ fail = self.cov.report(file=outf) != 100.0 |
+ summary = outf.getvalue().replace('%- 15s' % 'Name', 'Coverage Report', 1) |
+ if verbose: |
+ print summary |
+ elif fail: |
+ lines = summary.splitlines() |
+ lines[2:-2] = [l for l in lines[2:-2] |
+ if not l.strip().endswith('100%')] |
+ print '\n'.join(lines) |
+ print 'FATAL: Test coverage is not at 100%.' |
+ |
+ return not fail |
+ |
+ def create_subprocess_context(self): |
+ if self.enabled: |
+ return _cover(**self.opts) |
+ else: |
+ return _nop_cover() |