use custom test runner

the other approach eliminated the ability to pass args like individual
tests

Author: timgraham

performance_tests/perftest/runner.py

@@ -0,0 +1,9 @@
+from django.test.runner import DiscoverRunner
+
+from .base import write_output_file
+
+
+class TestRunner(DiscoverRunner):
+    def teardown_test_environment(self, **kwargs):
+        write_output_file()
+        super().teardown_test_environment(**kwargs)

performance_tests/perftest/settings.py

@@ -11,3 +11,5 @@
 INSTALLED_APPS = ["perftest"]
 
 DEFAULT_AUTO_FIELD = "django_mongodb_backend.fields.ObjectIdAutoField"
+
+TEST_RUNNER = "perftest.runner.TestRunner"

performance_tests/runtests.py

@@ -3,17 +3,11 @@
 import sys
 
 import django
-from django.conf import settings
-from django.test.utils import get_runner
-from perftest.base import write_output_file
+from django.core.management import call_command
 
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "perftest.settings")
 
 
 if __name__ == "__main__":
     django.setup()
-    TestRunner = get_runner(settings)
-    test_runner = TestRunner()
-    failures = test_runner.run_tests(["perftest"])
-    write_output_file()
-    sys.exit(bool(failures))
+    call_command("test", sys.argv[1:])