Coverage for tests/test_route_scope.py: 100%

1import pytest 5 ctx1bacde

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

3from fastapi.routing import APIRoute, APIWebSocketRoute 5 ctx1bacde

4from fastapi.testclient import TestClient 5 ctx1bacde

5 

6app = FastAPI() 5 ctx1bacde

7 

8 

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

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

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

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

13 

14 

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

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

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

18 await websocket.accept() 5 ctx1bacde

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

20 

21 

22client = TestClient(app) 5 ctx1bacde

23 

24 

25def test_get(): 5 ctx1bacde

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

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

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

29 

30 

31def test_invalid_method_doesnt_match(): 5 ctx1bacde

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

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

34 

35 

36def test_invalid_path_doesnt_match(): 5 ctx1bacde

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

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

39 

40 

41def test_websocket(): 5 ctx1bacde

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

43 data = websocket.receive_json() 5 ctx1bacde

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

45 

46 

47def test_websocket_invalid_path_doesnt_match(): 5 ctx1bacde

48 with pytest.raises(WebSocketDisconnect): 5 ctx1bacde

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

50 pass 1 ctx1a