def test_retries_function_given_number_of_times(self):
a, k = (1, 2, 3), {'1': 1, '2': 2, '3': 3}
globals()['counter'] = 5
def test_func(*args, **kwargs):
global counter
if counter > 0:
counter -= 1
raise Exception()
return args, kwargs
wrapped = retry(test_func, (Exception), attempts=6)
self.assertEqual((wrapped(*a, **k)), (a, k))
def test_retries_function_given_number_of_times(self):
a, k = (1, 2, 3), {'1': 1, '2': 2, '3': 3}
globals()['counter'] = 5
def test_func(*args, **kwargs):
global counter
if counter > 0:
counter -= 1
raise Exception()
return args, kwargs
wrapped = retry(test_func, (Exception), attempts=6)
self.assertEqual((wrapped(*a, **k)), (a, k))
def test_raises_error_if_function_still_erroring(self):
a, k = (1, 2, 3), {'1': 1, '2': 2, '3': 3}
globals()['counter'] = 5
def test_func(*args, **kwargs):
global counter
if counter > 0:
counter -= 1
raise Exception()
return args, kwargs
wrapped = retry(test_func, (Exception), attempts=5)
with self.assertRaises(Exception):
wrapped(*a, **k)
def test_raises_error_if_function_still_erroring(self):
a, k = (1, 2, 3), {'1': 1, '2': 2, '3': 3}
globals()['counter'] = 5
def test_func(*args, **kwargs):
global counter
if counter > 0:
counter -= 1
raise Exception()
return args, kwargs
wrapped = retry(test_func, (Exception), attempts=5)
with self.assertRaises(Exception):
wrapped(*a, **k)
