zulip/tools/test-backend

#!/usr/bin/env python

from __future__ import print_function
from __future__ import absolute_import
import glob
import optparse
import os
import sys
import subprocess

# check for the venv
from lib import sanity_check
sanity_check.check_venv(__file__)

import django
from django.conf import settings
from django.test.utils import get_runner

target_fully_covered = {path for target in [
    'analytics/tests/*.py',
    'analytics/lib/*.py',
    'zerver/context_processors.py',
    'zerver/lib/alert_words.py',
    'zerver/lib/attachments.py',
    'zerver/lib/avatar_hash.py',
    'zerver/lib/context_managers.py',
    'zerver/lib/domains.py',
    'zerver/lib/emoji.py',
    'zerver/lib/i18n.py',
    'zerver/lib/mention.py',
    'zerver/lib/message.py',
    'zerver/lib/name_restrictions.py',
    'zerver/lib/retention.py',
    'zerver/lib/streams.py',
    'zerver/lib/users.py',
    'zerver/lib/webhooks/*.py',
    'zerver/views/*.py',
    # Once we have a nice negative tests system, we can add these:
    # 'zerver/webhooks/*/*.py',
    # 'zerver/webhooks/*/*/*.py',
    'zproject/backends.py',
    # Uncovered but in exclude list and we'd like to have included soon
    'confirmation/models.py',
    'zerver/decorator.py',
    'zerver/lib/actions.py',
    'zerver/lib/events.py',
    'zerver/lib/bugdown/__init__.py',
    'zerver/lib/events.py',
    'zerver/lib/integrations.py',
    'zerver/lib/message.py',
    'zerver/lib/narrow.py',
    'zerver/lib/notifications.py',
    'zerver/lib/push_notifications.py',
    'zerver/lib/request.py',
    'zerver/lib/upload.py',
    'zerver/lib/validator.py',
    'zerver/models.py',
] for path in glob.glob(target)}

not_yet_fully_covered = {
    # Goal is for analytics to have 100% coverage
    'analytics/lib/counts.py',
    'analytics/lib/fixtures.py',
    'analytics/lib/time_utils.py',
    # Major lib files should have 100% coverage
    'confirmation/models.py',
    'zerver/decorator.py',
    'zerver/lib/actions.py',
    'zerver/lib/events.py',
    'zerver/lib/bugdown/__init__.py',
    'zerver/lib/events.py',
    'zerver/lib/i18n.py',
    'zerver/lib/integrations.py',
    'zerver/lib/message.py',
    'zerver/lib/narrow.py',
    'zerver/lib/notifications.py',
    'zerver/lib/push_notifications.py',
    'zerver/lib/request.py',
    'zerver/lib/upload.py',
    'zerver/lib/validator.py',
    'zerver/models.py',
    # Test files should have full coverage; it's a bug in the test if
    # they don't!
    'zerver/tests/test_auth_backends.py',
    'zerver/tests/test_bugdown.py',
    'zerver/tests/test_decorators.py',
    'zerver/tests/test_email_mirror.py',
    'zerver/tests/test_events.py',
    'zerver/tests/test_narrow.py',
    'zerver/tests/test_signup.py',
    'zerver/tests/test_subs.py',
    'zerver/tests/test_templates.py',
    'zerver/tests/test_tornado.py',
    'zerver/tests/test_urls.py',
    'zerver/tests/tests.py',
    # Getting views file coverage to 100% is a major project goal
    'zerver/views/auth.py',
    'zerver/views/integrations.py',
    'zerver/views/messages.py',
    'zerver/views/report.py',
    'zerver/views/unsubscribe.py',
    'zerver/views/zephyr.py',
    'zerver/views/invite.py',
    'zerver/views/home.py',
    'zerver/views/registration.py',
    'zerver/views/events_register.py',
    # This one has 1 line that isn't covered in Python 3
    'zerver/views/upload.py',
    # Getting this to 100% is a major project goal.
    'zproject/backends.py',
}

enforce_fully_covered = sorted(target_fully_covered - not_yet_fully_covered)

if __name__ == "__main__":
    TOOLS_DIR = os.path.dirname(os.path.abspath(__file__))
    os.chdir(os.path.dirname(TOOLS_DIR))
    sys.path.insert(0, os.path.dirname(TOOLS_DIR))
    from zerver.lib.test_fixtures import is_template_database_current

    from tools.lib.test_script import (
        get_provisioning_status,
    )

    os.environ['DJANGO_SETTINGS_MODULE'] = 'zproject.test_settings'
    # "-u" uses unbuffered IO, which is important when wrapping it in subprocess
    os.environ['PYTHONUNBUFFERED'] = 'y'

    usage = """%prog [options]
    test-backend # Runs all backend tests
    test-backend zerver.tests.test_bugdown # run all tests in a test module
    test-backend zerver/tests/test_bugdown.py # run all tests in a test module
    test-backend test_bugdown # run all tests in a test module
    test-backend zerver.tests.test_bugdown.BugdownTest # run all tests in a test class
    test-backend BugdownTest # run all tests in a test class
    test-backend zerver.tests.test_bugdown.BugdownTest.test_inline_youtube # run a single test
    test-backend BugdownTest.test_inline_youtube # run a single test"""

    parser = optparse.OptionParser(usage)

    parser.add_option('--nonfatal-errors', action="store_false", default=True,
                      dest="fatal_errors", help="Continue past test failures to run all tests")
    parser.add_option('--coverage', dest='coverage',
                      action="store_true",
                      default=False, help='Compute test coverage.')
    parser.add_option('--no-verbose-coverage', dest='verbose_coverage',
                      action="store_false",
                      default=True, help='Disable verbose print of coverage report.')
    parser.add_option('--profile', dest='profile',
                      action="store_true",
                      default=False, help='Profile test runtime.')
    parser.add_option('--force', dest='force',
                      action="store_true",
                      default=False, help='Run tests despite possible problems.')
    parser.add_option('--no-shallow', dest='no_shallow',
                      action="store_true",
                      default=False,
                      help="Don't allow shallow testing of templates (deprecated)")
    parser.add_option('--verbose', dest='verbose',
                      action="store_true",
                      default=False,
                      help="Show detailed output")
    parser.add_option('--no-generate-fixtures', action="store_false", default=True,
                      dest="generate_fixtures",
                      help=("Reduce running time by not calling generate-fixtures. "
                            "This may cause spurious failures for some tests."))
    parser.add_option('--report-slow-tests', dest='report_slow_tests',
                      action="store_true",
                      default=False,
                      help="Show which tests are slowest.")

    (options, args) = parser.parse_args()

    zerver_test_dir = 'zerver/tests/'

    # to transform forward slashes '/' present in the argument into dots '.'
    for suite in args:
        args[args.index(suite)] = suite.rstrip('/').replace("/", ".")

    def rewrite_arguments(search_key):
        # type: (str) -> None
        for root, dirs, files_names in os.walk(zerver_test_dir, topdown=False):
            for file_name in files_names:
                # Check for files starting with alphanumeric characters and ending with '.py'
                # Ignore backup files if any
                if not file_name[0].isalnum() or not file_name.endswith(".py"):
                    continue
                filepath = os.path.join(root, file_name)
                for line in open(filepath):
                    if search_key not in line:
                        continue
                    new_suite = filepath.replace(".py", ".") + suite
                    args[args.index(suite)] = new_suite
                    return

    for suite in args:
        if suite[0].isupper() and "test_" in suite:
            classname = suite.rsplit('.', 1)[0]
            rewrite_arguments(classname)
        elif suite[0].isupper():
            rewrite_arguments(suite)

    for suite in args:
        if suite.startswith('test'):
            for root, dirs, files_names in os.walk(zerver_test_dir):
                for file_name in files_names:
                    if file_name == suite or file_name == suite + ".py":
                        new_suite = os.path.join(root, file_name)
                        args[args.index(suite)] = new_suite
                        break

    for suite in args:
        args[args.index(suite)] = suite.replace(".py", "")

    # to transform forward slashes '/' introduced by the zerver_test_dir into dots '.'
    # taking care of any forward slashes that might be present
    for suite in args:
        args[args.index(suite)] = suite.replace("/", ".")

    full_suite = len(args) == 0

    if len(args) == 0:
        suites = ["zerver.tests",
                  "zerver.webhooks",
                  "analytics.tests"]
    else:
        suites = args

    if not options.force:
        ok, msg = get_provisioning_status()
        if not ok:
            print(msg)
            print('If you really know what you are doing, use --force to run anyway.')
            sys.exit(1)

    if options.coverage:
        import coverage
        cov = coverage.Coverage(omit=["*/zulip-venv-cache/*",
                                      "*/migrations/*",
                                      "*/management/commands/*"])
        cov.start()
    if options.profile:
        import cProfile
        prof = cProfile.Profile()
        prof.enable()

    # This is kind of hacky, but it's the most reliable way
    # to make sure instrumentation decorators know the
    # setting when they run.
    os.environ['TEST_INSTRUMENT_URL_COVERAGE'] = 'TRUE'

    # setup() needs to be called after coverage is started to get proper coverage reports of model
    # files, since part of setup is importing the models for all applications in INSTALLED_APPS.
    django.setup()

    if options.generate_fixtures:
        generate_fixtures_command = [os.path.join(TOOLS_DIR, 'setup', 'generate-fixtures')]
        if not is_template_database_current():
            generate_fixtures_command.append('--force')

        subprocess.call(generate_fixtures_command)

    TestRunner = get_runner(settings)
    test_runner = TestRunner(failfast=options.fatal_errors, verbosity=2)
    failures = test_runner.run_tests(suites, full_suite=full_suite)

    templates_not_rendered = test_runner.get_shallow_tested_templates()
    # We only check the templates if all the tests ran and passed
    if not failures and full_suite and templates_not_rendered:
        missed_count = len(templates_not_rendered)
        print("\nError: %s templates have no tests!" % (missed_count,))
        for template in templates_not_rendered:
            print('  {}'.format(template))
        print("See zerver/tests/test_templates.py for the exclude list.")
        failures = True

    if options.coverage:
        cov.stop()
        cov.save()
        if options.verbose_coverage:
            print("Printing coverage data")
            cov.report(show_missing=False)
        cov.html_report(directory='var/coverage')
        print("HTML report saved to var/coverage")
    if full_suite and not failures and options.coverage:
        # Assert that various files have full coverage
        for path in enforce_fully_covered:
            missing_lines = cov.analysis2(path)[3]
            if len(missing_lines) > 0:
                print("ERROR: %s no longer has complete backend test coverage" % (path,))
                print("  Lines missing coverage: %s" % (missing_lines,))
                print()
                failures = True
        if failures:
            print("It looks like your changes lost 100% test coverage in one or more files")
            print("Usually, the right fix for this is to add some tests.")
            print("But also check out the include/exclude lists in tools/test-backend.")
            print("To run this check locally, use `test-backend --coverage`.")
    if options.profile:
        prof.disable()
        prof.dump_stats("/tmp/profile.data")
        print("Profile data saved to /tmp/profile.data")
        print("You can visualize it using e.g. `runsnake /tmp/profile.data`")

    if options.report_slow_tests:
        from zerver.lib.test_runner import report_slow_tests
        # We do this even with failures, since slowness can be
        # an important clue as to why tests fail.
        report_slow_tests()

    # We'll have printed whether tests passed or failed above
    sys.exit(bool(failures))