Coverage for tests/test_router_redirect_slashes.py: 100%

1from fastapi import APIRouter, FastAPI 7 ctx1opqrstu

2from fastapi.testclient import TestClient 7 ctx1opqrstu

3 

4 

5def test_redirect_slashes_enabled(): 7 ctx1opqrstu

6 app = FastAPI() 7 ctx1abcdefg

7 router = APIRouter() 7 ctx1abcdefg

8 

9 @router.get("/hello/") 7 ctx1abcdefg

10 def hello_page() -> str: 7 ctx1abcdefg

11 return "Hello, World!" 7 ctx1abcdefg

12 

13 app.include_router(router) 7 ctx1abcdefg

14 

15 client = TestClient(app) 7 ctx1abcdefg

16 

17 response = client.get("/hello/", follow_redirects=False) 7 ctx1abcdefg

18 assert response.status_code == 200 7 ctx1abcdefg

19 

20 response = client.get("/hello", follow_redirects=False) 7 ctx1abcdefg

21 assert response.status_code == 307 7 ctx1abcdefg

22 

23 

24def test_redirect_slashes_disabled(): 7 ctx1opqrstu

25 app = FastAPI(redirect_slashes=False) 7 ctx1hijklmn

26 router = APIRouter() 7 ctx1hijklmn

27 

28 @router.get("/hello/") 7 ctx1hijklmn

29 def hello_page() -> str: 7 ctx1hijklmn

30 return "Hello, World!" 7 ctx1hijklmn

31 

32 app.include_router(router) 7 ctx1hijklmn

33 

34 client = TestClient(app) 7 ctx1hijklmn

35 

36 response = client.get("/hello/", follow_redirects=False) 7 ctx1hijklmn

37 assert response.status_code == 200 7 ctx1hijklmn

38 

39 response = client.get("/hello", follow_redirects=False) 7 ctx1hijklmn

40 assert response.status_code == 404 7 ctx1hijklmn