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"] 4 ctx1ijkl

12 return {"user_id": user_id, "path": route.path} 4 ctx1ijkl

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"] 4 ctx1efgh

18 await websocket.accept() 4 ctx1efgh

19 await websocket.send_json({"item_id": item_id, "path": route.path}) 4 ctx1efgh

20 

21 

22client = TestClient(app) 4 ctx1abcd

23 

24 

25def test_get(): 4 ctx1abcd

26 response = client.get("/users/rick") 4 ctx1ijkl

27 assert response.status_code == 200, response.text 4 ctx1ijkl

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

29 

30 

31def test_invalid_method_doesnt_match(): 4 ctx1abcd

32 response = client.post("/users/rick") 4 ctx1mnop

33 assert response.status_code == 405, response.text 4 ctx1mnop

34 

35 

36def test_invalid_path_doesnt_match(): 4 ctx1abcd

37 response = client.post("/usersx/rick") 4 ctx1qrst

38 assert response.status_code == 404, response.text 4 ctx1qrst

39 

40 

41def test_websocket(): 4 ctx1abcd

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

43 data = websocket.receive_json() 4 ctx1efgh

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

45 

46 

47def test_websocket_invalid_path_doesnt_match(): 4 ctx1abcd

48 with pytest.raises(WebSocketDisconnect): 4 ctx1uvwx

49 with client.websocket_connect("/itemsx/portal-gun"): 4 ctx1uvwx

50 pass # pragma: no cover