Coverage for tests/test_request_param_model_by_alias.py: 100%

1from dirty_equals import IsPartialDict 7 ctx1abcdefg

2from fastapi import Cookie, FastAPI, Header, Query 7 ctx1abcdefg

3from fastapi._compat import PYDANTIC_V2 7 ctx1abcdefg

4from fastapi.testclient import TestClient 7 ctx1abcdefg

5from pydantic import BaseModel, Field 7 ctx1abcdefg

6 

7app = FastAPI() 7 ctx1abcdefg

8 

9 

10class Model(BaseModel): 7 ctx1abcdefg

11 param: str = Field(alias="param_alias") 7 ctx1abcdefg

12 

13 

14@app.get("/query") 7 ctx1abcdefg

15async def query_model(data: Model = Query()): 7 ctx1abcdefg

16 return {"param": data.param} 7 ctx1JKLMNOP

17 

18 

19@app.get("/header") 7 ctx1abcdefg

20async def header_model(data: Model = Header()): 7 ctx1abcdefg

21 return {"param": data.param} 7 ctx1QRSTUVW

22 

23 

24@app.get("/cookie") 7 ctx1abcdefg

25async def cookie_model(data: Model = Cookie()): 7 ctx1abcdefg

26 return {"param": data.param} 7 ctx1opqrstu

27 

28 

29def test_query_model_with_alias(): 7 ctx1abcdefg

30 client = TestClient(app) 7 ctx1JKLMNOP

31 response = client.get("/query", params={"param_alias": "value"}) 7 ctx1JKLMNOP

32 assert response.status_code == 200, response.text 7 ctx1JKLMNOP

33 assert response.json() == {"param": "value"} 7 ctx1JKLMNOP

34 

35 

36def test_header_model_with_alias(): 7 ctx1abcdefg

37 client = TestClient(app) 7 ctx1QRSTUVW

38 response = client.get("/header", headers={"param_alias": "value"}) 7 ctx1QRSTUVW

39 assert response.status_code == 200, response.text 7 ctx1QRSTUVW

40 assert response.json() == {"param": "value"} 7 ctx1QRSTUVW

41 

42 

43def test_cookie_model_with_alias(): 7 ctx1abcdefg

44 client = TestClient(app) 7 ctx1opqrstu

45 client.cookies.set("param_alias", "value") 7 ctx1opqrstu

46 response = client.get("/cookie") 7 ctx1opqrstu

47 assert response.status_code == 200, response.text 7 ctx1opqrstu

48 assert response.json() == {"param": "value"} 7 ctx1opqrstu

49 

50 

51def test_query_model_with_alias_by_name(): 7 ctx1abcdefg

52 client = TestClient(app) 7 ctx1vwxyzAB

53 response = client.get("/query", params={"param": "value"}) 7 ctx1vwxyzAB

54 assert response.status_code == 422, response.text 7 ctx1vwxyzAB

55 details = response.json() 7 ctx1vwxyzAB

56 if PYDANTIC_V2: 7 ctx1vwxyzAB

57 assert details["detail"][0]["input"] == {"param": "value"} 7 ctx1vwxyzAB

58 

59 

60def test_header_model_with_alias_by_name(): 7 ctx1abcdefg

61 client = TestClient(app) 7 ctx1CDEFGHI

62 response = client.get("/header", headers={"param": "value"}) 7 ctx1CDEFGHI

63 assert response.status_code == 422, response.text 7 ctx1CDEFGHI

64 details = response.json() 7 ctx1CDEFGHI

65 if PYDANTIC_V2: 7 ctx1CDEFGHI

66 assert details["detail"][0]["input"] == IsPartialDict({"param": "value"}) 7 ctx1CDEFGHI

67 

68 

69def test_cookie_model_with_alias_by_name(): 7 ctx1abcdefg

70 client = TestClient(app) 7 ctx1hijklmn

71 client.cookies.set("param", "value") 7 ctx1hijklmn

72 response = client.get("/cookie") 7 ctx1hijklmn

73 assert response.status_code == 422, response.text 7 ctx1hijklmn

74 details = response.json() 7 ctx1hijklmn

75 if PYDANTIC_V2: 7 ctx1hijklmn

76 assert details["detail"][0]["input"] == {"param": "value"} 7 ctx1hijklmn