diff --git a/testbench/common.py b/testbench/common.py index 5b1f405a..5f7e2433 100644 --- a/testbench/common.py +++ b/testbench/common.py @@ -893,6 +893,7 @@ def wrapper(*args, **kwargs): return retry_test + def get_stall_uploads_after_bytes( database, request, context=None, transport="HTTP" ): @@ -948,6 +949,7 @@ def get_retry_uploads_error_after_bytes( return 0, 0, "" + def handle_stall_uploads_after_bytes( upload, data, diff --git a/testbench/rest_server.py b/testbench/rest_server.py index 11c06d02..2efddee3 100644 --- a/testbench/rest_server.py +++ b/testbench/rest_server.py @@ -1170,7 +1170,6 @@ def resumable_upload_chunk(bucket_name): return upload.resumable_status_rest(override_308=override_308) - @upload.route("/b//o", methods=["DELETE"]) @retry_test(method="storage.objects.delete") def delete_resumable_upload(bucket_name): diff --git a/tests/test_testbench_retry.py b/tests/test_testbench_retry.py index 9c7cf7d6..668fd3d8 100644 --- a/tests/test_testbench_retry.py +++ b/tests/test_testbench_retry.py @@ -665,18 +665,22 @@ def test_write_retry_test_stall_after_bytes(self): # Setup a stall for reading back the object. response = self.client.post( "/retry_test", - data=json.dumps({ - "instructions": { - "storage.objects.insert": [ - "stall-for-1s-after-250K", - "stall-for-1s-after-300K" - ] + data=json.dumps( + { + "instructions": { + "storage.objects.insert": [ + "stall-for-1s-after-250K", + "stall-for-1s-after-300K" + ] + } } - }), + ), content_type='application/json' ) self.assertEqual(response.status_code, 200) - self.assertTrue(response.headers.get("content-type").startswith("application/json")) + self.assertTrue( + response.headers.get("content-type").startswith("application/json") + ) create_rest = json.loads(response.data) self.assertIn("id", create_rest)