From 06901abc3cd49a78843c26fc800a2ed7dc2de79a Mon Sep 17 00:00:00 2001
From: Serhiy Storchaka <storchaka@gmail.com>
Date: Tue, 5 Sep 2023 08:36:43 +0300
Subject: [PATCH] [3.12] gh-89392: Remove support of test_main() in libregrtest
 (GH-108876). (cherry picked from commit
 04a0830b00879efe057e3dfe75e9aa9c0caf1a26)

Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
---
 Lib/test/libregrtest/runtest.py                          | 9 ++++-----
 Lib/test/test_regrtest.py                                | 8 ++++----
 .../Tests/2023-09-04-15-18-14.gh-issue-89392.8A4T5p.rst  | 2 ++
 3 files changed, 10 insertions(+), 9 deletions(-)
 create mode 100644 Misc/NEWS.d/next/Tests/2023-09-04-15-18-14.gh-issue-89392.8A4T5p.rst

diff --git a/Lib/test/libregrtest/runtest.py b/Lib/test/libregrtest/runtest.py
index 6fa60697371b72..2c30269fc9fd6d 100644
--- a/Lib/test/libregrtest/runtest.py
+++ b/Lib/test/libregrtest/runtest.py
@@ -345,11 +345,10 @@ def _load_run_test(result: TestResult, ns: Namespace) -> None:
 
     the_module = importlib.import_module(abstest)
 
-    # If the test has a test_main, that will run the appropriate
-    # tests.  If not, use normal unittest test loading.
-    test_func = getattr(the_module, "test_main", None)
-    if test_func is None:
-        test_func = functools.partial(_test_module, the_module)
+    if hasattr(the_module, "test_main"):
+        # https://github.com/python/cpython/issues/89392
+        raise Exception(f"Module {result.test_name} defines test_main() which is no longer supported by regrtest")
+    test_func = functools.partial(_test_module, the_module)
 
     try:
         with save_env(ns, result.test_name):
diff --git a/Lib/test/test_regrtest.py b/Lib/test/test_regrtest.py
index 14f5a962356aa3..68f71360d6914c 100644
--- a/Lib/test/test_regrtest.py
+++ b/Lib/test/test_regrtest.py
@@ -1678,9 +1678,9 @@ def my_function():
                 7948648
                 """
 
-            def test_main():
-                testmod = sys.modules[__name__]
-                return support.run_doctest(testmod)
+            def load_tests(loader, tests, pattern):
+                tests.addTest(doctest.DocTestSuite())
+                return tests
         ''')
         testname = self.create_test(code=code)
 
@@ -1689,7 +1689,7 @@ def test_main():
         self.check_executed_tests(output, [testname],
                                   failed=[testname],
                                   randomize=True,
-                                  stats=TestStats(3, 2, 0))
+                                  stats=TestStats(1, 1, 0))
 
 
 class TestUtils(unittest.TestCase):
diff --git a/Misc/NEWS.d/next/Tests/2023-09-04-15-18-14.gh-issue-89392.8A4T5p.rst b/Misc/NEWS.d/next/Tests/2023-09-04-15-18-14.gh-issue-89392.8A4T5p.rst
new file mode 100644
index 00000000000000..e1dea8e78cdd4e
--- /dev/null
+++ b/Misc/NEWS.d/next/Tests/2023-09-04-15-18-14.gh-issue-89392.8A4T5p.rst
@@ -0,0 +1,2 @@
+Removed support of ``test_main()`` function in tests. They now always use
+normal unittest test runner.