Coverage for tests/test_query_cookie_header_model_extra_params.py: 100%

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

2from fastapi._compat import PYDANTIC_V2 7 ctx1abcdefg

3from fastapi.testclient import TestClient 7 ctx1abcdefg

4from pydantic import BaseModel 7 ctx1abcdefg

5 

6app = FastAPI() 7 ctx1abcdefg

7 

8 

9class Model(BaseModel): 7 ctx1abcdefg

10 param: str 7 ctx1abcdefg

11 

12 if PYDANTIC_V2: 7 ctx1abcdefg

13 model_config = {"extra": "allow"} 7 ctx1abcdefg

14 else: 

15 

16 class Config: 6 ctx1abcdef

17 extra = "allow" 6 ctx1abcdef

18 

19 

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

21async def query_model_with_extra(data: Model = Query()): 7 ctx1abcdefg

22 return data 14 ctx1CDEFGHIJKLMNOP

23 

24 

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

26async def header_model_with_extra(data: Model = Header()): 7 ctx1abcdefg

27 return data 14 ctx1opqrstuvwxyzAB

28 

29 

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

31async def cookies_model_with_extra(data: Model = Cookie()): 7 ctx1abcdefg

32 return data 7 ctx1hijklmn

33 

34 

35def test_query_pass_extra_list(): 7 ctx1abcdefg

36 client = TestClient(app) 7 ctx1CEGIKMO

37 resp = client.get( 7 ctx1CEGIKMO

38 "/query", 

39 params={ 

40 "param": "123", 

41 "param2": ["456", "789"], # Pass a list of values as extra parameter 

42 }, 

43 ) 

44 assert resp.status_code == 200 7 ctx1CEGIKMO

45 assert resp.json() == { 7 ctx1CEGIKMO

46 "param": "123", 

47 "param2": ["456", "789"], 

48 } 

49 

50 

51def test_query_pass_extra_single(): 7 ctx1abcdefg

52 client = TestClient(app) 7 ctx1DFHJLNP

53 resp = client.get( 7 ctx1DFHJLNP

54 "/query", 

55 params={ 

56 "param": "123", 

57 "param2": "456", 

58 }, 

59 ) 

60 assert resp.status_code == 200 7 ctx1DFHJLNP

61 assert resp.json() == { 7 ctx1DFHJLNP

62 "param": "123", 

63 "param2": "456", 

64 } 

65 

66 

67def test_header_pass_extra_list(): 7 ctx1abcdefg

68 client = TestClient(app) 7 ctx1oqsuwyA

69 

70 resp = client.get( 7 ctx1oqsuwyA

71 "/header", 

72 headers=[ 

73 ("param", "123"), 

74 ("param2", "456"), # Pass a list of values as extra parameter 

75 ("param2", "789"), 

76 ], 

77 ) 

78 assert resp.status_code == 200 7 ctx1oqsuwyA

79 resp_json = resp.json() 7 ctx1oqsuwyA

80 assert "param2" in resp_json 7 ctx1oqsuwyA

81 assert resp_json["param2"] == ["456", "789"] 7 ctx1oqsuwyA

82 

83 

84def test_header_pass_extra_single(): 7 ctx1abcdefg

85 client = TestClient(app) 7 ctx1prtvxzB

86 

87 resp = client.get( 7 ctx1prtvxzB

88 "/header", 

89 headers=[ 

90 ("param", "123"), 

91 ("param2", "456"), 

92 ], 

93 ) 

94 assert resp.status_code == 200 7 ctx1prtvxzB

95 resp_json = resp.json() 7 ctx1prtvxzB

96 assert "param2" in resp_json 7 ctx1prtvxzB

97 assert resp_json["param2"] == "456" 7 ctx1prtvxzB

98 

99 

100def test_cookie_pass_extra_list(): 7 ctx1abcdefg

101 client = TestClient(app) 7 ctx1hijklmn

102 client.cookies = [ 7 ctx1hijklmn

103 ("param", "123"), 

104 ("param2", "456"), # Pass a list of values as extra parameter 

105 ("param2", "789"), 

106 ] 

107 resp = client.get("/cookie") 7 ctx1hijklmn

108 assert resp.status_code == 200 7 ctx1hijklmn

109 resp_json = resp.json() 7 ctx1hijklmn

110 assert "param2" in resp_json 7 ctx1hijklmn

111 assert resp_json["param2"] == "789" # Cookies only keep the last value 7 ctx1hijklmn