Test PathEx parameter parsing with custom validator

Author: jceipek

tests/main.py

@@ -1,9 +1,10 @@
-from typing import List, Optional
+from typing import Annotated, List, Optional
 from uuid import UUID
 
+import pydantic
 from django.urls import register_converter
 
-from ninja import Field, Path, Query, Router, Schema
+from ninja import Field, P, Path, PathEx, Query, Router, Schema
 
 router = Router()
 
@@ -38,6 +39,29 @@ def get_bool_id(request, item_id: bool):
     return item_id
 
 
+def custom_validator(value: int) -> int:
+    if value != 42:
+        raise ValueError("Input should pass this custom validator")
+    return value
+
+
+CustomValidatedInt = Annotated[
+    int,
+    pydantic.AfterValidator(custom_validator),
+    pydantic.WithJsonSchema({
+        "type": "int",
+        "example": "42",
+    }),
+]
+
+
+@router.get("/path/param_ex/{item_id}")
+def get_path_param_ex_id(
+    request, item_id: PathEx[CustomValidatedInt, P(description="path_ex description")]
+):
+    return item_id
+
+
 @router.get("/path/param/{item_id}")
 def get_path_param_id(request, item_id: str = Path(None)):
     return item_id

tests/test_path.py

@@ -33,6 +33,17 @@ def test_text_get():
     ]
 }
 
+response_not_valid_custom = {
+    "detail": [
+        {
+            "ctx": {"error": "Input should pass this custom validator"},
+            "loc": ["path", "item_id"],
+            "msg": "Value error, Input should pass this custom validator",
+            "type": "value_error",
+        }
+    ]
+}
+
 response_not_valid_int_float = {
     "detail": [
         {
@@ -210,6 +221,9 @@ def test_text_get():
         ("/path/bool/true", 200, True),
         ("/path/bool/False", 200, False),
         ("/path/bool/false", 200, False),
+        ("/path/param_ex/True", 422, response_not_valid_int),
+        ("/path/param_ex/0", 422, response_not_valid_custom),
+        ("/path/param_ex/42", 200, 42),
         ("/path/param/foo", 200, "foo"),
         ("/path/param-required/foo", 200, "foo"),
         ("/path/param-minlength/foo", 200, "foo"),