TMSS-417: Add tests for update actions

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