Coverage for tests / test_route_scope.py: 100%

1import pytest 4 ctx1abcd

2from fastapi import FastAPI, Request, WebSocket, WebSocketDisconnect 4 ctx1abcd

3from fastapi.routing import APIRoute, APIWebSocketRoute 4 ctx1abcd

4from fastapi.testclient import TestClient 4 ctx1abcd

5 

6app = FastAPI() 4 ctx1abcd

7 

8 

9@app.get("/users/{user_id}") 4 ctx1abcd

10async def get_user(user_id: str, request: Request): 4 ctx1abcd

11 route: APIRoute = request.scope["route"] 3 ctx1hij

12 return {"user_id": user_id, "path": route.path} 3 ctx1hij

13 

14 

15@app.websocket("/items/{item_id}") 4 ctx1abcd

16async def websocket_item(item_id: str, websocket: WebSocket): 4 ctx1abcd

17 route: APIWebSocketRoute = websocket.scope["route"] 3 ctx1efg

18 await websocket.accept() 3 ctx1efg

19 await websocket.send_json({"item_id": item_id, "path": route.path}) 3 ctx1efg

20 

21 

22client = TestClient(app) 4 ctx1abcd

23 

24 

25def test_get(): 4 ctx1abcd

26 response = client.get("/users/rick") 3 ctx1hij

27 assert response.status_code == 200, response.text 3 ctx1hij

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

29 

30 

31def test_invalid_method_doesnt_match(): 4 ctx1abcd

32 response = client.post("/users/rick") 3 ctx1klm

33 assert response.status_code == 405, response.text 3 ctx1klm

34 

35 

36def test_invalid_path_doesnt_match(): 4 ctx1abcd

37 response = client.post("/usersx/rick") 3 ctx1nop

38 assert response.status_code == 404, response.text 3 ctx1nop

39 

40 

41def test_websocket(): 4 ctx1abcd

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

43 data = websocket.receive_json() 3 ctx1efg

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

45 

46 

47def test_websocket_invalid_path_doesnt_match(): 4 ctx1abcd

48 with pytest.raises(WebSocketDisconnect): 3 ctx1qrs

49 with client.websocket_connect("/itemsx/portal-gun"): 3 ctx1qrs

50 pass # pragma: no cover