Improve aiohttp JSONMatcher support (#139)

* Improve aiohttp JSONMatcher support. * Lint * Small simplification --------- Co-authored-by: sarayourfriend <[email protected]>
h2non · Oct 5, 2024 · 2683231 · 2683231
1 parent d9e739f
commit 2683231
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
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