OLD | NEW |
| (Empty) |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | |
2 # Use of this source code is governed by a BSD-style license that can be | |
3 # found in the LICENSE file. | |
4 | |
5 """Unittests for timeout_and_retry.py.""" | |
6 | |
7 import unittest | |
8 | |
9 from pylib.utils import reraiser_thread | |
10 from pylib.utils import timeout_retry | |
11 | |
12 | |
13 class TestException(Exception): | |
14 pass | |
15 | |
16 | |
17 def _NeverEnding(tries): | |
18 tries[0] += 1 | |
19 while True: | |
20 pass | |
21 | |
22 | |
23 def _CountTries(tries): | |
24 tries[0] += 1 | |
25 raise TestException | |
26 | |
27 | |
28 class TestRun(unittest.TestCase): | |
29 """Tests for timeout_retry.Run.""" | |
30 | |
31 def testRun(self): | |
32 self.assertTrue(timeout_retry.Run( | |
33 lambda x: x, 30, 3, [True], {})) | |
34 | |
35 def testTimeout(self): | |
36 tries = [0] | |
37 self.assertRaises(reraiser_thread.TimeoutError, | |
38 timeout_retry.Run, lambda: _NeverEnding(tries), 0, 3) | |
39 self.assertEqual(tries[0], 4) | |
40 | |
41 def testRetries(self): | |
42 tries = [0] | |
43 self.assertRaises(TestException, | |
44 timeout_retry.Run, lambda: _CountTries(tries), 30, 3) | |
45 self.assertEqual(tries[0], 4) | |
46 | |
47 def testReturnValue(self): | |
48 self.assertTrue(timeout_retry.Run(lambda: True, 30, 3)) | |
49 | |
50 | |
51 if __name__ == '__main__': | |
52 unittest.main() | |
OLD | NEW |