name : test_keyboard.py # 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")