Coverage for tests / test_request_param_model_by_alias.py: 100%

1from dirty_equals import IsPartialDict 4 ctx1abcd

2from fastapi import Cookie, FastAPI, Header, Query 4 ctx1abcd

3from fastapi.testclient import TestClient 4 ctx1abcd

4from pydantic import BaseModel, Field 4 ctx1abcd

5 

6app = FastAPI() 4 ctx1abcd

7 

8 

9class Model(BaseModel): 4 ctx1abcd

10 param: str = Field(alias="param_alias") 4 ctx1abcd

11 

12 

13@app.get("/query") 4 ctx1abcd

14async def query_model(data: Model = Query()): 4 ctx1abcd

15 return {"param": data.param} 3 ctx1klm

16 

17 

18@app.get("/header") 4 ctx1abcd

19async def header_model(data: Model = Header()): 4 ctx1abcd

20 return {"param": data.param} 3 ctx1nop

21 

22 

23@app.get("/cookie") 4 ctx1abcd

24async def cookie_model(data: Model = Cookie()): 4 ctx1abcd

25 return {"param": data.param} 3 ctx1efg

26 

27 

28def test_query_model_with_alias(): 4 ctx1abcd

29 client = TestClient(app) 3 ctx1klm

30 response = client.get("/query", params={"param_alias": "value"}) 3 ctx1klm

31 assert response.status_code == 200, response.text 3 ctx1klm

32 assert response.json() == {"param": "value"} 3 ctx1klm

33 

34 

35def test_header_model_with_alias(): 4 ctx1abcd

36 client = TestClient(app) 3 ctx1nop

37 response = client.get("/header", headers={"param_alias": "value"}) 3 ctx1nop

38 assert response.status_code == 200, response.text 3 ctx1nop

39 assert response.json() == {"param": "value"} 3 ctx1nop

40 

41 

42def test_cookie_model_with_alias(): 4 ctx1abcd

43 client = TestClient(app) 3 ctx1efg

44 client.cookies.set("param_alias", "value") 3 ctx1efg

45 response = client.get("/cookie") 3 ctx1efg

46 assert response.status_code == 200, response.text 3 ctx1efg

47 assert response.json() == {"param": "value"} 3 ctx1efg

48 

49 

50def test_query_model_with_alias_by_name(): 4 ctx1abcd

51 client = TestClient(app) 3 ctx1qrs

52 response = client.get("/query", params={"param": "value"}) 3 ctx1qrs

53 assert response.status_code == 422, response.text 3 ctx1qrs

54 details = response.json() 3 ctx1qrs

55 assert details["detail"][0]["input"] == {"param": "value"} 3 ctx1qrs

56 

57 

58def test_header_model_with_alias_by_name(): 4 ctx1abcd

59 client = TestClient(app) 3 ctx1tuv

60 response = client.get("/header", headers={"param": "value"}) 3 ctx1tuv

61 assert response.status_code == 422, response.text 3 ctx1tuv

62 details = response.json() 3 ctx1tuv

63 assert details["detail"][0]["input"] == IsPartialDict({"param": "value"}) 3 ctx1tuv

64 

65 

66def test_cookie_model_with_alias_by_name(): 4 ctx1abcd

67 client = TestClient(app) 3 ctx1hij

68 client.cookies.set("param", "value") 3 ctx1hij

69 response = client.get("/cookie") 3 ctx1hij

70 assert response.status_code == 422, response.text 3 ctx1hij

71 details = response.json() 3 ctx1hij

72 assert details["detail"][0]["input"] == {"param": "value"} 3 ctx1hij