| 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..2eab0580440f6a997c47e3d713d67cb105b6b340
|
| --- /dev/null
|
| +++ b/scripts/slave/unittests/expect_tests/cover.py
|
| @@ -0,0 +1,86 @@
|
| +# 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
|
| + 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(enabled, maybe_kwargs):
|
| + if enabled:
|
| + c = coverage.coverage(**maybe_kwargs)
|
| + c._warn_no_data = False # pylint: disable=protected-access
|
| + c.start()
|
| + try:
|
| + yield
|
| + finally:
|
| + c.stop()
|
| + c.save()
|
| + else:
|
| + 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
|
| + 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)
|
|
|