| Index: build/android/pylib/utils/timeout_retry_unittest.py
|
| diff --git a/build/android/pylib/utils/timeout_retry_unittest.py b/build/android/pylib/utils/timeout_retry_unittest.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..dc36c42ba5cea70edd93c84c078573af3291b6c5
|
| --- /dev/null
|
| +++ b/build/android/pylib/utils/timeout_retry_unittest.py
|
| @@ -0,0 +1,52 @@
|
| +# Copyright 2013 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +"""Unittests for timeout_and_retry.py."""
|
| +
|
| +import unittest
|
| +
|
| +from pylib.utils import reraiser_thread
|
| +from pylib.utils import timeout_retry
|
| +
|
| +
|
| +class TestException(Exception):
|
| + pass
|
| +
|
| +
|
| +def _NeverEnding(tries):
|
| + tries[0] += 1
|
| + while True:
|
| + pass
|
| +
|
| +
|
| +def _CountTries(tries):
|
| + tries[0] += 1
|
| + raise TestException
|
| +
|
| +
|
| +class TestRun(unittest.TestCase):
|
| + """Tests for timeout_retry.Run."""
|
| +
|
| + def testRun(self):
|
| + self.assertTrue(timeout_retry.Run(
|
| + lambda x: x, 30, 3, [True], {}))
|
| +
|
| + def testTimeout(self):
|
| + tries = [0]
|
| + self.assertRaises(reraiser_thread.TimeoutError,
|
| + timeout_retry.Run, lambda: _NeverEnding(tries), 0, 3)
|
| + self.assertEqual(tries[0], 4)
|
| +
|
| + def testRetries(self):
|
| + tries = [0]
|
| + self.assertRaises(TestException,
|
| + timeout_retry.Run, lambda: _CountTries(tries), 30, 3)
|
| + self.assertEqual(tries[0], 4)
|
| +
|
| + def testReturnValue(self):
|
| + self.assertTrue(timeout_retry.Run(lambda: True, 30, 3))
|
| +
|
| +
|
| +if __name__ == '__main__':
|
| + unittest.main()
|
|
|