Issue python#116: Test suite: use test.regrtest as test driver in runAll.py

Anselm Kruis · Anselm Kruis · commit 1f0db0934c7b · 2017-01-05T22:43:01.000+01:00
Change Stackless/unittests/runAll.py to use the test driver of the C-Python test suite. Now we can use all CPython test suite options with Stackless too. https://bitbucket.org/stackless-dev/stackless/issues/116 (grafted from 28757ff577e0a8b796ef33c9ffb31f6372ba222f)
diff --git a/Stackless/unittests/runAll.py b/Stackless/unittests/runAll.py
@@ -1,19 +1,32 @@
 """
 runAll.py
 
-Runs all testcases in files named test_*.py.
+Runs all test-cases in files named test_*.py.
 Should be run in the folder Stackless/unittests.
+
+This driver reuses the CPython test.regrtest driver and
+supports its options.
 """
+from __future__ import absolute_import
 
 import os
-import unittest
+# just to make sure, it can be imported
+import support  # @UnusedImport
+
+from test import regrtest
+from test import __path__ as test_path
+
+# path of the Stackless unittest directory
+path = os.path.split(__file__)[0]
+# add the stackless unittest dir to the package "test"
+test_path.insert(0, path)
+
+# monkey patch: remove all standard tests
+del regrtest.STDTESTS[:]
 
 
 def main():
-    path = os.path.split(__file__)[0]
-    print(path)
-    testSuite = unittest.TestLoader().discover(path)
-    unittest.TextTestRunner(verbosity=2).run(testSuite)
+    return regrtest.main(testdir=path)
 
 if __name__ == '__main__':
     main()
diff --git a/Stackless/unittests/support.py b/Stackless/unittests/support.py
@@ -32,6 +32,7 @@
 import io
 import contextlib
 import gc
+from test.support import run_unittest
 
 # emit warnings about uncollectable objects
 gc.set_debug(gc.DEBUG_UNCOLLECTABLE)
@@ -444,3 +445,23 @@ def helper():
         result = c.receive()
         assert self._ran_AsTaskletTestCase_setUp
         return result
+
+
+def test_main():
+    """Main function for the CPython :mod:`test.regrtest` test driver.
+
+    Import this function into your test_ module. It uses introspection
+    to find the module name and run your tests.
+    """
+    stack = inspect.stack(0)
+    for i in range(1, 5):
+        try:
+            the_module = stack[i][0].f_locals["the_module"]
+            break
+        except KeyError:
+            pass
+    else:
+        raise RuntimeError("can't find local variable 'the_module'")
+
+    test_suite = unittest.TestLoader().loadTestsFromModule(the_module)
+    return run_unittest(test_suite)
diff --git a/Stackless/unittests/test_chan_cb.py b/Stackless/unittests/test_chan_cb.py
@@ -1,6 +1,8 @@
+from __future__ import absolute_import
 import unittest
 
 from support import StacklessTestCase
+from support import test_main  # @UnusedImport
 
 
 class ChannelMonitor:
diff --git a/Stackless/unittests/test_channel.py b/Stackless/unittests/test_channel.py
@@ -1,3 +1,4 @@
+from __future__ import absolute_import
 import unittest
 import stackless
 try:
@@ -8,6 +9,7 @@
 import sys
 import traceback
 import contextlib
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase, require_one_thread
 
 
diff --git a/Stackless/unittests/test_defects.py b/Stackless/unittests/test_defects.py
@@ -13,6 +13,7 @@
     withThreads = False
 
 from stackless import _test_nostacklesscall as apply_not_stackless
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase, captured_stderr, require_one_thread
 
 
diff --git a/Stackless/unittests/test_exception.py b/Stackless/unittests/test_exception.py
@@ -1,5 +1,7 @@
+from __future__ import absolute_import
 import unittest
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 
diff --git a/Stackless/unittests/test_generator.py b/Stackless/unittests/test_generator.py
@@ -1,8 +1,10 @@
+from __future__ import absolute_import
 import unittest
 import gc
 import stackless
 import types
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 
diff --git a/Stackless/unittests/test_miscell.py b/Stackless/unittests/test_miscell.py
@@ -1,3 +1,4 @@
+from __future__ import absolute_import
 # import common
 
 import unittest
@@ -18,6 +19,7 @@
 except:
     withThreads = False
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase, AsTaskletTestCase, require_one_thread
 
 
diff --git a/Stackless/unittests/test_outside.py b/Stackless/unittests/test_outside.py
@@ -1,7 +1,9 @@
+from __future__ import absolute_import
 import unittest
 from stackless import test_cframe, test_cframe_nr, test_outside, test_cstate
 from stackless import tasklet, channel, run
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 
diff --git a/Stackless/unittests/test_pickle.py b/Stackless/unittests/test_pickle.py
@@ -6,6 +6,7 @@
 
 from stackless import schedule, tasklet, stackless
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase, StacklessPickleTestCase
 
 
diff --git a/Stackless/unittests/test_sched_cb.py b/Stackless/unittests/test_sched_cb.py
@@ -1,9 +1,11 @@
+from __future__ import absolute_import
 import unittest
 import stackless
 import gc
 import traceback
 import sys
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 
diff --git a/Stackless/unittests/test_shutdown.py b/Stackless/unittests/test_shutdown.py
@@ -48,6 +48,7 @@
     withThreads = True
 except:
     withThreads = False
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 
diff --git a/Stackless/unittests/test_thread.py b/Stackless/unittests/test_thread.py
@@ -1,3 +1,4 @@
+from __future__ import absolute_import
 # import common
 
 import unittest
@@ -6,6 +7,7 @@
 import time
 from stackless import _test_nostacklesscall as apply_not_stackless
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase, AsTaskletTestCase
 try:
     import threading
diff --git a/Stackless/unittests/test_tpflags_have_stackless_call.py b/Stackless/unittests/test_tpflags_have_stackless_call.py
@@ -21,10 +21,12 @@
 This is a test for issue http://www.stackless.com/ticket/18
 """
 
+from __future__ import absolute_import
 import unittest
 import stackless
 import gc
 import sys
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 
diff --git a/Stackless/unittests/test_tracing.py b/Stackless/unittests/test_tracing.py
@@ -1,7 +1,9 @@
+from __future__ import absolute_import
 import unittest
 import stackless
 import sys
 import inspect
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 
diff --git a/Stackless/unittests/test_tstate.py b/Stackless/unittests/test_tstate.py
@@ -1,9 +1,11 @@
+from __future__ import absolute_import
 import unittest
 import sys
 from pickle import Pickler, Unpickler
 from io import BytesIO
 from stackless import *
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 # import os
diff --git a/Stackless/unittests/test_watchdog.py b/Stackless/unittests/test_watchdog.py
@@ -1,8 +1,10 @@
+from __future__ import absolute_import
 import sys
 import random
 import unittest
 import stackless
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase, require_one_thread
 
 
diff --git a/Stackless/unittests/test_weakref.py b/Stackless/unittests/test_weakref.py
@@ -1,8 +1,10 @@
+from __future__ import absolute_import
 import unittest
 import weakref
 
 from stackless import *
 
+from support import test_main  # @UnusedImport
 from support import StacklessTestCase
 
 

Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,7 @@`
`6`	`6`
`7`	`7`	`from stackless import schedule, tasklet, stackless`
`8`	`8`
	`9`	`+from support import test_main # @UnusedImport`
`9`	`10`	`from support import StacklessTestCase, StacklessPickleTestCase`
`10`	`11`
`11`	`12`