Coverage for microservice_websocket/app/blueprints/api/alert.py: 100%

1from fastapi import APIRouter, Depends 

2 

3from ...services.database import Alert, User 

4from ...services.jwt import get_user_from_jwt 

5from ...utils.alert import get_alert, handle_alert 

6from .models import HandlePayload 

7 

8alert_router = APIRouter(prefix="/alert") 

9 

10 

11@alert_router.post("/{alertID}") 

12async def handle_alert_route( 

13 alertID: str, payload: HandlePayload, user: User = Depends(get_user_from_jwt) 

14): 

15 from ... import socketManager 

16 

17 await handle_alert(alertID, payload, user) 

18 

19 await socketManager.emit("change") 

20 await socketManager.emit("change-reading") 

21 

22 return {"message": "Handled"} 

23 

24 

25@alert_router.get("/{alertID}", response_model=Alert.Serialized) 

26async def get_alert_route(alertID: str): 

27 response: Alert = await get_alert(alertID) 

28 

29 return await response.serialize()