clean runtests

Author: Joachim Jablon

runtests.py

@@ -1,22 +1,11 @@
 #!/usr/bin/env python
 import sys
 import os
+import contextlib
 
-try:
-    from django import setup
-except ImportError:
-    import traceback
-    traceback.print_exc()
-    msg = "To fix this error, run: pip install -r requirements_test.txt"
-    raise ImportError(msg)
 
-from django.test.utils import get_runner
-from django.conf import settings
-os.environ["DJANGO_SETTINGS_MODULE"] = "tests.readonly_project.settings"
-setup()
-
-
-def run_tests(*test_args):
+@contextlib.contextmanager
+def cover():
     do_coverage = "COVERAGE" in os.environ
 
     if do_coverage:
@@ -25,12 +14,32 @@ def run_tests(*test_args):
         cov.start()
         print("Coverage will be generated")
 
-    from django.core.management import execute_from_command_line
-    execute_from_command_line(["", "test", ] + sys.argv[1:])
+    try:
+        yield
+    finally:
+        if do_coverage:
+            cov.stop()
+            cov.save()
 
-    if do_coverage:
-        cov.stop()
-        cov.save()
+
+def run_tests(*test_args):
+
+    with cover():
+        try:
+            from django import setup
+        except ImportError:
+            import traceback
+            traceback.print_exc()
+            msg = ("To fix this error, run: "
+                   "pip install -r requirements_test.txt")
+            raise ImportError(msg)
+
+        module = "tests.readonly_project.settings"
+        os.environ["DJANGO_SETTINGS_MODULE"] = module
+        setup()
+
+        from django.core.management import execute_from_command_line
+        execute_from_command_line(["", "test", ] + sys.argv[1:])
 
 
 if __name__ == '__main__':