Time setup separately from test

realpython#19 (comment)
drewbrew · Oct 19, 2020 · 1d4c37b · 1d4c37b
1 parent 2c5673c
commit 1d4c37b
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/django_slowtests/testrunner.py b/django_slowtests/testrunner.py
@@ -1,4 +1,5 @@
 import glob
+import importlib
 import json
 import time
 import os
@@ -105,9 +106,8 @@ def run(self, result, debug=False):
             if result.shouldStop:
                 break
 
-            start_time = _time()
-
             if _isnotsuite(test):
+                start_time = _time()
                 self._tearDownPreviousClass(test, result)
                 self._handleModuleFixture(test, result)
                 self._handleClassSetUp(test, result)
@@ -116,6 +116,13 @@ def run(self, result, debug=False):
                 if (getattr(test.__class__, '_classSetupFailed', False) or
                         getattr(result, '_moduleSetUpFailed', False)):
                     continue
+                self.save_test_time(
+                    f'{test.__class__.__module__}.{test.__class__.__name__}.setUp '
+                    f'({importlib.import_module(test.__class__.__module__).__file__})',
+                    _time() - start_time
+                )
+
+            start_time = _time()
 
             if not debug:
                 test(result)