diff --git a/tests/test_introspection/test_digital_ocean.py b/tests/test_introspection/test_digital_ocean.py index 167b35c5..076a2904 100644 --- a/tests/test_introspection/test_digital_ocean.py +++ b/tests/test_introspection/test_digital_ocean.py @@ -69,7 +69,7 @@ def test_cloud_mode_with_signature(self) -> None: assert isinstance( net.validate_response_sig( body=res.get_data(), - headers={k: v for k, v in res.headers}, + headers=res.headers, mode=server_lib.ServerKind.CLOUD, signing_key=self.signing_key, ), diff --git a/tests/test_introspection/test_flask.py b/tests/test_introspection/test_flask.py index d2f258ac..4c636327 100644 --- a/tests/test_introspection/test_flask.py +++ b/tests/test_introspection/test_flask.py @@ -76,7 +76,7 @@ def test_cloud_mode_with_signature(self) -> None: assert isinstance( net.validate_response_sig( body=res.get_data(), - headers={k: v for k, v in res.headers}, + headers=res.headers, mode=server_lib.ServerKind.CLOUD, signing_key=self.signing_key, ), @@ -115,7 +115,7 @@ def test_cloud_mode_with_signature_fallback(self) -> None: assert isinstance( net.validate_response_sig( body=res.get_data(), - headers={k: v for k, v in res.headers}, + headers=res.headers, mode=server_lib.ServerKind.CLOUD, signing_key=signing_key_fallback, ), diff --git a/tests/test_probes/test_flask.py b/tests/test_probes/test_flask.py index 993a1c41..4cd52704 100644 --- a/tests/test_probes/test_flask.py +++ b/tests/test_probes/test_flask.py @@ -46,7 +46,7 @@ def test_signed(self) -> None: assert isinstance( net.validate_response_sig( body=res.get_data(), - headers={k: v for k, v in res.headers}, + headers=res.headers, mode=server_lib.ServerKind.CLOUD, signing_key=self.signing_key, ),