def test_exceptions3(self):
with self.assertRaises(AutoReconnect):
self.func_to_retry.exceptions_to_raise = [
AutoReconnect, AutoReconnect, AutoReconnect, AutoReconnect
]
autoretry_read(tries=4)(self.func_to_retry)()
self.assertEqual(self.func_to_retry.call_count, 4)
def test_exceptions1(self):
with self.assertRaises(ValueError):
self.func_to_retry.exceptions_to_raise = [ValueError]
autoretry_read(tries=3)(self.func_to_retry)()
self.assertEqual(self.func_to_retry.call_count, 1)
def test_exceptions2(self):
self.func_to_retry.exceptions_to_raise = [AutoReconnect]
self.assertEqual(
autoretry_read(tries=3)(self.func_to_retry)(), "Finished.")
self.assertEqual(self.func_to_retry.call_count, 2)
def test_no_exceptions2(self):
self.assertEqual(
autoretry_read(tries=3)(self.func_to_retry)(), "Finished.")
self.assertEqual(self.func_to_retry.call_count, 1)
def test_exceptions3(self):
with self.assertRaises(AutoReconnect):
self.func_to_retry.exceptions_to_raise = [AutoReconnect, AutoReconnect, AutoReconnect, AutoReconnect]
autoretry_read(tries=4)(self.func_to_retry)()
self.assertEqual(self.func_to_retry.call_count, 4)
def test_exceptions2(self):
self.func_to_retry.exceptions_to_raise = [AutoReconnect]
self.assertEqual(autoretry_read(tries=3)(self.func_to_retry)(), "Finished.")
self.assertEqual(self.func_to_retry.call_count, 2)
def test_exceptions1(self):
with self.assertRaises(ValueError):
self.func_to_retry.exceptions_to_raise = [ValueError]
autoretry_read(tries=3)(self.func_to_retry)()
self.assertEqual(self.func_to_retry.call_count, 1)
def test_no_exceptions2(self):
self.assertEqual(autoretry_read(tries=3)(self.func_to_retry)(), "Finished.")
self.assertEqual(self.func_to_retry.call_count, 1)
