| 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 from pylib.utils import reraiser_thread | 9 import reraiser_thread |
| 10 from pylib.utils import timeout_retry | 10 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): | 17 def _NeverEnding(tries=[0]): |
| 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 |