Coverage for microservice_websocket/app/blueprints/api/command.py: 85%

1from fastapi import APIRouter, Depends 

2from pydantic import BaseModel 

3 

4from ...services.jwt import jwt_required 

5from ...utils.command import handle_command 

6from ...utils.enums import CommandType 

7 

8command_router = APIRouter(prefix="/command") 

9 

10 

11class MqttCommandPayload(BaseModel): 

12 nodeID: int 

13 applicationID: str 

14 

15 

16@command_router.post("/{command_type}", dependencies=[Depends(jwt_required)]) 

17async def send_mqtt_command_route( 

18 command_type: CommandType, payload: MqttCommandPayload 

19): 

20 await handle_command(command_type, payload.applicationID, payload.nodeID) 

21 

22 return {"message": "Sent"}