run_tests.py

"""
Creating standalone Django apps is a PITA because you're not in a project, so
you don't have a settings.py file.  I can never remember to define
DJANGO_SETTINGS_MODULE, so I run these commands which get the right env
automatically.
"""
import argparse
import os
import sys
from subprocess import call, check_output

NAME = os.path.basename(os.path.dirname(__file__))
ROOT = os.path.abspath(os.path.dirname(__file__))

os.environ["PYTHONPATH"] = os.pathsep.join([ROOT, os.path.join(ROOT, "examples")])

SETTINGS = (
    "locmem_settings",
    "settings",
    "memcache_byid",
    "custom_backend",
    "redis_settings",
    "redis_byid",
    "django_redis_settings",
)


def main():
    parser = argparse.ArgumentParser(
        description="Run the tests for django-cache-machine. "
        "If no options are specified, tests will be run with "
        "all settings files and without coverage.py."
    )
    parser.add_argument(
        "--with-coverage",
        action="store_true",
        help="Run tests with coverage.py and display coverage report",
    )
    parser.add_argument(
        "--settings",
        choices=SETTINGS,
        help="Run tests only for the specified settings file",
    )
    args = parser.parse_args()
    settings = args.settings and [args.settings] or SETTINGS
    results = []
    django_admin = check_output(["which", "django-admin"]).strip()
    for i, settings_module in enumerate(settings):
        print("Running tests for: %s" % settings_module)
        os.environ["DJANGO_SETTINGS_MODULE"] = "cache_machine.%s" % settings_module
        # append to the existing coverage data for all but the first run
        if args.with_coverage and i > 0:
            test_cmd = ["coverage", "run", "--append"]
        elif args.with_coverage:
            test_cmd = ["coverage", "run"]
        else:
            test_cmd = []
        test_cmd += [django_admin, "test", "--keepdb"]
        results.append(call(test_cmd))
        if args.with_coverage:
            results.append(call(["coverage", "report", "-m", "--fail-under", "70"]))
    sys.exit(any(results) and 1 or 0)


if __name__ == "__main__":
    main()