Coverage for tests/test_router_redirect_slashes.py: 100%

1from fastapi import APIRouter, FastAPI 6 ctx1mnopqr

2from fastapi.testclient import TestClient 6 ctx1mnopqr

3 

4 

5def test_redirect_slashes_enabled(): 6 ctx1mnopqr

6 app = FastAPI() 6 ctx1abcdef

7 router = APIRouter() 6 ctx1abcdef

8 

9 @router.get("/hello/") 6 ctx1abcdef

10 def hello_page() -> str: 6 ctx1abcdef

11 return "Hello, World!" 6 ctx1abcdef

12 

13 app.include_router(router) 6 ctx1abcdef

14 

15 client = TestClient(app) 6 ctx1abcdef

16 

17 response = client.get("/hello/", follow_redirects=False) 6 ctx1abcdef

18 assert response.status_code == 200 6 ctx1abcdef

19 

20 response = client.get("/hello", follow_redirects=False) 6 ctx1abcdef

21 assert response.status_code == 307 6 ctx1abcdef

22 

23 

24def test_redirect_slashes_disabled(): 6 ctx1mnopqr

25 app = FastAPI(redirect_slashes=False) 6 ctx1ghijkl

26 router = APIRouter() 6 ctx1ghijkl

27 

28 @router.get("/hello/") 6 ctx1ghijkl

29 def hello_page() -> str: 6 ctx1ghijkl

30 return "Hello, World!" 6 ctx1ghijkl

31 

32 app.include_router(router) 6 ctx1ghijkl

33 

34 client = TestClient(app) 6 ctx1ghijkl

35 

36 response = client.get("/hello/", follow_redirects=False) 6 ctx1ghijkl

37 assert response.status_code == 200 6 ctx1ghijkl

38 

39 response = client.get("/hello", follow_redirects=False) 6 ctx1ghijkl

40 assert response.status_code == 404 6 ctx1ghijkl