fix pydantic return type serializer to support lists (#17)

pyproject.toml

@@ -7,7 +7,7 @@ name = "rolo"
 authors = [
     { name = "LocalStack Contributors", email = "[email protected]" }
 ]
-version = "0.6.0"
+version = "0.6.1"
 description = "A Python framework for building HTTP-based server applications"
 dependencies = [
     "requests>=2.20",

rolo/dispatcher.py

@@ -120,6 +120,14 @@ def _dispatch(request: Request, endpoint: Handler, args: RequestArguments) -> Re
 
             if isinstance(result, pydantic.BaseModel):
                 result = result.model_dump()
+            if isinstance(result, (list, tuple)):
+                converted = []
+                for element in result:
+                    if isinstance(element, pydantic.BaseModel):
+                        converted.append(element.model_dump())
+                    else:
+                        converted.append(element)
+                result = converted
 
         else:
             result = endpoint(request, **args)

tests/test_pydantic.py

@@ -82,6 +82,31 @@ def handler(_request: Request, item_id: int) -> MyItem:
             "is_offer": None,
         }
 
+    def test_response_list(self):
+        router = Router(dispatcher=handler_dispatcher())
+
+        def handler(_request: Request) -> list[MyItem]:
+            return [
+                MyItem(name="rolo", price=420.69),
+                MyItem(name="twiks", price=1.23, is_offer=True),
+            ]
+
+        router.add("/items", handler)
+
+        request = Request("GET", "/items")
+        assert router.dispatch(request).get_json() == [
+            {
+                "name": "rolo",
+                "price": 420.69,
+                "is_offer": None,
+            },
+            {
+                "name": "twiks",
+                "price": 1.23,
+                "is_offer": True,
+            },
+        ]
+
     def test_request_arg_validation_error(self):
         router = Router(dispatcher=handler_dispatcher())