From c3591b32b950066a29c903c9204110df7b30a275 Mon Sep 17 00:00:00 2001 From: Lucas Wiman Date: Fri, 24 Jun 2022 12:00:01 -0700 Subject: [PATCH] Make tests more end-to-end to keep code coverage high. --- tests/forward_reference_type.py | 2 +- tests/test_wrapped_method_forward_reference.py | 10 ++++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/forward_reference_type.py b/tests/forward_reference_type.py index 02be56bec..b9cb9c22c 100644 --- a/tests/forward_reference_type.py +++ b/tests/forward_reference_type.py @@ -2,7 +2,7 @@ from pydantic import BaseModel def forwardref_method(input: "ForwardRef") -> "ForwardRef": - return ForwardRef() + return ForwardRef(x=input.x + 1) class ForwardRef(BaseModel): diff --git a/tests/test_wrapped_method_forward_reference.py b/tests/test_wrapped_method_forward_reference.py index 2b4d25b81..f01fe1474 100644 --- a/tests/test_wrapped_method_forward_reference.py +++ b/tests/test_wrapped_method_forward_reference.py @@ -1,6 +1,7 @@ import functools from fastapi import FastAPI +from fastapi.testclient import TestClient from .forward_reference_type import forwardref_method @@ -21,5 +22,10 @@ def test_wrapped_method_type_inference(): references. """ app = FastAPI() - app.get("/endpoint")(passthrough(forwardref_method)) - app.get("/endpoint2")(passthrough(passthrough(forwardref_method))) + client = TestClient(app) + app.post("/endpoint")(passthrough(forwardref_method)) + app.post("/endpoint2")(passthrough(passthrough(forwardref_method))) + with client: + response = client.post("/endpoint", json={"input": {"x": 0}}) + response2 = client.post("/endpoint2", json={"input": {"x": 0}}) + assert response.json() == response2.json() == {"x": 1}