Coverage for tests / test_wrapped_method_forward_reference.py: 100%

1import functools 4 ctx1efgh

2 

3from fastapi import FastAPI 4 ctx1efgh

4from fastapi.testclient import TestClient 4 ctx1efgh

5 

6from .forward_reference_type import forwardref_method 4 ctx1efgh

7 

8 

9def passthrough(f): 4 ctx1efgh

10 @functools.wraps(f) 4 ctx1abcd

11 def method(*args, **kwargs): 4 ctx1abcd

12 return f(*args, **kwargs) 4 ctx1abcd

13 

14 return method 4 ctx1abcd

15 

16 

17def test_wrapped_method_type_inference(): 4 ctx1efgh

18 """ 

19 Regression test ensuring that when a method imported from another module 

20 is decorated with something that sets the __wrapped__ attribute (functools.wraps), 

21 then the types are still processed correctly, including dereferencing of forward 

22 references. 

23 """ 

24 app = FastAPI() 4 ctx1abcd

25 client = TestClient(app) 4 ctx1abcd

26 app.post("/endpoint")(passthrough(forwardref_method)) 4 ctx1abcd

27 app.post("/endpoint2")(passthrough(passthrough(forwardref_method))) 4 ctx1abcd

28 with client: 4 ctx1abcd

29 response = client.post("/endpoint", json={"input": {"x": 0}}) 4 ctx1abcd

30 response2 = client.post("/endpoint2", json={"input": {"x": 0}}) 4 ctx1abcd

31 assert response.json() == response2.json() == {"x": 1} 4 ctx1abcd