diff --git a/testbench/common.py b/testbench/common.py index 5f7e2433..be43cb6b 100644 --- a/testbench/common.py +++ b/testbench/common.py @@ -922,6 +922,7 @@ def get_stall_uploads_after_bytes( return 0, 0, "" + def get_retry_uploads_error_after_bytes( database, request, context=None, transport="HTTP" ): diff --git a/tests/test_testbench_retry.py b/tests/test_testbench_retry.py index 202d7436..ebc0d8f4 100644 --- a/tests/test_testbench_retry.py +++ b/tests/test_testbench_retry.py @@ -675,7 +675,7 @@ def test_write_retry_test_stall_after_bytes(self): } } ), - content_type="application/json" + content_type="application/json", ) self.assertEqual(response.status_code, 200) self.assertTrue( @@ -761,6 +761,7 @@ def test_write_retry_test_stall_after_bytes(self): self.assertLess(elapsed_time, 1) self.assertEqual(response.status_code, 200, msg=response.data) + class TestTestbenchRetryGrpc(unittest.TestCase): def setUp(self): rest_server.db.clear()