Modify unittest

0b01001001 · May 13, 2021 · ac337ad · ac337ad
1 parent f1b68c9
commit ac337ad
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 9 deletions.
diff --git a/tests/test_plugin.py b/tests/test_plugin.py
@@ -13,7 +13,8 @@
 )
 def test_plugin_spec(api):
     models = {
-        m.__name__: m.schema(ref_template="#/components/schemas/{model}")
+        f"{m.__module__}.{m.__name__}": m.schema(
+            ref_template="#/components/schemas/{model}")
         for m in (Query, JSON, Resp, Cookies, Headers)
     }
     for name, schema in models.items():
@@ -49,7 +50,7 @@ def test_plugin_spec(api):
     assert user["tags"] == ["API", "test"]
     assert (
         user["requestBody"]["content"]["application/json"]["schema"]["$ref"]
-        == "#/components/schemas/JSON"
+        == "#/components/schemas/tests.common.JSON"
     )
     assert len(user["responses"]) == 3
 

diff --git a/tests/test_response.py b/tests/test_response.py
@@ -40,11 +40,11 @@ def test_response_spec():
     assert spec["422"]["description"] == DEFAULT_CODE_DESC["HTTP_422"]
     assert (
         spec["201"]["content"]["application/json"]["schema"]["$ref"].split("/")[-1]
-        == "DemoModel"
+        == "tests.common.DemoModel"
     )
     assert (
         spec["422"]["content"]["application/json"]["schema"]["$ref"].split("/")[-1]
-        == "UnprocessableEntity"
+        == "spectree.models.UnprocessableEntity"
     )
 
     assert spec.get(200) is None

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -85,25 +85,26 @@ def test_parse_resp():
     assert resp_spec["422"]["description"] == "Unprocessable Entity"
     assert (
         resp_spec["422"]["content"]["application/json"]["schema"]["$ref"]
-        == "#/components/schemas/UnprocessableEntity"
+        == "#/components/schemas/spectree.models.UnprocessableEntity"
     )
     assert (
         resp_spec["200"]["content"]["application/json"]["schema"]["$ref"]
-        == "#/components/schemas/DemoModel"
+        == "#/components/schemas/tests.common.DemoModel"
     )
 
 
 def test_parse_request():
     assert (
         parse_request(demo_func)["content"]["application/json"]["schema"]["$ref"]
-        == "#/components/schemas/DemoModel"
+        == "#/components/schemas/tests.common.DemoModel"
     )
     assert parse_request(demo_class.demo_method) == {}
 
 
 def test_parse_params():
     models = {
-        "DemoModel": DemoModel.schema(ref_template="#/components/schemas/{model}")
+        "tests.common.DemoModel": DemoModel.schema(
+            ref_template="#/components/schemas/{model}")
     }
     assert parse_params(demo_func, [], models) == []
     params = parse_params(demo_class.demo_method, [], models)
@@ -120,7 +121,8 @@ def test_parse_params():
 
 def test_parse_params_with_route_param_keywords():
     models = {
-        "DemoQuery": DemoQuery.schema(ref_template="#/components/schemas/{model}")
+        "tests.common.DemoQuery": DemoQuery.schema(
+            ref_template="#/components/schemas/{model}")
     }
     params = parse_params(demo_func_with_query, [], models)
     assert params == [