coverage_utils.py 2.48 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
# Copyright 2013 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 distutils.version
import os
import sys
import textwrap
import unittest

ROOT_PATH = os.path.abspath(os.path.join(
    os.path.dirname(os.path.dirname(__file__))))


def native_error(msg, version):
  print textwrap.dedent("""\
  ERROR: Native python-coverage (version: %s) is required to be
  installed on your PYTHONPATH to run this test. Recommendation:
szager@chromium.org's avatar
szager@chromium.org committed
19 20
     sudo apt-get install pip
     sudo pip install --upgrade coverage
21 22 23
  %s""") % (version, msg)
  sys.exit(1)

24 25
def covered_main(includes, require_native=None, required_percentage=100.0,
                 disable_coverage=True):
26 27 28 29 30 31 32 33 34
  """Equivalent of unittest.main(), except that it gathers coverage data, and
  asserts if the test is not at 100% coverage.

  Args:
    includes (list(str) or str) - List of paths to include in coverage report.
      May also be a single path instead of a list.
    require_native (str) - If non-None, will require that
      at least |require_native| version of coverage is installed on the
      system with CTracer.
35 36
    disable_coverage (bool) - If True, just run unittest.main() without any
      coverage tracking. Bug: crbug.com/662277
37
  """
38 39 40 41
  if disable_coverage:
    unittest.main()
    return

42 43 44 45
  try:
    import coverage
    if require_native is not None:
      got_ver = coverage.__version__
szager@chromium.org's avatar
szager@chromium.org committed
46
      if not getattr(coverage.collector, 'CTracer', None):
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
        native_error((
            "Native python-coverage module required.\n"
            "Pure-python implementation (version: %s) found: %s"
          ) % (got_ver, coverage), require_native)
      if got_ver < distutils.version.LooseVersion(require_native):
        native_error("Wrong version (%s) found: %s" % (got_ver, coverage),
                     require_native)
  except ImportError:
    if require_native is None:
      sys.path.insert(0, os.path.join(ROOT_PATH, 'third_party'))
      import coverage
    else:
      print ("ERROR: python-coverage (%s) is required to be installed on your "
             "PYTHONPATH to run this test." % require_native)
      sys.exit(1)

  COVERAGE = coverage.coverage(include=includes)
  COVERAGE.start()

  retcode = 0
  try:
    unittest.main()
  except SystemExit as e:
    retcode = e.code or retcode

  COVERAGE.stop()
73
  if COVERAGE.report() < required_percentage:
74
    print 'FATAL: not at required %f%% coverage.' % required_percentage
75 76 77
    retcode = 2

  return retcode