Coverage for tests/test_repeated_cookie_headers.py: 100%

1from fastapi import Depends, FastAPI, Response 6 ctx1abcdef

2from fastapi.testclient import TestClient 6 ctx1abcdef

3 

4app = FastAPI() 6 ctx1abcdef

5 

6 

7def set_cookie(*, response: Response): 6 ctx1abcdef

8 response.set_cookie("cookie-name", "cookie-value") 6 ctx1ghijkl

9 return {} 6 ctx1ghijkl

10 

11 

12def set_indirect_cookie(*, dep: str = Depends(set_cookie)): 6 ctx1abcdef

13 return dep 6 ctx1ghijkl

14 

15 

16@app.get("/directCookie") 6 ctx1abcdef

17def get_direct_cookie(dep: str = Depends(set_cookie)): 6 ctx1abcdef

18 return {"dep": dep} 6 ctx1ghijkl

19 

20 

21@app.get("/indirectCookie") 6 ctx1abcdef

22def get_indirect_cookie(dep: str = Depends(set_indirect_cookie)): 6 ctx1abcdef

23 return {"dep": dep} 6 ctx1ghijkl

24 

25 

26client = TestClient(app) 6 ctx1abcdef

27 

28 

29def test_cookie_is_set_once(): 6 ctx1abcdef

30 direct_response = client.get("/directCookie") 6 ctx1ghijkl

31 indirect_response = client.get("/indirectCookie") 6 ctx1ghijkl

32 assert ( 6 ctx1ghijkl

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

34 )