Coverage for tests / test_query_cookie_header_model_extra_params.py: 100%

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

2from fastapi.testclient import TestClient 4 ctx1abcd

3from pydantic import BaseModel 4 ctx1abcd

4 

5app = FastAPI() 4 ctx1abcd

6 

7 

8class Model(BaseModel): 4 ctx1abcd

9 param: str 4 ctx1abcd

10 

11 model_config = {"extra": "allow"} 4 ctx1abcd

12 

13 

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

15async def query_model_with_extra(data: Model = Query()): 4 ctx1abcd

16 return data 6 ctx1nopqrs

17 

18 

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

20async def header_model_with_extra(data: Model = Header()): 4 ctx1abcd

21 return data 6 ctx1hijklm

22 

23 

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

25async def cookies_model_with_extra(data: Model = Cookie()): 4 ctx1abcd

26 return data 3 ctx1efg

27 

28 

29def test_query_pass_extra_list(): 4 ctx1abcd

30 client = TestClient(app) 3 ctx1npr

31 resp = client.get( 3 ctx1npr

32 "/query", 

33 params={ 

34 "param": "123", 

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

36 }, 

37 ) 

38 assert resp.status_code == 200 3 ctx1npr

39 assert resp.json() == { 3 ctx1npr

40 "param": "123", 

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

42 } 

43 

44 

45def test_query_pass_extra_single(): 4 ctx1abcd

46 client = TestClient(app) 3 ctx1oqs

47 resp = client.get( 3 ctx1oqs

48 "/query", 

49 params={ 

50 "param": "123", 

51 "param2": "456", 

52 }, 

53 ) 

54 assert resp.status_code == 200 3 ctx1oqs

55 assert resp.json() == { 3 ctx1oqs

56 "param": "123", 

57 "param2": "456", 

58 } 

59 

60 

61def test_header_pass_extra_list(): 4 ctx1abcd

62 client = TestClient(app) 3 ctx1hjl

63 

64 resp = client.get( 3 ctx1hjl

65 "/header", 

66 headers=[ 

67 ("param", "123"), 

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

69 ("param2", "789"), 

70 ], 

71 ) 

72 assert resp.status_code == 200 3 ctx1hjl

73 resp_json = resp.json() 3 ctx1hjl

74 assert "param2" in resp_json 3 ctx1hjl

75 assert resp_json["param2"] == ["456", "789"] 3 ctx1hjl

76 

77 

78def test_header_pass_extra_single(): 4 ctx1abcd

79 client = TestClient(app) 3 ctx1ikm

80 

81 resp = client.get( 3 ctx1ikm

82 "/header", 

83 headers=[ 

84 ("param", "123"), 

85 ("param2", "456"), 

86 ], 

87 ) 

88 assert resp.status_code == 200 3 ctx1ikm

89 resp_json = resp.json() 3 ctx1ikm

90 assert "param2" in resp_json 3 ctx1ikm

91 assert resp_json["param2"] == "456" 3 ctx1ikm

92 

93 

94def test_cookie_pass_extra_list(): 4 ctx1abcd

95 client = TestClient(app) 3 ctx1efg

96 client.cookies = [ 3 ctx1efg

97 ("param", "123"), 

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

99 ("param2", "789"), 

100 ] 

101 resp = client.get("/cookie") 3 ctx1efg

102 assert resp.status_code == 200 3 ctx1efg

103 resp_json = resp.json() 3 ctx1efg

104 assert "param2" in resp_json 3 ctx1efg

105 assert resp_json["param2"] == "789" # Cookies only keep the last value 3 ctx1efg