Coverage for yaptide/routes/task_routes.py: 30%

1from flask import request 

2from flask_restful import Resource 

3 

4from yaptide.persistence.db_methods import (fetch_simulation_by_sim_id, 

5 fetch_task_by_sim_id_and_task_id, 

6 update_task_state) 

7from yaptide.routes.utils.response_templates import yaptide_response 

8 

9 

10class TasksResource(Resource): 

11 """Class responsible for updating tasks""" 

12 

13 @staticmethod 

14 def post(): 

15 """ 

16 Method updating task state 

17 Structure required by this method to work properly: 

18 { 

19 "simulation_id": <int>, 

20 "task_id": <string>, 

21 "update_key": <string>, 

22 "update_dict": <dict> 

23 } 

24 simulation_id and task_id self explanatory 

25 """ 

26 payload_dict: dict = request.get_json(force=True) 

27 required_keys = {"simulation_id", "task_id", "update_key", "update_dict"} 

28 if required_keys != set(payload_dict.keys()): 

29 diff = required_keys.difference(set(payload_dict.keys())) 

30 return yaptide_response(message=f"Missing keys in JSON payload: {diff}", code=400) 

31 

32 sim_id: int = payload_dict["simulation_id"] 

33 simulation = fetch_simulation_by_sim_id(sim_id=sim_id) 

34 

35 if not simulation: 

36 return yaptide_response(message=f"Simulation {sim_id} does not exist", code=400) 

37 

38 if not simulation.check_update_key(payload_dict["update_key"]): 

39 return yaptide_response(message="Invalid update key", code=400) 

40 

41 task = fetch_task_by_sim_id_and_task_id(sim_id=simulation.id, task_id=payload_dict["task_id"]) 

42 

43 if not task: 

44 return yaptide_response(message=f"Task {payload_dict['task_id']} does not exist", code=400) 

45 

46 update_task_state(task=task, update_dict=payload_dict["update_dict"]) 

47 

48 return yaptide_response(message="Task updated", code=202)