tests - fastapi model name generation changes

commonism · Oct 20, 2023 · c073188 · c073188
1 parent 30b3935
commit c073188
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/apiv1_test.py b/tests/apiv1_test.py
@@ -51,7 +51,7 @@ def randomPet(name=None):
 @pytest.mark.skipif(sys.version_info < (3, 9), reason="requires asyncio.to_thread")
 async def test_createPet(event_loop, server, client):
     h, r = await asyncio.to_thread(client._.createPet, **randomPet())
-    assert type(r).model_json_schema() == client.components.schemas["Pet-Input"].get_type().model_json_schema()
+    assert type(r).model_json_schema() == client.components.schemas["Pet"].get_type().model_json_schema()
     assert h["X-Limit-Remain"] == 5
 
     r = await asyncio.to_thread(client._.createPet, data={"pet": {"name": r.name}})

diff --git a/tests/apiv2_test.py b/tests/apiv2_test.py
@@ -161,7 +161,7 @@ async def test_Request(event_loop, server, client):
 @pytest.mark.asyncio
 async def test_createPet(event_loop, server, client):
     data = {
-        "pet": client.components.schemas["WhiteCat-Input"]
+        "pet": client.components.schemas["WhiteCat"]
         .model(
             {
                 "name": str(uuid.uuid4()),