Coverage for tests / test_router_redirect_slashes.py: 100%

1from fastapi import APIRouter, FastAPI 4 ctx1ghij

2from fastapi.testclient import TestClient 4 ctx1ghij

3 

4 

5def test_redirect_slashes_enabled(): 4 ctx1ghij

6 app = FastAPI() 3 ctx1abc

7 router = APIRouter() 3 ctx1abc

8 

9 @router.get("/hello/") 3 ctx1abc

10 def hello_page() -> str: 3 ctx1abc

11 return "Hello, World!" 3 ctx1abc

12 

13 app.include_router(router) 3 ctx1abc

14 

15 client = TestClient(app) 3 ctx1abc

16 

17 response = client.get("/hello/", follow_redirects=False) 3 ctx1abc

18 assert response.status_code == 200 3 ctx1abc

19 

20 response = client.get("/hello", follow_redirects=False) 3 ctx1abc

21 assert response.status_code == 307 3 ctx1abc

22 

23 

24def test_redirect_slashes_disabled(): 4 ctx1ghij

25 app = FastAPI(redirect_slashes=False) 3 ctx1def

26 router = APIRouter() 3 ctx1def

27 

28 @router.get("/hello/") 3 ctx1def

29 def hello_page() -> str: 3 ctx1def

30 return "Hello, World!" 3 ctx1def

31 

32 app.include_router(router) 3 ctx1def

33 

34 client = TestClient(app) 3 ctx1def

35 

36 response = client.get("/hello/", follow_redirects=False) 3 ctx1def

37 assert response.status_code == 200 3 ctx1def

38 

39 response = client.get("/hello", follow_redirects=False) 3 ctx1def

40 assert response.status_code == 404 3 ctx1def