From e249797eb79a8163146d34e7e384ef0017b859b7 Mon Sep 17 00:00:00 2001 From: Tulsi Shah Date: Fri, 18 Oct 2024 04:23:21 +0000 Subject: [PATCH] lint fixes --- testbench/common.py | 1 + tests/test_testbench_retry.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) 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()