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()