2
from twisted.trial import unittest
3
from twisted.trial import reporter, runner
6
class TrialTest(unittest.TestCase):
8
self.output = StringIO.StringIO()
9
self.reporter = reporter.Reporter(stream=self.output)
10
self.loader = runner.TestLoader()
11
self.root = runner.TrialRoot(self.reporter)
14
class TestInterruptInTest(TrialTest):
15
class InterruptedTest(unittest.TestCase):
16
def test_02_raiseInterrupt(self):
17
raise KeyboardInterrupt
19
def test_01_doNothing(self):
22
def test_03_doNothing(self):
23
TestInterruptInTest.test_03_doNothing_run = True
26
super(TestInterruptInTest, self).setUp()
27
self.suite = self.loader.loadClass(TestInterruptInTest.InterruptedTest)
28
TestInterruptInTest.test_03_doNothing_run = None
30
def test_setUpOK(self):
31
self.failUnlessEqual(3, self.suite.countTestCases())
32
self.failUnlessEqual(0, self.reporter.testsRun)
33
self.failIf(self.reporter.shouldStop)
35
def test_interruptInTest(self):
36
self.root.run(self.suite)
37
self.failUnless(self.reporter.shouldStop)
38
self.failUnlessEqual(2, self.reporter.testsRun)
39
self.failIf(TestInterruptInTest.test_03_doNothing_run,
40
"test_03_doNothing ran.")
44
class TestInterruptInSetUp(TrialTest):
47
class InterruptedTest(unittest.TestCase):
49
if TestInterruptInSetUp.testsRun > 0:
50
raise KeyboardInterrupt
53
TestInterruptInSetUp.testsRun += 1
56
TestInterruptInSetUp.testsRun += 1
57
TestInterruptInSetUp.test_02_run = True
60
super(TestInterruptInSetUp, self).setUp()
61
self.suite = self.loader.loadClass(
62
TestInterruptInSetUp.InterruptedTest)
63
TestInterruptInSetUp.test_02_run = False
64
TestInterruptInSetUp.testsRun = 0
66
def test_setUpOK(self):
67
self.failUnlessEqual(0, TestInterruptInSetUp.testsRun)
68
self.failUnlessEqual(2, self.suite.countTestCases())
69
self.failUnlessEqual(0, self.reporter.testsRun)
70
self.failIf(self.reporter.shouldStop)
72
def test_interruptInSetUp(self):
73
self.root.run(self.suite)
74
self.failUnless(self.reporter.shouldStop)
75
self.failUnlessEqual(2, self.reporter.testsRun)
76
self.failIf(TestInterruptInSetUp.test_02_run,
80
class TestInterruptInTearDown(TrialTest):
83
class InterruptedTest(unittest.TestCase):
85
if TestInterruptInTearDown.testsRun > 0:
86
raise KeyboardInterrupt
89
TestInterruptInTearDown.testsRun += 1
92
TestInterruptInSetUp.testsRun += 1
93
TestInterruptInTearDown.test_02_run = True
96
super(TestInterruptInTearDown, self).setUp()
97
self.suite = self.loader.loadClass(
98
TestInterruptInTearDown.InterruptedTest)
99
TestInterruptInTearDown.testsRun = 0
100
TestInterruptInTearDown.test_02_run = False
102
def test_setUpOK(self):
103
self.failUnlessEqual(0, TestInterruptInTearDown.testsRun)
104
self.failUnlessEqual(2, self.suite.countTestCases())
105
self.failUnlessEqual(0, self.reporter.testsRun)
106
self.failIf(self.reporter.shouldStop)
108
def test_interruptInSetUp(self):
109
self.root.run(self.suite)
110
self.failUnless(self.reporter.shouldStop)
111
self.failUnlessEqual(1, self.reporter.testsRun)
112
self.failIf(TestInterruptInTearDown.test_02_run,