Coverage for tests/test_router_redirect_slashes.py: 100%

1from fastapi import APIRouter, FastAPI 5 ctx1abcde

2from fastapi.testclient import TestClient 5 ctx1abcde

3 

4 

5def test_redirect_slashes_enabled(): 5 ctx1abcde

6 app = FastAPI() 5 ctx1abcde

7 router = APIRouter() 5 ctx1abcde

8 

9 @router.get("/hello/") 5 ctx1abcde

10 def hello_page() -> str: 5 ctx1abcde

11 return "Hello, World!" 5 ctx1abcde

12 

13 app.include_router(router) 5 ctx1abcde

14 

15 client = TestClient(app) 5 ctx1abcde

16 

17 response = client.get("/hello/", follow_redirects=False) 5 ctx1abcde

18 assert response.status_code == 200 5 ctx1abcde

19 

20 response = client.get("/hello", follow_redirects=False) 5 ctx1abcde

21 assert response.status_code == 307 5 ctx1abcde

22 

23 

24def test_redirect_slashes_disabled(): 5 ctx1abcde

25 app = FastAPI(redirect_slashes=False) 5 ctx1abcde

26 router = APIRouter() 5 ctx1abcde

27 

28 @router.get("/hello/") 5 ctx1abcde

29 def hello_page() -> str: 5 ctx1abcde

30 return "Hello, World!" 5 ctx1abcde

31 

32 app.include_router(router) 5 ctx1abcde

33 

34 client = TestClient(app) 5 ctx1abcde

35 

36 response = client.get("/hello/", follow_redirects=False) 5 ctx1abcde

37 assert response.status_code == 200 5 ctx1abcde

38 

39 response = client.get("/hello", follow_redirects=False) 5 ctx1abcde

40 assert response.status_code == 404 5 ctx1abcde