Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(206)

Unified Diff: third_party/recipe_engine/expect_tests/cover.py

Issue 1347263002: Revert of Cross-repo recipe package system. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/tools/build
Patch Set: Created 5 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: third_party/recipe_engine/expect_tests/cover.py
diff --git a/third_party/recipe_engine/expect_tests/cover.py b/third_party/recipe_engine/expect_tests/cover.py
new file mode 100644
index 0000000000000000000000000000000000000000..d5411166ac2c7fdf4def118ab6c086181f2afe05
--- /dev/null
+++ b/third_party/recipe_engine/expect_tests/cover.py
@@ -0,0 +1,85 @@
+# 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.
+
+from cStringIO import StringIO
+
+import coverage
+
+# This is instead of a contextmanager because it causes old pylints to crash :(
+class _Cover(object):
+ def __init__(self, enabled, maybe_kwargs):
+ self.enabled = enabled
+ self.kwargs = maybe_kwargs or {}
+ self.c = None
+
+ def __call__(self, **kwargs):
+ new_kwargs = self.kwargs
+ if self.enabled:
+ new_kwargs = new_kwargs.copy()
+ new_kwargs.update(kwargs)
+ return _Cover(self.enabled, new_kwargs)
+
+ def __enter__(self):
+ if self.enabled:
+ if self.c is None:
+ self.c = coverage.coverage(**self.kwargs)
+ self.c._warn_no_data = False # pylint: disable=protected-access
+ self.c.start()
+
+ def __exit__(self, *_):
+ if self.enabled:
+ self.c.stop()
+ self.c.save()
+
+
+class CoverageContext(object):
+ def __init__(self, name, cover_branches, html_report, enabled=True):
+ self.opts = None
+ self.cov = None
+ self.enabled = enabled
+
+ self.html_report = html_report
+
+ if enabled:
+ self.opts = {
+ 'data_file': '.%s_coverage' % name,
+ 'data_suffix': True,
+ 'branch': cover_branches,
+ }
+ self.cov = coverage.coverage(**self.opts)
+ self.cov.erase()
+
+ def cleanup(self):
+ if self.enabled:
+ self.cov.combine()
+
+ def report(self, verbose, omit=None):
+ fail = False
+
+ if self.enabled:
+ if self.html_report:
+ self.cov.html_report(directory=self.html_report, omit=None)
+
+ outf = StringIO()
+ fail = self.cov.report(file=outf, omit=omit) != 100.0
+ summary = outf.getvalue().replace('%- 15s' % 'Name', 'Coverage Report', 1)
+ if verbose:
+ print
+ print summary
+ elif fail:
+ print
+ lines = summary.splitlines()
+ lines[2:-2] = [l for l in lines[2:-2]
+ if not l.strip().endswith('100%')]
+ print '\n'.join(lines)
+ print
+ print 'FATAL: Test coverage is not at 100%.'
+
+ return not fail
+
+ def create_subprocess_context(self):
+ # Can't have this method be the contextmanager because otherwise
+ # self (and self.cov) will get pickled to the subprocess, and we don't want
+ # that :(
+ return _Cover(self.enabled, self.opts)
« no previous file with comments | « third_party/recipe_engine/expect_tests/__init__.py ('k') | third_party/recipe_engine/expect_tests/handle_debug.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698