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)