OLD | NEW |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 """Unittests for timeout_and_retry.py.""" | 5 """Unittests for timeout_and_retry.py.""" |
6 | 6 |
7 import unittest | 7 import unittest |
8 | 8 |
9 import reraiser_thread | 9 from pylib.utils import reraiser_thread |
10 import timeout_retry | 10 from pylib.utils import timeout_retry |
11 | 11 |
12 | 12 |
13 class TestException(Exception): | 13 class TestException(Exception): |
14 pass | 14 pass |
15 | 15 |
16 | 16 |
17 def _NeverEnding(tries=[0]): | 17 def _NeverEnding(tries): |
18 tries[0] += 1 | 18 tries[0] += 1 |
19 while True: | 19 while True: |
20 pass | 20 pass |
21 | 21 |
22 | 22 |
23 def _CountTries(tries): | 23 def _CountTries(tries): |
24 tries[0] += 1 | 24 tries[0] += 1 |
25 raise TestException | 25 raise TestException |
26 | 26 |
27 | 27 |
(...skipping 15 matching lines...) Expand all Loading... |
43 self.assertRaises(TestException, | 43 self.assertRaises(TestException, |
44 timeout_retry.Run, lambda: _CountTries(tries), 30, 3) | 44 timeout_retry.Run, lambda: _CountTries(tries), 30, 3) |
45 self.assertEqual(tries[0], 4) | 45 self.assertEqual(tries[0], 4) |
46 | 46 |
47 def testReturnValue(self): | 47 def testReturnValue(self): |
48 self.assertTrue(timeout_retry.Run(lambda: True, 30, 3)) | 48 self.assertTrue(timeout_retry.Run(lambda: True, 30, 3)) |
49 | 49 |
50 | 50 |
51 if __name__ == '__main__': | 51 if __name__ == '__main__': |
52 unittest.main() | 52 unittest.main() |
OLD | NEW |