add tests of list response

Author: huangsong

pydantic_client/schema/method_info.py

@@ -1,4 +1,4 @@
-from typing import Any, Callable, Dict, Optional, Type, Union
+from typing import Any, Callable, Dict, Optional
 
 from pydantic import BaseModel
 
@@ -10,5 +10,5 @@ class MethodInfo(BaseModel):
     http_method: str
     url: str
     request_type: Dict[str, Any]
-    response_type: Optional[Type]
+    response_type: Optional[Any]
     form_body: bool

tests/conftest.py

@@ -1,4 +1,5 @@
 import pytest
+import typing
 from aiohttp import ClientSession
 from httpx import AsyncClient
 from requests import Session
@@ -12,7 +13,6 @@
 
 config_1 = ClientConfig(
     base_url=server_url,
-    timeout=10
 )
 
 config_2 = ClientConfig(
@@ -51,6 +51,10 @@
 @pydantic_client_manager.register(config_1)
 class R:
 
+    @get("/book_list")
+    def get_books(self) -> typing.List:
+        ...
+
     @get("/books/{book_id}?query={query}")
     def get_book(self, book_id: int, query: str) -> Book:
         ...
@@ -88,6 +92,11 @@ def download(self) -> File:
 
 @pydantic_client_manager.register(config_3)
 class AsyncR:
+
+    @get("/book_list")
+    async def get_books(self) -> typing.List:
+        ...
+
     @get("/books/{book_id}?query={query}")
     async def get_book(self, book_id: int, query: str) -> Book:
         ...
@@ -150,7 +159,6 @@ def fastapi_server_url() -> str:
     from threading import Thread
     host = "localhost"
     port = 12098  # TODO: add port availability check
-
     def start_server():
         run(app, host=host, port=port)
 

tests/fastapi_service.py

@@ -1,4 +1,5 @@
 import json
+import typing
 
 from fastapi import FastAPI, Form
 from starlette.responses import FileResponse
@@ -14,6 +15,11 @@ async def get() -> Book:
     return get_the_book()
 
 
+@app.get("/book_list")
+async def get_list() -> typing.List:
+    return ["1", "2"]
+
+
 @app.get("/books/{book_id}")
 def get_raw_book(book_id: int):
     return get_the_book()

tests/test_client.py

@@ -15,6 +15,15 @@ async def test_get(clients):
         assert book.age == 1
 
 
+@pytest.mark.asyncio
+async def test_get(clients):
+    for cl in clients:
+        book = cl.get_books()
+        if inspect.isawaitable(book):
+            book = await book
+        assert book == ["1", "2"]
+
+
 @pytest.mark.asyncio
 async def test_get_num_pages(clients):
     for cl in clients: