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_no_exceptions2(self): self.assertEqual(autoretry_read(tries=3)(self.func_to_retry)(), "Finished.") self.assertEqual(self.func_to_retry.call_count, 1)