Coverage for tests/test_custom_route_class.py: 100%

1import pytest 6 ctx1abcdef

2from fastapi import APIRouter, FastAPI 6 ctx1abcdef

3from fastapi.routing import APIRoute 6 ctx1abcdef

4from fastapi.testclient import TestClient 6 ctx1abcdef

5from starlette.routing import Route 6 ctx1abcdef

6 

7app = FastAPI() 6 ctx1abcdef

8 

9 

10class APIRouteA(APIRoute): 6 ctx1abcdef

11 x_type = "A" 6 ctx1abcdef

12 

13 

14class APIRouteB(APIRoute): 6 ctx1abcdef

15 x_type = "B" 6 ctx1abcdef

16 

17 

18class APIRouteC(APIRoute): 6 ctx1abcdef

19 x_type = "C" 6 ctx1abcdef

20 

21 

22router_a = APIRouter(route_class=APIRouteA) 6 ctx1abcdef

23router_b = APIRouter(route_class=APIRouteB) 6 ctx1abcdef

24router_c = APIRouter(route_class=APIRouteC) 6 ctx1abcdef

25 

26 

27@router_a.get("/") 6 ctx1abcdef

28def get_a(): 6 ctx1abcdef

29 return {"msg": "A"} 6 ctx1mnopqr

30 

31 

32@router_b.get("/") 6 ctx1abcdef

33def get_b(): 6 ctx1abcdef

34 return {"msg": "B"} 6 ctx1mnopqr

35 

36 

37@router_c.get("/") 6 ctx1abcdef

38def get_c(): 6 ctx1abcdef

39 return {"msg": "C"} 6 ctx1mnopqr

40 

41 

42router_b.include_router(router=router_c, prefix="/c") 6 ctx1abcdef

43router_a.include_router(router=router_b, prefix="/b") 6 ctx1abcdef

44app.include_router(router=router_a, prefix="/a") 6 ctx1abcdef

45 

46 

47client = TestClient(app) 6 ctx1abcdef

48 

49 

50@pytest.mark.parametrize( 6 ctx1abcdef

51 "path,expected_status,expected_response", 

52 [ 

53 ("/a", 200, {"msg": "A"}), 

54 ("/a/b", 200, {"msg": "B"}), 

55 ("/a/b/c", 200, {"msg": "C"}), 

56 ], 

57) 

58def test_get_path(path, expected_status, expected_response): 6 ctx1abcdef

59 response = client.get(path) 6 ctx1mnopqr

60 assert response.status_code == expected_status 6 ctx1mnopqr

61 assert response.json() == expected_response 6 ctx1mnopqr

62 

63 

64def test_route_classes(): 6 ctx1abcdef

65 routes = {} 6 ctx1ghijkl

66 for r in app.router.routes: 6 ctx1ghijkl

67 assert isinstance(r, Route) 6 ctx1ghijkl

68 routes[r.path] = r 6 ctx1ghijkl

69 assert getattr(routes["/a/"], "x_type") == "A" # noqa: B009 6 ctx1ghijkl

70 assert getattr(routes["/a/b/"], "x_type") == "B" # noqa: B009 6 ctx1ghijkl

71 assert getattr(routes["/a/b/c/"], "x_type") == "C" # noqa: B009 6 ctx1ghijkl

72 

73 

74def test_openapi_schema(): 6 ctx1abcdef

75 response = client.get("/openapi.json") 6 ctx1stuvwx

76 assert response.status_code == 200, response.text 6 ctx1stuvwx

77 assert response.json() == { 6 ctx1stuvwx

78 "openapi": "3.1.0", 

79 "info": {"title": "FastAPI", "version": "0.1.0"}, 

80 "paths": { 

81 "/a/": { 

82 "get": { 

83 "responses": { 

84 "200": { 

85 "description": "Successful Response", 

86 "content": {"application/json": {"schema": {}}}, 

87 } 

88 }, 

89 "summary": "Get A", 

90 "operationId": "get_a_a__get", 

91 } 

92 }, 

93 "/a/b/": { 

94 "get": { 

95 "responses": { 

96 "200": { 

97 "description": "Successful Response", 

98 "content": {"application/json": {"schema": {}}}, 

99 } 

100 }, 

101 "summary": "Get B", 

102 "operationId": "get_b_a_b__get", 

103 } 

104 }, 

105 "/a/b/c/": { 

106 "get": { 

107 "responses": { 

108 "200": { 

109 "description": "Successful Response", 

110 "content": {"application/json": {"schema": {}}}, 

111 } 

112 }, 

113 "summary": "Get C", 

114 "operationId": "get_c_a_b_c__get", 

115 } 

116 }, 

117 }, 

118 }