Improve aiohttp JSONMatcher support (#139)

KyleJamesWalker · sarayourfriend · web-flow · commit 26832315c713 · 2024-10-04T22:25:42.000-06:00
* Improve aiohttp JSONMatcher support.

* Lint

* Small simplification

---------

Co-authored-by: sarayourfriend &lt;git@sarayourfriend.pictures&gt;
diff --git a/src/pook/interceptors/aiohttp.py b/src/pook/interceptors/aiohttp.py
@@ -78,6 +78,12 @@ async def _on_request(
                 str(url) + "?" + urlencode([(x, y) for x, y in kw["params"].items()])
             )
 
+        # If a json payload is provided, serialize it for JSONMatcher support
+        if json_body := kw.get("json"):
+            req.json = json_body
+            if "Content-Type" not in req.headers:
+                req.headers["Content-Type"] = "application/json"
+
         # Match the request against the registered mocks in pook
         mock = self.engine.match(req)
 
diff --git a/tests/unit/interceptors/aiohttp_test.py b/tests/unit/interceptors/aiohttp_test.py
@@ -1,5 +1,6 @@
 import pytest
 import aiohttp
+import json
 
 import pook
 
@@ -55,3 +56,21 @@ async def test_binary_body(URL):
     async with aiohttp.ClientSession() as session:
         req = await session.get(URL)
         assert await req.read() == BINARY_FILE
+
+
+@pytest.mark.asyncio
+async def test_json_matcher_data_payload(URL):
+    payload = {"foo": "bar"}
+    pook.post(URL).json(payload).reply(200).body(BINARY_FILE)
+    async with aiohttp.ClientSession() as session:
+        req = await session.post(URL, data=json.dumps(payload))
+        assert await req.read() == BINARY_FILE
+
+
+@pytest.mark.asyncio
+async def test_json_matcher_json_payload(URL):
+    payload = {"foo": "bar"}
+    pook.post(URL).json(payload).reply(200).body(BINARY_FILE)
+    async with aiohttp.ClientSession() as session:
+        req = await session.post(URL, json=payload)
+        assert await req.read() == BINARY_FILE
