Coverage for tests / test_query_cookie_header_model_extra_params.py: 100%

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

2from fastapi.testclient import TestClient 4 ctx1adbc

3from pydantic import BaseModel 4 ctx1adbc

4 

5app = FastAPI() 4 ctx1adbc

6 

7 

8class Model(BaseModel): 4 ctx1adbc

9 param: str 3 ctx1abc

10 

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

12 

13 

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

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

16 return data 8 ctx1qrstuvwx

17 

18 

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

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

21 return data 8 ctx1ijklmnop

22 

23 

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

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

26 return data 4 ctx1efgh

27 

28 

29def test_query_pass_extra_list(): 4 ctx1adbc

30 client = TestClient(app) 4 ctx1qsuw

31 resp = client.get( 4 ctx1qsuw

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 4 ctx1qsuw

39 assert resp.json() == { 4 ctx1qsuw

40 "param": "123", 

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

42 } 

43 

44 

45def test_query_pass_extra_single(): 4 ctx1adbc

46 client = TestClient(app) 4 ctx1rtvx

47 resp = client.get( 4 ctx1rtvx

48 "/query", 

49 params={ 

50 "param": "123", 

51 "param2": "456", 

52 }, 

53 ) 

54 assert resp.status_code == 200 4 ctx1rtvx

55 assert resp.json() == { 4 ctx1rtvx

56 "param": "123", 

57 "param2": "456", 

58 } 

59 

60 

61def test_header_pass_extra_list(): 4 ctx1adbc

62 client = TestClient(app) 4 ctx1ikmo

63 

64 resp = client.get( 4 ctx1ikmo

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 4 ctx1ikmo

73 resp_json = resp.json() 4 ctx1ikmo

74 assert "param2" in resp_json 4 ctx1ikmo

75 assert resp_json["param2"] == ["456", "789"] 4 ctx1ikmo

76 

77 

78def test_header_pass_extra_single(): 4 ctx1adbc

79 client = TestClient(app) 4 ctx1jlnp

80 

81 resp = client.get( 4 ctx1jlnp

82 "/header", 

83 headers=[ 

84 ("param", "123"), 

85 ("param2", "456"), 

86 ], 

87 ) 

88 assert resp.status_code == 200 4 ctx1jlnp

89 resp_json = resp.json() 4 ctx1jlnp

90 assert "param2" in resp_json 4 ctx1jlnp

91 assert resp_json["param2"] == "456" 4 ctx1jlnp

92 

93 

94def test_cookie_pass_extra_list(): 4 ctx1adbc

95 client = TestClient(app) 4 ctx1efgh

96 client.cookies = [ 4 ctx1efgh

97 ("param", "123"), 

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

99 ("param2", "789"), 

100 ] 

101 resp = client.get("/cookie") 4 ctx1efgh

102 assert resp.status_code == 200 4 ctx1efgh

103 resp_json = resp.json() 4 ctx1efgh

104 assert "param2" in resp_json 4 ctx1efgh

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