Coverage for tests/test_repeated_cookie_headers.py: 100%

1from fastapi import Depends, FastAPI, Response 5 ctx1abcde

2from fastapi.testclient import TestClient 5 ctx1abcde

3 

4app = FastAPI() 5 ctx1abcde

5 

6 

7def set_cookie(*, response: Response): 5 ctx1abcde

8 response.set_cookie("cookie-name", "cookie-value") 5 ctx1abcde

9 return {} 5 ctx1abcde

10 

11 

12def set_indirect_cookie(*, dep: str = Depends(set_cookie)): 5 ctx1abcde

13 return dep 5 ctx1abcde

14 

15 

16@app.get("/directCookie") 5 ctx1abcde

17def get_direct_cookie(dep: str = Depends(set_cookie)): 5 ctx1abcde

18 return {"dep": dep} 5 ctx1abcde

19 

20 

21@app.get("/indirectCookie") 5 ctx1abcde

22def get_indirect_cookie(dep: str = Depends(set_indirect_cookie)): 5 ctx1abcde

23 return {"dep": dep} 5 ctx1abcde

24 

25 

26client = TestClient(app) 5 ctx1abcde

27 

28 

29def test_cookie_is_set_once(): 5 ctx1abcde

30 direct_response = client.get("/directCookie") 5 ctx1abcde

31 indirect_response = client.get("/indirectCookie") 5 ctx1abcde

32 assert ( 5 ctx1abcde

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

34 )