| 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 import time | 5 import time |
| 6 | 6 |
| 7 from testing_support import auto_stub | 7 from testing_support import auto_stub |
| 8 | 8 |
| 9 from infra.libs.service_utils import outer_loop | 9 from infra.libs.service_utils import outer_loop |
| 10 | 10 |
| 11 from infra_libs.ts_mon.common import interface | 11 from infra_libs.ts_mon.common import interface |
| 12 from infra_libs.ts_mon.common.test import stubs | |
| 13 | 12 |
| 14 | 13 |
| 15 class TestOuterLoop(auto_stub.TestCase): | 14 class TestOuterLoop(auto_stub.TestCase): |
| 16 | 15 |
| 17 class MyTime(): | 16 class MyTime(): |
| 18 def __init__(self): | 17 def __init__(self): |
| 19 self.sleeps = [] | 18 self.sleeps = [] |
| 20 self.now = 0 | 19 self.now = 0 |
| 21 def sleep(self, t): | 20 def sleep(self, t): |
| 22 self.sleeps.append(t) | 21 self.sleeps.append(t) |
| (...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 85 if t == 'err': | 84 if t == 'err': |
| 86 raise Exception('Horrible error') | 85 raise Exception('Horrible error') |
| 87 if t == 'false': | 86 if t == 'false': |
| 88 return False | 87 return False |
| 89 return True | 88 return True |
| 90 ret = outer_loop.loop(task, sleep_timeout=lambda: 1, max_errors=3, | 89 ret = outer_loop.loop(task, sleep_timeout=lambda: 1, max_errors=3, |
| 91 time_mod=self.time_mod) | 90 time_mod=self.time_mod) |
| 92 self.assertEqual(outer_loop.LoopResults(False, 5), ret) | 91 self.assertEqual(outer_loop.LoopResults(False, 5), ret) |
| 93 self.assertEqual(['skipped'], tasks) | 92 self.assertEqual(['skipped'], tasks) |
| 94 self.assertEqual([1, 1, 1, 1, 1, 1], self.time_mod.sleeps) | 93 self.assertEqual([1, 1, 1, 1, 1, 1], self.time_mod.sleeps) |
| OLD | NEW |