Coverage for tests/test_route_scope.py: 100%

1import pytest 5 ctx1abcde

2from fastapi import FastAPI, Request, WebSocket, WebSocketDisconnect 5 ctx1abcde

3from fastapi.routing import APIRoute, APIWebSocketRoute 5 ctx1abcde

4from fastapi.testclient import TestClient 5 ctx1abcde

5 

6app = FastAPI() 5 ctx1abcde

7 

8 

9@app.get("/users/{user_id}") 5 ctx1abcde

10async def get_user(user_id: str, request: Request): 5 ctx1abcde

11 route: APIRoute = request.scope["route"] 5 ctx1klmno

12 return {"user_id": user_id, "path": route.path} 5 ctx1klmno

13 

14 

15@app.websocket("/items/{item_id}") 5 ctx1abcde

16async def websocket_item(item_id: str, websocket: WebSocket): 5 ctx1abcde

17 route: APIWebSocketRoute = websocket.scope["route"] 5 ctx1fghij

18 await websocket.accept() 5 ctx1fghij

19 await websocket.send_json({"item_id": item_id, "path": route.path}) 5 ctx1fghij

20 

21 

22client = TestClient(app) 5 ctx1abcde

23 

24 

25def test_get(): 5 ctx1abcde

26 response = client.get("/users/rick") 5 ctx1klmno

27 assert response.status_code == 200, response.text 5 ctx1klmno

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

29 

30 

31def test_invalid_method_doesnt_match(): 5 ctx1abcde

32 response = client.post("/users/rick") 5 ctx1qrstu

33 assert response.status_code == 405, response.text 5 ctx1qrstu

34 

35 

36def test_invalid_path_doesnt_match(): 5 ctx1abcde

37 response = client.post("/usersx/rick") 5 ctx1vwxyz

38 assert response.status_code == 404, response.text 5 ctx1vwxyz

39 

40 

41def test_websocket(): 5 ctx1abcde

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

43 data = websocket.receive_json() 5 ctx1fghij

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

45 

46 

47def test_websocket_invalid_path_doesnt_match(): 5 ctx1abcde

48 with pytest.raises(WebSocketDisconnect): 5 ctx1ApBCD

49 with client.websocket_connect("/itemsx/portal-gun"): 5 ctx1ApBCD

50 pass 1 ctx1p