Coverage for tests/test_router_redirect_slashes.py: 100%

1from fastapi import APIRouter, FastAPI 5 ctx1klmno

2from fastapi.testclient import TestClient 5 ctx1klmno

3 

4 

5def test_redirect_slashes_enabled(): 5 ctx1klmno

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 ctx1klmno

25 app = FastAPI(redirect_slashes=False) 5 ctx1fghij

26 router = APIRouter() 5 ctx1fghij

27 

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

29 def hello_page() -> str: 5 ctx1fghij

30 return "Hello, World!" 5 ctx1fghij

31 

32 app.include_router(router) 5 ctx1fghij

33 

34 client = TestClient(app) 5 ctx1fghij

35 

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

37 assert response.status_code == 200 5 ctx1fghij

38 

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

40 assert response.status_code == 404 5 ctx1fghij