RAHHH

backend/lib/python3.12/site-packages/twisted/trial/test/test_keyboard.py

@@ -0,0 +1,120 @@
+# Copyright (c) Twisted Matrix Laboratories.
+# See LICENSE for details.
+
+"""
+Tests for interrupting tests with Control-C.
+"""
+from __future__ import annotations
+
+from io import StringIO
+
+from twisted.trial import reporter, runner, unittest
+
+
+class TrialTest(unittest.SynchronousTestCase):
+    def setUp(self) -> None:
+        self.output = StringIO()
+        self.reporter = reporter.TestResult()
+        self.loader = runner.TestLoader()
+
+
+class InterruptInTestTests(TrialTest):
+    test_03_doNothing_run: bool | None
+
+    class InterruptedTest(unittest.TestCase):
+        def test_02_raiseInterrupt(self) -> None:
+            raise KeyboardInterrupt
+
+        def test_01_doNothing(self) -> None:
+            pass
+
+        def test_03_doNothing(self) -> None:
+            InterruptInTestTests.test_03_doNothing_run = True
+
+    def setUp(self) -> None:
+        super().setUp()
+        self.suite = self.loader.loadClass(InterruptInTestTests.InterruptedTest)
+        InterruptInTestTests.test_03_doNothing_run = None
+
+    def test_setUpOK(self) -> None:
+        self.assertEqual(3, self.suite.countTestCases())
+        self.assertEqual(0, self.reporter.testsRun)
+        self.assertFalse(self.reporter.shouldStop)
+
+    def test_interruptInTest(self) -> None:
+        runner.TrialSuite([self.suite]).run(self.reporter)
+        self.assertTrue(self.reporter.shouldStop)
+        self.assertEqual(2, self.reporter.testsRun)
+        self.assertFalse(
+            InterruptInTestTests.test_03_doNothing_run, "test_03_doNothing ran."
+        )
+
+
+class InterruptInSetUpTests(TrialTest):
+    testsRun = 0
+    test_02_run: bool
+
+    class InterruptedTest(unittest.TestCase):
+        def setUp(self) -> None:
+            if InterruptInSetUpTests.testsRun > 0:
+                raise KeyboardInterrupt
+
+        def test_01(self) -> None:
+            InterruptInSetUpTests.testsRun += 1
+
+        def test_02(self) -> None:
+            InterruptInSetUpTests.testsRun += 1
+            InterruptInSetUpTests.test_02_run = True
+
+    def setUp(self) -> None:
+        super().setUp()
+        self.suite = self.loader.loadClass(InterruptInSetUpTests.InterruptedTest)
+        InterruptInSetUpTests.test_02_run = False
+        InterruptInSetUpTests.testsRun = 0
+
+    def test_setUpOK(self) -> None:
+        self.assertEqual(0, InterruptInSetUpTests.testsRun)
+        self.assertEqual(2, self.suite.countTestCases())
+        self.assertEqual(0, self.reporter.testsRun)
+        self.assertFalse(self.reporter.shouldStop)
+
+    def test_interruptInSetUp(self) -> None:
+        runner.TrialSuite([self.suite]).run(self.reporter)
+        self.assertTrue(self.reporter.shouldStop)
+        self.assertEqual(2, self.reporter.testsRun)
+        self.assertFalse(InterruptInSetUpTests.test_02_run, "test_02 ran")
+
+
+class InterruptInTearDownTests(TrialTest):
+    testsRun = 0
+    test_02_run: bool
+
+    class InterruptedTest(unittest.TestCase):
+        def tearDown(self) -> None:
+            if InterruptInTearDownTests.testsRun > 0:
+                raise KeyboardInterrupt
+
+        def test_01(self) -> None:
+            InterruptInTearDownTests.testsRun += 1
+
+        def test_02(self) -> None:
+            InterruptInTearDownTests.testsRun += 1
+            InterruptInTearDownTests.test_02_run = True
+
+    def setUp(self) -> None:
+        super().setUp()
+        self.suite = self.loader.loadClass(InterruptInTearDownTests.InterruptedTest)
+        InterruptInTearDownTests.testsRun = 0
+        InterruptInTearDownTests.test_02_run = False
+
+    def test_setUpOK(self) -> None:
+        self.assertEqual(0, InterruptInTearDownTests.testsRun)
+        self.assertEqual(2, self.suite.countTestCases())
+        self.assertEqual(0, self.reporter.testsRun)
+        self.assertFalse(self.reporter.shouldStop)
+
+    def test_interruptInTearDown(self) -> None:
+        runner.TrialSuite([self.suite]).run(self.reporter)
+        self.assertEqual(1, self.reporter.testsRun)
+        self.assertTrue(self.reporter.shouldStop)
+        self.assertFalse(InterruptInTearDownTests.test_02_run, "test_02 ran")