Coverage for tests/test_repeated_cookie_headers.py: 100%

1from fastapi import Depends, FastAPI, Response 7 ctx1abcdefg

2from fastapi.testclient import TestClient 7 ctx1abcdefg

3 

4app = FastAPI() 7 ctx1abcdefg

5 

6 

7def set_cookie(*, response: Response): 7 ctx1abcdefg

8 response.set_cookie("cookie-name", "cookie-value") 7 ctx1hijklmn

9 return {} 7 ctx1hijklmn

10 

11 

12def set_indirect_cookie(*, dep: str = Depends(set_cookie)): 7 ctx1abcdefg

13 return dep 7 ctx1hijklmn

14 

15 

16@app.get("/directCookie") 7 ctx1abcdefg

17def get_direct_cookie(dep: str = Depends(set_cookie)): 7 ctx1abcdefg

18 return {"dep": dep} 7 ctx1hijklmn

19 

20 

21@app.get("/indirectCookie") 7 ctx1abcdefg

22def get_indirect_cookie(dep: str = Depends(set_indirect_cookie)): 7 ctx1abcdefg

23 return {"dep": dep} 7 ctx1hijklmn

24 

25 

26client = TestClient(app) 7 ctx1abcdefg

27 

28 

29def test_cookie_is_set_once(): 7 ctx1abcdefg

30 direct_response = client.get("/directCookie") 7 ctx1hijklmn

31 indirect_response = client.get("/indirectCookie") 7 ctx1hijklmn

32 assert ( 7 ctx1hijklmn

33 direct_response.headers["set-cookie"] == indirect_response.headers["set-cookie"] 

34 )