def __init__(self, task_runner, timeout):
"""
Initialise with the TaskRunner and a timeout period in seconds.
"""
message = _('%s Timed out') % six.text_type(task_runner)
super(Timeout, self).__init__(message)
self._duration = timeutils.Duration(timeout)
def pause_between_batch(pause_sec):
duration = timeutils.Duration(pause_sec)
while not duration.expired():
yield
def test_duration_expired(self):
self.assertTrue(util.Duration(0.1).expired())
def test_duration_not_expired(self):
self.assertFalse(util.Duration(1.0).expired())
