OLD | NEW |
| (Empty) |
1 import StringIO | |
2 from twisted.trial import unittest | |
3 from twisted.trial import reporter, runner | |
4 | |
5 | |
6 class TrialTest(unittest.TestCase): | |
7 def setUp(self): | |
8 self.output = StringIO.StringIO() | |
9 self.reporter = reporter.TestResult() | |
10 self.loader = runner.TestLoader() | |
11 | |
12 | |
13 class TestInterruptInTest(TrialTest): | |
14 class InterruptedTest(unittest.TestCase): | |
15 def test_02_raiseInterrupt(self): | |
16 raise KeyboardInterrupt | |
17 | |
18 def test_01_doNothing(self): | |
19 pass | |
20 | |
21 def test_03_doNothing(self): | |
22 TestInterruptInTest.test_03_doNothing_run = True | |
23 | |
24 def setUp(self): | |
25 super(TestInterruptInTest, self).setUp() | |
26 self.suite = self.loader.loadClass(TestInterruptInTest.InterruptedTest) | |
27 TestInterruptInTest.test_03_doNothing_run = None | |
28 | |
29 def test_setUpOK(self): | |
30 self.failUnlessEqual(3, self.suite.countTestCases()) | |
31 self.failUnlessEqual(0, self.reporter.testsRun) | |
32 self.failIf(self.reporter.shouldStop) | |
33 | |
34 def test_interruptInTest(self): | |
35 runner.TrialSuite([self.suite]).run(self.reporter) | |
36 self.failUnless(self.reporter.shouldStop) | |
37 self.failUnlessEqual(2, self.reporter.testsRun) | |
38 self.failIf(TestInterruptInTest.test_03_doNothing_run, | |
39 "test_03_doNothing ran.") | |
40 | |
41 | |
42 class TestInterruptInSetUp(TrialTest): | |
43 testsRun = 0 | |
44 | |
45 class InterruptedTest(unittest.TestCase): | |
46 def setUp(self): | |
47 if TestInterruptInSetUp.testsRun > 0: | |
48 raise KeyboardInterrupt | |
49 | |
50 def test_01(self): | |
51 TestInterruptInSetUp.testsRun += 1 | |
52 | |
53 def test_02(self): | |
54 TestInterruptInSetUp.testsRun += 1 | |
55 TestInterruptInSetUp.test_02_run = True | |
56 | |
57 def setUp(self): | |
58 super(TestInterruptInSetUp, self).setUp() | |
59 self.suite = self.loader.loadClass( | |
60 TestInterruptInSetUp.InterruptedTest) | |
61 TestInterruptInSetUp.test_02_run = False | |
62 TestInterruptInSetUp.testsRun = 0 | |
63 | |
64 def test_setUpOK(self): | |
65 self.failUnlessEqual(0, TestInterruptInSetUp.testsRun) | |
66 self.failUnlessEqual(2, self.suite.countTestCases()) | |
67 self.failUnlessEqual(0, self.reporter.testsRun) | |
68 self.failIf(self.reporter.shouldStop) | |
69 | |
70 def test_interruptInSetUp(self): | |
71 runner.TrialSuite([self.suite]).run(self.reporter) | |
72 self.failUnless(self.reporter.shouldStop) | |
73 self.failUnlessEqual(2, self.reporter.testsRun) | |
74 self.failIf(TestInterruptInSetUp.test_02_run, | |
75 "test_02 ran") | |
76 | |
77 | |
78 class TestInterruptInTearDown(TrialTest): | |
79 testsRun = 0 | |
80 | |
81 class InterruptedTest(unittest.TestCase): | |
82 def tearDown(self): | |
83 if TestInterruptInTearDown.testsRun > 0: | |
84 raise KeyboardInterrupt | |
85 | |
86 def test_01(self): | |
87 TestInterruptInTearDown.testsRun += 1 | |
88 | |
89 def test_02(self): | |
90 TestInterruptInTearDown.testsRun += 1 | |
91 TestInterruptInTearDown.test_02_run = True | |
92 | |
93 def setUp(self): | |
94 super(TestInterruptInTearDown, self).setUp() | |
95 self.suite = self.loader.loadClass( | |
96 TestInterruptInTearDown.InterruptedTest) | |
97 TestInterruptInTearDown.testsRun = 0 | |
98 TestInterruptInTearDown.test_02_run = False | |
99 | |
100 def test_setUpOK(self): | |
101 self.failUnlessEqual(0, TestInterruptInTearDown.testsRun) | |
102 self.failUnlessEqual(2, self.suite.countTestCases()) | |
103 self.failUnlessEqual(0, self.reporter.testsRun) | |
104 self.failIf(self.reporter.shouldStop) | |
105 | |
106 def test_interruptInTearDown(self): | |
107 runner.TrialSuite([self.suite]).run(self.reporter) | |
108 self.failUnlessEqual(1, self.reporter.testsRun) | |
109 self.failUnless(self.reporter.shouldStop) | |
110 self.failIf(TestInterruptInTearDown.test_02_run, | |
111 "test_02 ran") | |
112 | |
113 | |
OLD | NEW |