Coverage for tests / test_repeated_cookie_headers.py: 100%

1from fastapi import Depends, FastAPI, Response 4 ctx1abcd

2from fastapi.testclient import TestClient 4 ctx1abcd

3 

4app = FastAPI() 4 ctx1abcd

5 

6 

7def set_cookie(*, response: Response): 4 ctx1abcd

8 response.set_cookie("cookie-name", "cookie-value") 3 ctx1efg

9 return {} 3 ctx1efg

10 

11 

12def set_indirect_cookie(*, dep: str = Depends(set_cookie)): 4 ctx1abcd

13 return dep 3 ctx1efg

14 

15 

16@app.get("/directCookie") 4 ctx1abcd

17def get_direct_cookie(dep: str = Depends(set_cookie)): 4 ctx1abcd

18 return {"dep": dep} 3 ctx1efg

19 

20 

21@app.get("/indirectCookie") 4 ctx1abcd

22def get_indirect_cookie(dep: str = Depends(set_indirect_cookie)): 4 ctx1abcd

23 return {"dep": dep} 3 ctx1efg

24 

25 

26client = TestClient(app) 4 ctx1abcd

27 

28 

29def test_cookie_is_set_once(): 4 ctx1abcd

30 direct_response = client.get("/directCookie") 3 ctx1efg

31 indirect_response = client.get("/indirectCookie") 3 ctx1efg

32 assert ( 2 ctx1ef

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

34 )