Coverage for tests / test_wrapped_method_forward_reference.py: 100%

1import functools 4 ctx1defg

2 

3from fastapi import FastAPI 4 ctx1defg

4from fastapi.testclient import TestClient 4 ctx1defg

5 

6from .forward_reference_type import forwardref_method 4 ctx1defg

7 

8 

9def passthrough(f): 4 ctx1defg

10 @functools.wraps(f) 3 ctx1abc

11 def method(*args, **kwargs): 3 ctx1abc

12 return f(*args, **kwargs) 3 ctx1abc

13 

14 return method 3 ctx1abc

15 

16 

17def test_wrapped_method_type_inference(): 4 ctx1defg

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() 3 ctx1abc

25 client = TestClient(app) 3 ctx1abc

26 app.post("/endpoint")(passthrough(forwardref_method)) 3 ctx1abc

27 app.post("/endpoint2")(passthrough(passthrough(forwardref_method))) 3 ctx1abc

28 with client: 3 ctx1abc

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

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

31 assert response.json() == response2.json() == {"x": 1} 3 ctx1abc