add tests

Author: slv0

tests/test_apidocs.py

@@ -3,7 +3,7 @@
 from starlette.testclient import TestClient
 
 from star_resty.apidocs import setup_spec
-from .utils.method import CreateUser, GetUser, SearchUser
+from .utils.method import CreateUser, GetUser, SearchUser, GetItemsEcho
 
 
 def test_generate_api_docs():
@@ -104,3 +104,25 @@ def test_generate_api_docs_for_path():
                 'responses': {
                     '200': {'schema': {'$ref': '#/definitions/tests.utils.method.CreateUserResponse'}},
                     '400': {'description': 'Bad request'}}}}}
+
+
+def test_generate_api_docs_for_nested():
+    app = Starlette()
+
+    setup_spec(app, title='test')
+    app.add_route('/items', GetItemsEcho.as_endpoint(), methods=['GET'])
+
+    client = TestClient(app)
+    resp = client.get('/apidocs.json')
+    assert resp is not None
+    body = resp.json()
+    assert body is not None
+    assert body.get('paths') == {
+        '/items': {
+            'get': {'tags': ['items'], 'description': 'get items', 'produces': ['application/json'], 'parameters': [
+                {'in': 'query', 'name': 'id', 'required': False, 'collectionFormat': 'multi', 'type': 'array',
+                 'items': {'type': 'integer', 'format': 'int32'}}],
+                    'responses': {'200': {'schema': {'$ref': '#/definitions/tests.utils.method.ItemsModel'}},
+                                  '400': {'description': 'Bad request'}}}
+        }
+    }

tests/test_endpoints.py

@@ -3,7 +3,7 @@
 from starlette.responses import JSONResponse
 from starlette.testclient import TestClient
 
-from tests.utils.method import SearchUser
+from tests.utils.method import SearchUser, GetItemsEcho
 
 
 def test_search_user():
@@ -31,3 +31,14 @@ async def error(request, exc: ValidationError):
     client = TestClient(app)
     resp = client.get('/users/invalid?q=Test')
     assert resp.status_code == 400
+
+
+def test_nested_result():
+    app = Starlette()
+
+    app.add_route('/items', GetItemsEcho.as_endpoint(), methods=['GET'])
+    client = TestClient(app)
+    resp = client.get('/items?id=1&id=2&id=3')
+    assert resp.status_code == 200
+    data = resp.json()
+    assert data == {'items': [{'id': 1}, {'id': 2}, {'id': 3}]}

tests/utils/method.py

@@ -27,6 +27,25 @@ class SearchUserResponse(Schema):
     q = fields.String()
 
 
+class ItemsModel(Schema):
+    class Item(Schema):
+        id = fields.Integer()
+
+    items = fields.List(fields.Nested(Item))
+    success = fields.Bool(missing=True)
+
+
+class GetItemsEcho(Method):
+    class Input(Schema):
+        id = fields.List(fields.Integer())
+
+    meta = Operation(tag='items', description='get items')
+    response_schema = ItemsModel
+
+    async def execute(self, items: query(Input)):
+        return {'items': [{'id': item_id} for item_id in items['id']]}
+
+
 class CreateUser(Method):
     meta = Operation(tag='users', description='create user')
     response_schema = CreateUserResponse