Ensure aiohttp headers matchable from both session and request

History.rst

@@ -1,6 +1,11 @@
 History
 =======
 
+v2.1.3 / 2024-12-16
+-------------------------
+
+  * Ensure aiohttp headers can be matched from both the session and request in the same matcher
+
 v2.1.2 / 2024-11-21
 -------------------------
 

pyproject.toml

@@ -10,6 +10,7 @@ readme = "README.rst"
 license = "MIT"
 authors = [
     { name = "Tomas Aparicio", email = "[email protected]" },
+    { name = "Sara Marcondes", email = "[email protected]" },
 ]
 classifiers = [
     "Development Status :: 5 - Production/Stable",

src/pook/__init__.py

@@ -44,4 +44,4 @@
 __license__ = "MIT"
 
 # Current version
-__version__ = "2.1.2"
+__version__ = "2.1.3"

src/pook/interceptors/aiohttp.py

@@ -58,18 +58,15 @@ def set_headers(self, req, headers) -> None:
         # ``pook.request`` only allows a dict, so we need to map the iterable to the matchable interface
         if headers:
             if isinstance(headers, Mapping):
-                req.headers = headers
+                req.headers.update(**headers)
             else:
-                req_headers: dict[str, str] = {}
                 # If it isn't a mapping, then its an Iterable[Tuple[Union[str, istr], str]]
                 for req_header, req_header_value in headers:
                     normalised_header = req_header.lower()
-                    if normalised_header in req_headers:
-                        req_headers[normalised_header] += f", {req_header_value}"
+                    if normalised_header in req.headers:
+                        req.headers[normalised_header] += f", {req_header_value}"
                     else:
-                        req_headers[normalised_header] = req_header_value
-
-                req.headers = req_headers
+                        req.headers[normalised_header] = req_header_value
 
     async def _on_request(
         self,

tests/unit/interceptors/aiohttp_test.py

@@ -86,3 +86,31 @@ async def test_client_headers(httpbin):
         res = await session.get(httpbin + "/status/404")
         assert res.status == 200
         assert await res.read() == b"hello from pook"
+
+
+@pytest.mark.asyncio
+async def test_client_headers_merged(httpbin):
+    """Headers set on the client should be matched even if request-specific headers are sent."""
+    pook.get(httpbin + "/status/404").header("x-pook", "hello").reply(200).body(
+        "hello from pook"
+    )
+    async with aiohttp.ClientSession(headers={"x-pook": "hello"}) as session:
+        res = await session.get(
+            httpbin + "/status/404", headers={"x-pook-secondary": "xyz"}
+        )
+        assert res.status == 200
+        assert await res.read() == b"hello from pook"
+
+
+@pytest.mark.asyncio
+async def test_client_headers_both_session_and_request(httpbin):
+    """Headers should be matchable from both the session and request in the same matcher"""
+    pook.get(httpbin + "/status/404").header("x-pook-session", "hello").header(
+        "x-pook-request", "hey"
+    ).reply(200).body("hello from pook")
+    async with aiohttp.ClientSession(headers={"x-pook-session": "hello"}) as session:
+        res = await session.get(
+            httpbin + "/status/404", headers={"x-pook-request": "hey"}
+        )
+        assert res.status == 200
+        assert await res.read() == b"hello from pook"

tests/unit/interceptors/base.py

@@ -208,7 +208,7 @@ def test_header_sent(self, url_404):
         assert body == b"hello from pook"
 
     @pytest.mark.pook
-    def test_mocked_resposne_headers(self, url_404):
+    def test_mocked_response_headers(self, url_404):
         """Mocked response headers are appropriately returned."""
         pook.get(url_404).reply(200).header("x-hello", "from pook")
 
@@ -238,3 +238,11 @@ def test_mutli_value_response_headers(self, url_404):
 
         assert status == 200
         assert headers["x-hello"] == "from pook, another time"
+
+    @pytest.mark.pook(allow_pending_mocks=True)
+    def test_unmatched_headers_none_sent(self, url_404):
+        """Header matching will run, but not match, on requests that send no headers."""
+        pook.get(url_404).header("x-hello", "from pook").reply(200)
+
+        with pytest.raises(PookNoMatches):
+            self.make_request("GET", url_404)