def test_custom_retry_failure(error_code):
def callback():
raise ClientError({'Error': {'Code': 'test-error'}}, 'test-op')
with pytest.raises(ClientError) as exc_info:
batch_versioning.custom_retry(callback, error_code, 3)
assert exc_info.value.response['Error']['Code'] == 'test-error'
def test_custom_retry():
def callback():
return 'Success!'
response = batch_versioning.custom_retry(callback, 'test-error', 3)
assert response == 'Success!'