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} 4 ctx1mnop

16 

17 

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

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

20 return {"param": data.param} 4 ctx1qrst

21 

22 

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

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

25 return {"param": data.param} 4 ctx1efgh

26 

27 

28def test_query_model_with_alias(): 4 ctx1abcd

29 client = TestClient(app) 4 ctx1mnop

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

31 assert response.status_code == 200, response.text 4 ctx1mnop

32 assert response.json() == {"param": "value"} 4 ctx1mnop

33 

34 

35def test_header_model_with_alias(): 4 ctx1abcd

36 client = TestClient(app) 4 ctx1qrst

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

38 assert response.status_code == 200, response.text 4 ctx1qrst

39 assert response.json() == {"param": "value"} 4 ctx1qrst

40 

41 

42def test_cookie_model_with_alias(): 4 ctx1abcd

43 client = TestClient(app) 4 ctx1efgh

44 client.cookies.set("param_alias", "value") 4 ctx1efgh

45 response = client.get("/cookie") 4 ctx1efgh

46 assert response.status_code == 200, response.text 4 ctx1efgh

47 assert response.json() == {"param": "value"} 4 ctx1efgh

48 

49 

50def test_query_model_with_alias_by_name(): 4 ctx1abcd

51 client = TestClient(app) 4 ctx1uvwx

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

53 assert response.status_code == 422, response.text 4 ctx1uvwx

54 details = response.json() 4 ctx1uvwx

55 assert details["detail"][0]["input"] == {"param": "value"} 4 ctx1uvwx

56 

57 

58def test_header_model_with_alias_by_name(): 4 ctx1abcd

59 client = TestClient(app) 4 ctx1yzAB

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

61 assert response.status_code == 422, response.text 4 ctx1yzAB

62 details = response.json() 4 ctx1yzAB

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

64 

65 

66def test_cookie_model_with_alias_by_name(): 4 ctx1abcd

67 client = TestClient(app) 4 ctx1ijkl

68 client.cookies.set("param", "value") 4 ctx1ijkl

69 response = client.get("/cookie") 4 ctx1ijkl

70 assert response.status_code == 422, response.text 4 ctx1ijkl

71 details = response.json() 4 ctx1ijkl

72 assert details["detail"][0]["input"] == {"param": "value"} 4 ctx1ijkl