"""
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()