Fix test assertions

abhinavsingh · Sep 21, 2024 · d6e220f · d6e220f
1 parent 2c6808a
commit d6e220f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/http/proxy/test_http_proxy_tls_interception.py b/tests/http/proxy/test_http_proxy_tls_interception.py
@@ -204,7 +204,7 @@ async def asyncReturn(val: T) -> T:
         )
         self.proxy_plugin.return_value.before_upstream_connection.assert_called()
         self.proxy_plugin.return_value.handle_client_request.assert_called_once()
-        self.proxy_plugin.return_value.do_intercept.assert_called_once()
+        self.assertEqual(self.proxy_plugin.return_value.do_intercept.call_count, 2)
         # All the invoked lifecycle callbacks will receive the CONNECT request
         # packet with / as the path
         callback_request: HttpParser = \
@@ -297,7 +297,7 @@ async def asyncReturn(val: T) -> T:
             self.proxy_plugin.return_value.handle_client_request.call_count,
             2,
         )
-        self.proxy_plugin.return_value.do_intercept.assert_called_once()
+        self.assertEqual(self.proxy_plugin.return_value.do_intercept.call_count, 2)
 
         callback_request = \
             self.proxy_plugin.return_value.handle_client_request.call_args_list[1][0][0]