diff --git a/tests/http/proxy/test_http_proxy_tls_interception.py b/tests/http/proxy/test_http_proxy_tls_interception.py index fad6eff189..497d80af90 100644 --- 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]