From 89e01b566d0829401e915cc76a658fa2db63d2d6 Mon Sep 17 00:00:00 2001
From: Mario Raciti <mario.raciti@inaf.it>
Date: Wed, 9 Dec 2020 12:38:02 +0100
Subject: [PATCH] TMSS-417: Add tests for update actions

---
 .../websocket/test/t_websocket_service.py     | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/SAS/TMSS/services/websocket/test/t_websocket_service.py b/SAS/TMSS/services/websocket/test/t_websocket_service.py
index 248e4fd4620..0291c69cc7a 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
 
 
-- 
GitLab