1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
"""
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()
|
