X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/7178ac911e7b160c1348da404b5aa07b9829d3c8..e6c2f09b14d03db040f2b81c451a7630bb78f1b1:/sdk/python/tests/test_retry.py diff --git a/sdk/python/tests/test_retry.py b/sdk/python/tests/test_retry.py index 8c3916b009..0c1110c5ce 100644 --- a/sdk/python/tests/test_retry.py +++ b/sdk/python/tests/test_retry.py @@ -7,7 +7,7 @@ import arvados.errors as arv_error import arvados.retry as arv_retry import mock -from arvados_testutil import fake_httplib2_response +from arvados_testutil import fake_requests_response class RetryLoopTestMixin(object): @staticmethod @@ -150,7 +150,7 @@ class RetryLoopBackoffTestCase(unittest.TestCase, RetryLoopTestMixin): class CheckHTTPResponseSuccessTestCase(unittest.TestCase): def results_map(self, *codes): for code in codes: - response = (fake_httplib2_response(code), None) + response = fake_requests_response(code, None) yield code, arv_retry.check_http_response_success(response) def check(assert_name):