qa: fix --loop with test cases implementing setUpClass/tearDownClass

3bfd4cd4 · Julien Muchembled · afd801c8 · 3bfd4cd4
Commit 3bfd4cd4 authored Jan 17, 2017 by Julien Muchembled
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

neo/scripts/runner.py neo/scripts/runner.py +5 -1

No files found.
--- a/neo/scripts/runner.py
+++ b/neo/scripts/runner.py
@@ -149,7 +149,11 @@ class NeoTestRunner(unittest.TextTestResult):
                traceback.print_exc()
                continue
            suite.addTests(loader.loadTestsFromModule(test_module))
-        suite.run(self)
+        try:
+            suite.run(self)
+        finally:
+            # Workaround weird behaviour of Python.
+            self._previousTestClass = None

    def startTest(self, test):
        super(NeoTestRunner, self).startTest(test)