From 0be9dac425aa5390275143e1aaae98bc85d6014f Mon Sep 17 00:00:00 2001
From: Aaron Harper <aaron@inngest.com>
Date: Fri, 22 Nov 2024 10:03:09 -0500
Subject: [PATCH] Revert some header changes

---
 tests/test_introspection/test_digital_ocean.py | 2 +-
 tests/test_introspection/test_flask.py         | 4 ++--
 tests/test_probes/test_flask.py                | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

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,
             ),