# Copyright 2013 The Chromium Authors
# 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:
     sudo apt-get install pip
     sudo pip install --upgrade coverage
  %s""")
        % (version, msg)
    )
    sys.exit(1)


def covered_main(
    includes,
    require_native=None,
    required_percentage=100.0,
    disable_coverage=True,
):
    """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.
        disable_coverage (bool) - If True, just run unittest.main() without any
            coverage tracking. Bug: crbug.com/662277
    """
    if disable_coverage:
        unittest.main()
        return

    try:
        import coverage

        if require_native is not None:
            got_ver = coverage.__version__
            if not getattr(coverage.collector, "CTracer", None):
                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()
    if COVERAGE.report() < required_percentage:
        print("FATAL: not at required %f%% coverage." % required_percentage)
        retcode = 2

    return retcode