def test_args_and_kwargs_passed(self): """ Args and kwargs passed to `retry` or to the function decorated with `@retry_on_error` should be passed to the wrapped function. """ def flakey(a, b, c=None): self.assertEqual(a, 1) self.assertEqual(b, 2) self.assertEqual(c, 3) retry(flakey, 1, 2, c=3) retry_on_error()(flakey)(1, 2, c=3)
def test_args_and_kwargs_passed(self): """ Args and kwargs passed to `retry` or to the function decorated with `@retry_on_error` should be passed to the wrapped function. """ def flakey(a, b, c=None): self.assertEqual(a, 1) self.assertEqual(b, 2) self.assertEqual(c, 3) retry(flakey, 1, 2, c=3) retry_on_error()(flakey)(1, 2, c=3)
def test_catch_param(self): """ It should only catch the exceptions given. """ def flakey(): flakey.attempts += 1 raise ValueError("Oops") flakey.attempts = 0 # If we only except TypeError then ValueError should raise immediately self.assertRaises(ValueError, retry_on_error(_catch=TypeError)(flakey)) self.assertEqual(flakey.attempts, 1) # Only 1 attempt should have been made # With the correct _catch param, it should catch our exception flakey.attempts = 0 # reset self.assertRaises(ValueError, retry_on_error(_catch=ValueError, _initial_wait=0, _attempts=5)(flakey)) self.assertEqual(flakey.attempts, 5)
def test_catch_param(self): """ It should only catch the exceptions given. """ def flakey(): flakey.attempts += 1 raise ValueError("Oops") flakey.attempts = 0 # If we only except TypeError then ValueError should raise immediately self.assertRaises(ValueError, retry_on_error(_catch=TypeError)(flakey)) self.assertEqual(flakey.attempts, 1) # Only 1 attempt should have been made # With the correct _catch param, it should catch our exception flakey.attempts = 0 # reset self.assertRaises( ValueError, retry_on_error(_catch=ValueError, _initial_wait=0, _attempts=5)(flakey)) self.assertEqual(flakey.attempts, 5)