def test_call(self, m_sleep, m_count):
m_handler = mock.Mock()
m_count.return_value = list(range(1, 5))
retry = h_retry.Retry(m_handler)
retry(mock.sentinel.event)
m_handler.assert_called_once_with(mock.sentinel.event)
m_sleep.assert_not_called()
def test_should_not_sleep(self, m_sleep, m_randint):
deadline = self.now - 1
retry = h_retry.Retry(mock.Mock())
ret = retry._sleep(deadline, 1, _EX1())
self.assertFalse(ret)
m_sleep.assert_not_called()
m_randint.assert_not_called()
def test_call_raises_no_retry(self, m_sleep, m_count):
event = mock.sentinel.event
m_handler = mock.Mock()
m_handler.side_effect = _EX1()
m_count.return_value = list(range(1, 5))
retry = h_retry.Retry(m_handler, exceptions=(_EX11, _EX2))
self.assertRaises(_EX1, retry, event)
m_handler.assert_called_once_with(event)
m_sleep.assert_not_called()
def _test_should_sleep(self, seconds_left, slept):
attempt = 5
timeout = 20
interval = 3
randint = 2
deadline = self.now + seconds_left
retry = h_retry.Retry(mock.Mock(), timeout=timeout, interval=interval)
with mock.patch('random.randint') as m_randint, \
mock.patch('time.sleep') as m_sleep:
m_randint.return_value = randint
ret = retry._sleep(deadline, attempt, _EX2())
self.assertEqual(slept, ret)
m_randint.assert_called_once_with(1, 2**attempt - 1)
m_sleep.assert_called_once_with(slept)
def test_call_retry(self, m_sleep, m_count):
attempts = 3
timeout = 10
deadline = self.now + timeout
failures = [_EX1()] * (attempts - 1)
event = mock.sentinel.event
m_handler = mock.Mock()
m_handler.side_effect = failures + [None]
m_sleep.return_value = 1
m_count.return_value = list(range(1, 5))
retry = h_retry.Retry(m_handler, timeout=timeout, exceptions=_EX1)
retry(event)
m_handler.assert_has_calls([mock.call(event)] * attempts)
m_sleep.assert_has_calls([
mock.call(deadline, i + 1, failures[i])
for i in range(len(failures))
])
def _wrap_consumer(self, consumer):
# TODO(ivc): tune retry interval/timeout
return h_log.LogExceptions(
h_retry.Retry(consumer, exceptions=exceptions.ResourceNotReady))