def test_total_failure_exception(self):
self.f.side_effect = Exception('some persistent error')
try:
retry( 5, self.f, self.fparms )
self.fail("Exception expected")
except Exception, e:
self.assertEqual( 'some persistent error', str(e) )
self.f.assert_called_with( 'parm1',2,True )
self.assertEqual(5, self.f.call_count)
def test_total_failure_exception(self):
self.f.side_effect = Exception('some persistent error')
try:
retry(5, self.f, self.fparms)
self.fail("Exception expected")
except Exception, e:
self.assertEqual('some persistent error', str(e))
self.f.assert_called_with('parm1', 2, True)
self.assertEqual(5, self.f.call_count)
def verify_retry_failures(self, retry_count, fail_count, sleep_time):
self.f.side_effect = ([ KeyError('somekey')]*fail_count) + [ None ]
retry( retry_count, self.f, self.fparms, sleep_secs=sleep_time )
self.f.assert_called_with( 'parm1',2,True )
self.assertTrue(self.f.call_count == fail_count + 1)
def test_retry_success_retval(self):
self.f.return_value = "Success"
retval = retry( 5, self.f, self.fparms )
self.f.assert_called_with( 'parm1',2,True )
self.assertEqual("Success", retval)
def test_retry_success(self):
retry( 5, self.f, self.fparms )
self.f.assert_called_with( 'parm1',2,True )
def verify_retry_failures(self, retry_count, fail_count, sleep_time):
self.f.side_effect = ([KeyError('somekey')] * fail_count) + [None]
retry(retry_count, self.f, self.fparms, sleep_secs=sleep_time)
self.f.assert_called_with('parm1', 2, True)
self.assertTrue(self.f.call_count == fail_count + 1)
def test_retry_success_retval(self):
self.f.return_value = "Success"
retval = retry(5, self.f, self.fparms)
self.f.assert_called_with('parm1', 2, True)
self.assertEqual("Success", retval)
def test_retry_success(self):
retry(5, self.f, self.fparms)
self.f.assert_called_with('parm1', 2, True)
