1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
import mock
import unittest
import nosetimer
from nose_parameterized import parameterized
class TestTimerPlugin(unittest.TestCase):
def setUp(self):
super(TestTimerPlugin, self).setUp()
self.plugin = nosetimer.TimerPlugin()
self.test_mock = mock.MagicMock(name='test')
self.test_mock.id.return_value = 1
self.opts_mock = mock.MagicMock(name='opts')
def test_addOptions(self):
parser = mock.MagicMock()
self.plugin.addOptions(parser)
self.assertEquals(parser.add_option.call_count, 5)
def test_configure(self):
attributes = ('config', 'timer_top_n', 'timer_ok',
'timer_warning', '_timed_tests')
for attr in attributes:
self.assertFalse(hasattr(self.plugin, attr))
self.plugin.configure(self.opts_mock, None)
for attr in attributes:
self.assertTrue(hasattr(self.plugin, attr))
def test_timeTaken(self):
self.assertFalse(hasattr(self.plugin, '_timer'))
self.assertEquals(self.plugin._timeTaken(), 0.0)
self.plugin.startTest(self.test_mock)
self.assertTrue(hasattr(self.plugin, '_timer'))
self.assertNotEquals(self.plugin._timeTaken(), 0.0)
@parameterized.expand([
('1', 1000), # seconds by default
('2s', 2000), # seconds
('500ms', 500), # milliseconds
])
def test_parse_time(self, value, expected_ms):
self.assertEqual(self.plugin._parse_time(value), expected_ms)
def test_parse_time_error(self):
self.assertRaises(ValueError, self.plugin._parse_time, '5seconds')
@parameterized.expand([
'timer_ok',
'timer_warning',
])
def test_parse_time_called(self, option):
time = '100ms'
with mock.patch.object(self.plugin, '_parse_time') as parse:
parse.return_value = time
mock_opts = mock.MagicMock(**{option: time})
self.plugin.configure(mock_opts, None)
self.assertEqual(getattr(self.plugin, option), time)
parse.assert_any_call(time)
def test_afterTest(self):
self.opts_mock.timer_verbose.return_value = True
self.plugin.configure(self.opts_mock, None)
# make sure there is no exception if test time result was not found
self.plugin.afterTest(self.test_mock)
|
