Coverage for tests / test_router_redirect_slashes.py: 100%

1from fastapi import APIRouter, FastAPI 4 ctx1ijkl

2from fastapi.testclient import TestClient 4 ctx1ijkl

3 

4 

5def test_redirect_slashes_enabled(): 4 ctx1ijkl

6 app = FastAPI() 4 ctx1abcd

7 router = APIRouter() 4 ctx1abcd

8 

9 @router.get("/hello/") 4 ctx1abcd

10 def hello_page() -> str: 4 ctx1abcd

11 return "Hello, World!" 4 ctx1abcd

12 

13 app.include_router(router) 4 ctx1abcd

14 

15 client = TestClient(app) 4 ctx1abcd

16 

17 response = client.get("/hello/", follow_redirects=False) 4 ctx1abcd

18 assert response.status_code == 200 4 ctx1abcd

19 

20 response = client.get("/hello", follow_redirects=False) 4 ctx1abcd

21 assert response.status_code == 307 4 ctx1abcd

22 

23 

24def test_redirect_slashes_disabled(): 4 ctx1ijkl

25 app = FastAPI(redirect_slashes=False) 4 ctx1efgh

26 router = APIRouter() 4 ctx1efgh

27 

28 @router.get("/hello/") 4 ctx1efgh

29 def hello_page() -> str: 4 ctx1efgh

30 return "Hello, World!" 4 ctx1efgh

31 

32 app.include_router(router) 4 ctx1efgh

33 

34 client = TestClient(app) 4 ctx1efgh

35 

36 response = client.get("/hello/", follow_redirects=False) 4 ctx1efgh

37 assert response.status_code == 200 4 ctx1efgh

38 

39 response = client.get("/hello", follow_redirects=False) 4 ctx1efgh

40 assert response.status_code == 404 4 ctx1efgh