Merge pull request #108 from kouk/async-support

Support async views. Fixes #91

Author: stv8

README.md

@@ -335,6 +335,10 @@ Feature requests and pull requests are welcome. For major changes, please open a
   python3 -m venv venv
   source venv/bin/activate
   ```
+- install runtime dependencies
+  ```bash
+  python3 -m pip install -e .
+  ```
 - install development requirements
   ```bash
   python3 -m pip install -r requirements/test.txt

flask_pydantic/core.py

@@ -282,7 +282,7 @@ def wrapper(*args, **kwargs):
                     return make_response(
                         jsonify({"validation_error": err}), status_code
                     )
-            res = func(*args, **kwargs)
+            res = current_app.ensure_sync(func)(*args, **kwargs)
 
             if response_many:
                 if is_iterable_of_models(res):

requirements/test.txt

@@ -3,3 +3,4 @@ pytest-flask
 pytest-coverage
 pytest-mock
 pytest-ruff
+Flask[async]

tests/func/test_app.py

@@ -1,3 +1,4 @@
+import asyncio
 import re
 from typing import List, Optional
 
@@ -147,6 +148,21 @@ def compute(query: RequestModel):
         )
 
 
+@pytest.fixture
+def app_with_async_route(app):
+    class RequestModel(BaseModel):
+        n: int
+
+    class ResultModel(BaseModel):
+        result: int
+
+    @app.route("/compute", methods=["POST"])
+    @validate()
+    async def compute(body: RequestModel):
+        await asyncio.sleep(0.1)
+        return ResultModel(result=2 * body.n)
+
+
 test_cases = [
     pytest.param(
         "?limit=limit",
@@ -469,3 +485,34 @@ def test_silent(self, client):
             response.json["body"],
         )
         assert response.status_code == 422
+
+
+@pytest.mark.usefixtures("app_with_async_route")
+class TestAsyncRoute:
+    def test_fails(self, client):
+        response = client.post("/compute", json={})
+
+        assert_matches(
+            {
+                "validation_error": {
+                    "body_params": [
+                        {
+                            "input": {},
+                            "loc": ["n"],
+                            "msg": "Field required",
+                            "type": "missing",
+                            "url": re.compile(
+                                r"https://errors\.pydantic\.dev/.*/v/missing"
+                            ),
+                        }
+                    ]
+                }
+            },
+            response.json,
+        )
+
+    def test_succeeds(self, client):
+        expected_response = {"result": 2}
+        response = client.post("/compute", json={"n": 1})
+
+        assert_matches(expected_response, response.json)