happyz
/
fastapi
mirror of https://github.com/tiangolo/fastapi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make tests more end-to-end to keep code coverage high.

This commit is contained in:
Lucas Wiman 2022-06-24 12:00:01 -07:00
parent c1691f5d94
commit c3591b32b9
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/forward_reference_type.py
View File

@ -2,7 +2,7 @@ from pydantic import BaseModel
def forwardref_method(input: "ForwardRef") -> "ForwardRef": def forwardref_method(input: "ForwardRef") -> "ForwardRef":
return ForwardRef() return ForwardRef(x=input.x + 1)
class ForwardRef(BaseModel): class ForwardRef(BaseModel):

10
tests/test_wrapped_method_forward_reference.py
View File

@ -1,6 +1,7 @@
import functools import functools
from fastapi import FastAPI from fastapi import FastAPI
from fastapi.testclient import TestClient
from .forward_reference_type import forwardref_method from .forward_reference_type import forwardref_method
@ -21,5 +22,10 @@ def test_wrapped_method_type_inference():
references. references.
""" """
app = FastAPI() app = FastAPI()
app.get("/endpoint")(passthrough(forwardref_method)) client = TestClient(app)
app.get("/endpoint2")(passthrough(passthrough(forwardref_method))) 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}