fix tests

Kludex · Kludex · commit cbdd4b9f77d2 · 2020-11-05T01:48:05.000+01:00
diff --git a/tests/test_typing_python39.py b/tests/test_typing_python39.py
@@ -1,27 +1,24 @@
-import pytest
 from fastapi import FastAPI
 from fastapi.testclient import TestClient
 
 from .utils import skip_py38
 
 
 @skip_py38
-@pytest.mark.parametrize(
-    "test_type,expect",
-    [
-        (list[int], [1, 2, 3]),
-        (dict[str, list[int]], {"a": [1, 2, 3], "b": [4, 5, 6]}),
-        (set[int], {1, 2, 3}),
-        (tuple[int], (1, 2, 3)),
-    ],
-)
-def test_typing(test_type, expect):
-    app = FastAPI()
+def test_typing():
+    types = {
+        list[int]: [1, 2, 3],
+        dict[str, list[int]]: {"a": [1, 2, 3], "b": [4, 5, 6]},
+        set[int]: [1, 2, 3],  # `set` is converted to `list`
+        tuple[int, ...]: [1, 2, 3],  # `tuple` is converted to `list`
+    }
+    for test_type, expect in types.items():
+        app = FastAPI()
 
-    @app.get("/", response_model=test_type)
-    def get_endpoint():
-        return expect
+        @app.post("/", response_model=test_type)
+        def post_endpoint(input: test_type):
+            return input
 
-    res = TestClient(app).get("/")
-    assert res.status_code == 200
-    assert res.json() == expect
+        res = TestClient(app).post("/", json=expect)
+        assert res.status_code == 200, res.json()
+        assert res.json() == expect
diff --git a/tests/utils.py b/tests/utils.py
@@ -3,4 +3,4 @@
 import pytest
 
 skip_py36 = pytest.mark.skipif(sys.version_info < (3, 7), reason="skip python3.6")
-skip_py38 = pytest.mark.skipif(sys.version_info < (3, 9), reason="skip python 3.8")
+skip_py38 = pytest.mark.skipif(sys.version_info < (3, 9), reason="requires python3.9")
