def test_retry_server_errors_validtion_error(self):
mock = Mock(side_effect=ValidationError("Mock Error", None))
test_function = intrinio_data.retry_server_errors(mock)
with self.assertRaises(ValidationError):
test_function()
self.assertEqual(mock.call_count, 1)
def test_retry_server_errors_api_error_501(self):
with patch.object(time, 'sleep', return_value=None):
mock = Mock(side_effect=DataError(
"Mock error", ApiException(501, "Mock Error")))
test_function = intrinio_data.retry_server_errors(mock)
with self.assertRaises(DataError):
test_function()
self.assertEqual(mock.call_count, self.RETRY_ERROR_COUNT)
def test_retry_server_errors_no_exception(self):
mock = Mock(return_value=None)
test_function = intrinio_data.retry_server_errors(mock)
test_function()