diff --git a/SAS/TMSS/services/websocket/test/t_websocket_service.py b/SAS/TMSS/services/websocket/test/t_websocket_service.py index 248e4fd46206904947075b85c4f8e7ee496d0434..0291c69cc7a09eb5fff899d4de916abd928de6a2 100755 --- a/SAS/TMSS/services/websocket/test/t_websocket_service.py +++ b/SAS/TMSS/services/websocket/test/t_websocket_service.py @@ -144,6 +144,32 @@ class TestSubtaskSchedulingService(unittest.TestCase): self.test_data_creator.Subtask(task_blueprint_url=task_blueprint['url']), '/subtask/') test_object(subtask, 'subtask', 'create') + # Test updates + with self.tmss_test_env.create_tmss_client() as client: + # Test subtask update + client.set_subtask_status(subtask['id'], 'scheduled') + if not self.sync_event.wait(timeout=5): + raise TimeoutError() + self.sync_event.clear() + json_test = self.msg_queue.popleft() + self.assertEqual({'id': subtask['id'], 'action': 'update', 'state_value': 'scheduled'}, + {'id': json_test['id'], 'action': json_test['action'], + 'state_value': json_test['object']['state_value']}) + + # Test task_blueprint update + if not self.sync_event.wait(timeout=5): + raise TimeoutError() + json_test = self.msg_queue.popleft() + self.assertEqual({'id': task_blueprint['id'], 'object_type': 'task_blueprint', 'action': 'update'}, + {'id': json_test['id'], 'object_type': json_test['object_type'], 'action': json_test['action']}) + # Test scheduling_unit_blueprint update + if not self.sync_event.wait(timeout=5): + raise TimeoutError() + self.sync_event.clear() + json_test = self.msg_queue.popleft() + self.assertEqual({'id': su_blueprint['id'], 'object_type': 'scheduling_unit_blueprint', 'action': 'update'}, + {'id': json_test['id'], 'object_type': json_test['object_type'], 'action': json_test['action']}) + # TODO: create/update/delete objects like SubTask, TaskBlueprint etc