Coverage for tests/test_route_scope.py: 100%

1import pytest 6 ctx1abcdef

2from fastapi import FastAPI, Request, WebSocket, WebSocketDisconnect 6 ctx1abcdef

3from fastapi.routing import APIRoute, APIWebSocketRoute 6 ctx1abcdef

4from fastapi.testclient import TestClient 6 ctx1abcdef

5 

6app = FastAPI() 6 ctx1abcdef

7 

8 

9@app.get("/users/{user_id}") 6 ctx1abcdef

10async def get_user(user_id: str, request: Request): 6 ctx1abcdef

11 route: APIRoute = request.scope["route"] 6 ctx1mnopqr

12 return {"user_id": user_id, "path": route.path} 6 ctx1mnopqr

13 

14 

15@app.websocket("/items/{item_id}") 6 ctx1abcdef

16async def websocket_item(item_id: str, websocket: WebSocket): 6 ctx1abcdef

17 route: APIWebSocketRoute = websocket.scope["route"] 6 ctx1ghijkl

18 await websocket.accept() 6 ctx1ghijkl

19 await websocket.send_json({"item_id": item_id, "path": route.path}) 6 ctx1ghijkl

20 

21 

22client = TestClient(app) 6 ctx1abcdef

23 

24 

25def test_get(): 6 ctx1abcdef

26 response = client.get("/users/rick") 6 ctx1mnopqr

27 assert response.status_code == 200, response.text 6 ctx1mnopqr

28 assert response.json() == {"user_id": "rick", "path": "/users/{user_id}"} 6 ctx1mnopqr

29 

30 

31def test_invalid_method_doesnt_match(): 6 ctx1abcdef

32 response = client.post("/users/rick") 6 ctx1tuvwxy

33 assert response.status_code == 405, response.text 6 ctx1tuvwxy

34 

35 

36def test_invalid_path_doesnt_match(): 6 ctx1abcdef

37 response = client.post("/usersx/rick") 6 ctx1zABCDE

38 assert response.status_code == 404, response.text 6 ctx1zABCDE

39 

40 

41def test_websocket(): 6 ctx1abcdef

42 with client.websocket_connect("/items/portal-gun") as websocket: 6 ctx1ghijkl

43 data = websocket.receive_json() 6 ctx1ghijkl

44 assert data == {"item_id": "portal-gun", "path": "/items/{item_id}"} 6 ctx1ghijkl

45 

46 

47def test_websocket_invalid_path_doesnt_match(): 6 ctx1abcdef

48 with pytest.raises(WebSocketDisconnect): 6 ctx1FsGHIJ

49 with client.websocket_connect("/itemsx/portal-gun"): 6 ctx1FsGHIJ

50 pass 1 ctx1s