| OLD | NEW |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 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 """ | 5 """ |
| 6 Unit tests for decorators.py. | 6 Unit tests for decorators.py. |
| 7 """ | 7 """ |
| 8 | 8 |
| 9 # pylint: disable=W0613 | 9 # pylint: disable=W0613 |
| 10 | 10 |
| (...skipping 262 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 273 self.function_call_counters['alwaysTimesOut'] += 1 | 273 self.function_call_counters['alwaysTimesOut'] += 1 |
| 274 time.sleep(100) | 274 time.sleep(100) |
| 275 self._test_case.assertFalse(True, msg='Failed to time out?') | 275 self._test_case.assertFalse(True, msg='Failed to time out?') |
| 276 | 276 |
| 277 @decorators.WithTimeoutAndRetriesFromInstance( | 277 @decorators.WithTimeoutAndRetriesFromInstance( |
| 278 'default_timeout', 'default_retries') | 278 'default_timeout', 'default_retries') |
| 279 def alwaysRaisesCommandFailedError(self, timeout=None, retries=None): | 279 def alwaysRaisesCommandFailedError(self, timeout=None, retries=None): |
| 280 self.function_call_counters['alwaysRaisesCommandFailedError'] += 1 | 280 self.function_call_counters['alwaysRaisesCommandFailedError'] += 1 |
| 281 raise device_errors.CommandFailedError('testCommand failed') | 281 raise device_errors.CommandFailedError('testCommand failed') |
| 282 | 282 |
| 283 # pylint: disable=R0201 | 283 # pylint: disable=no-self-use |
| 284 | 284 |
| 285 @decorators.WithTimeoutAndRetriesFromInstance( | 285 @decorators.WithTimeoutAndRetriesFromInstance( |
| 286 'default_timeout', 'default_retries') | 286 'default_timeout', 'default_retries') |
| 287 def alwaysReturnsTimeout(self, timeout=None, retries=None): | 287 def alwaysReturnsTimeout(self, timeout=None, retries=None): |
| 288 return timeout | 288 return timeout |
| 289 | 289 |
| 290 @decorators.WithTimeoutAndRetriesFromInstance( | 290 @decorators.WithTimeoutAndRetriesFromInstance( |
| 291 'default_timeout', 'default_retries') | 291 'default_timeout', 'default_retries') |
| 292 def alwaysReturnsRetries(self, timeout=None, retries=None): | 292 def alwaysReturnsRetries(self, timeout=None, retries=None): |
| 293 return retries | 293 return retries |
| 294 | 294 |
| 295 @decorators.WithTimeoutAndRetriesFromInstance( | 295 @decorators.WithTimeoutAndRetriesFromInstance( |
| 296 'default_timeout', 'default_retries') | 296 'default_timeout', 'default_retries') |
| 297 def alwaysRaisesProvidedException(self, exception, timeout=None, | 297 def alwaysRaisesProvidedException(self, exception, timeout=None, |
| 298 retries=None): | 298 retries=None): |
| 299 raise exception | 299 raise exception |
| 300 | 300 |
| 301 # pylint: enable=R0201 | 301 # pylint: enable=no-self-use |
| 302 | 302 |
| 303 | 303 |
| 304 def testMethodDecoratorDoesTimeout(self): | 304 def testMethodDecoratorDoesTimeout(self): |
| 305 """Tests that the method decorator handles timeout logic.""" | 305 """Tests that the method decorator handles timeout logic.""" |
| 306 test_obj = self._MethodDecoratorTestObject(self) | 306 test_obj = self._MethodDecoratorTestObject(self) |
| 307 start_time = time.time() | 307 start_time = time.time() |
| 308 with self.assertRaises(device_errors.CommandTimeoutError): | 308 with self.assertRaises(device_errors.CommandTimeoutError): |
| 309 try: | 309 try: |
| 310 test_obj.alwaysTimesOut(timeout=1, retries=0) | 310 test_obj.alwaysTimesOut(timeout=1, retries=0) |
| 311 except: | 311 except: |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 356 | 356 |
| 357 exception_desc = 'Reraiser thread timeout error' | 357 exception_desc = 'Reraiser thread timeout error' |
| 358 with self.assertRaises(device_errors.CommandTimeoutError) as e: | 358 with self.assertRaises(device_errors.CommandTimeoutError) as e: |
| 359 test_obj.alwaysRaisesProvidedException( | 359 test_obj.alwaysRaisesProvidedException( |
| 360 reraiser_thread.TimeoutError(exception_desc)) | 360 reraiser_thread.TimeoutError(exception_desc)) |
| 361 self.assertEquals(exception_desc, str(e.exception)) | 361 self.assertEquals(exception_desc, str(e.exception)) |
| 362 | 362 |
| 363 if __name__ == '__main__': | 363 if __name__ == '__main__': |
| 364 unittest.main(verbosity=2) | 364 unittest.main(verbosity=2) |
| 365 | 365 |
| OLD | NEW |