From 924c1d014afb6d1851dd003fdf265ed309dbc911 Mon Sep 17 00:00:00 2001
From: Jorrit Schaap <schaap@astron.nl>
Date: Tue, 2 Nov 2021 11:28:53 +0100
Subject: [PATCH] TMSS-1124: fixed tests

---
 .../test/t_tmssapp_scheduling_REST_API.py     | 41 +++++--------------
 SAS/TMSS/backend/test/tmss_test_data_rest.py  |  8 ++--
 2 files changed, 15 insertions(+), 34 deletions(-)

diff --git a/SAS/TMSS/backend/test/t_tmssapp_scheduling_REST_API.py b/SAS/TMSS/backend/test/t_tmssapp_scheduling_REST_API.py
index f784b6a6586..cfe84258a01 100755
--- a/SAS/TMSS/backend/test/t_tmssapp_scheduling_REST_API.py
+++ b/SAS/TMSS/backend/test/t_tmssapp_scheduling_REST_API.py
@@ -478,8 +478,8 @@ class SubtaskInputTestCase(unittest.TestCase):
         test_data_creator.wipe_cache()
         cls.subtask_data = test_data_creator.Subtask()
         cls.subtask_url = test_data_creator.post_data_and_get_url(cls.subtask_data, '/subtask/')
-        cls.task_relation_blueprint_data = test_data_creator.TaskRelationBlueprint()
-        cls.task_relation_blueprint_url = test_data_creator.post_data_and_get_url(cls.task_relation_blueprint_data, '/task_relation_blueprint/')
+        cls.task_connector_type_data = test_data_creator.TaskConnectorType()
+        cls.task_connector_type_url = test_data_creator.post_data_and_get_url(cls.task_connector_type_data, '/task_connector_type/')
         cls.dataproduct_urls = [test_data_creator.post_data_and_get_url(test_data_creator.Dataproduct(), '/dataproduct/'), test_data_creator.post_data_and_get_url(test_data_creator.Dataproduct(), '/dataproduct/')]
         cls.subtask_output_data = test_data_creator.SubtaskOutput()
         cls.subtask_output_url = test_data_creator.post_data_and_get_url(cls.subtask_output_data, '/subtask_output/')
@@ -494,7 +494,7 @@ class SubtaskInputTestCase(unittest.TestCase):
         GET_and_assert_equal_expected_code(self, BASE_URL + '/subtask_input/1234321/', 404)
 
     def test_subtask_input_POST_and_GET(self):
-        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, task_relation_blueprint_url=self.task_relation_blueprint_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
+        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, input_role_url=self.task_connector_type_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
 
         # POST and GET a new item and assert correctness
         r_dict = POST_and_assert_expected_response(self, BASE_URL + '/subtask_input/', sti_test_data, 201, sti_test_data)
@@ -502,12 +502,12 @@ class SubtaskInputTestCase(unittest.TestCase):
         GET_OK_and_assert_equal_expected_response(self, url, sti_test_data)
 
     def test_subtask_input_PUT_invalid_raises_error(self):
-        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, task_relation_blueprint_url=self.task_relation_blueprint_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
+        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, input_role_url=self.task_connector_type_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
 
         PUT_and_assert_expected_response(self, BASE_URL + '/subtask_input/9876789876/', sti_test_data, 404, {})
 
     def test_subtask_input_PUT(self):
-        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, task_relation_blueprint_url=self.task_relation_blueprint_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
+        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, input_role_url=self.task_connector_type_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
 
         # POST new item, verify
         r_dict = POST_and_assert_expected_response(self, BASE_URL + '/subtask_input/', sti_test_data, 201, sti_test_data)
@@ -515,12 +515,12 @@ class SubtaskInputTestCase(unittest.TestCase):
         GET_OK_and_assert_equal_expected_response(self, url, sti_test_data)
 
         # PUT new values, verify
-        sti_test_data2 = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, task_relation_blueprint_url=self.task_relation_blueprint_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
+        sti_test_data2 = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, input_role_url=self.task_connector_type_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
         PUT_and_assert_expected_response(self, url, sti_test_data2, 200, sti_test_data2)
         GET_OK_and_assert_equal_expected_response(self, url, sti_test_data2)
 
     def test_subtask_input_PATCH(self):
-        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, task_relation_blueprint_url=self.task_relation_blueprint_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
+        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, input_role_url=self.task_connector_type_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
 
         # POST new item, verify
         r_dict = POST_and_assert_expected_response(self, BASE_URL + '/subtask_input/', sti_test_data, 201, sti_test_data)
@@ -544,7 +544,7 @@ class SubtaskInputTestCase(unittest.TestCase):
         GET_OK_and_assert_equal_expected_response(self, url, expected_data)
 
     def test_subtask_input_DELETE(self):
-        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, task_relation_blueprint_url=self.task_relation_blueprint_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
+        sti_test_data = test_data_creator.SubtaskInput(subtask_url=self.subtask_url, input_role_url=self.task_connector_type_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
 
         # POST new item, verify
         r_dict = POST_and_assert_expected_response(self, BASE_URL + '/subtask_input/', sti_test_data, 201, sti_test_data)
@@ -560,7 +560,7 @@ class SubtaskInputTestCase(unittest.TestCase):
                                                                                         task_blueprint_url=test_data_creator.post_data_and_get_url(test_data_creator.TaskBlueprint(), '/task_blueprint/'),
                                                                                         specifications_template_url=self.subtask_data['specifications_template'],
                                                                                         specifications_doc=self.subtask_data['specifications_doc']), '/subtask/')
-        sti_test_data = test_data_creator.SubtaskInput(subtask_url=subtask_url, task_relation_blueprint_url=self.task_relation_blueprint_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
+        sti_test_data = test_data_creator.SubtaskInput(subtask_url=subtask_url, input_role_url=self.task_connector_type_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
 
         # POST new item, verify
         url = POST_and_assert_expected_response(self, BASE_URL + '/subtask_input/', sti_test_data, 201, sti_test_data)['url']
@@ -572,29 +572,10 @@ class SubtaskInputTestCase(unittest.TestCase):
         # assert item gone
         GET_and_assert_equal_expected_code(self, url, 404)
 
-    def test_subtask_input_SET_NULL_behavior_on_task_relation_blueprint_deleted(self):
-        # make new task_relation_blueprint instance, but reuse related data for speed
-        task_relation_blueprint_url = test_data_creator.post_data_and_get_url(test_data_creator.TaskRelationBlueprint(draft_url=self.task_relation_blueprint_data['draft'], template_url=self.task_relation_blueprint_data['selection_template'],
-                                                                                                                      input_role_url=self.task_relation_blueprint_data['input_role'], output_role_url=self.task_relation_blueprint_data['output_role'],
-                                                                                                                      consumer_url=self.task_relation_blueprint_data['consumer']), '/task_relation_blueprint/')
-        sti_test_data = test_data_creator.SubtaskInput(task_relation_blueprint_url=task_relation_blueprint_url, subtask_url=self.subtask_url, dataproduct_urls=self.dataproduct_urls, subtask_output_url=self.subtask_output_url, task_relation_selection_template_url=self.task_relation_selection_template_url)
-
-        # POST new item, verify
-        url = POST_and_assert_expected_response(self, BASE_URL + '/subtask_input/', sti_test_data, 201, sti_test_data)['url']
-        GET_OK_and_assert_equal_expected_response(self, url, sti_test_data)
-
-        # DELETE dependency and check it's gone
-        DELETE_and_assert_gone(self, task_relation_blueprint_url)
-
-        # assert item reference is set null
-        expected_data = dict(sti_test_data)
-        expected_data['task_relation_blueprint'] = None
-        GET_OK_and_assert_equal_expected_response(self, url, expected_data)
-
     def test_subtask_input_PROTECT_behavior_on_producer_deleted(self):
         # make new subtask_output_url instance, but reuse related data for speed
         subtask_output_url = test_data_creator.post_data_and_get_url(test_data_creator.SubtaskOutput(subtask_url=self.subtask_output_data['subtask']), '/subtask_output/')
-        sti_test_data = test_data_creator.SubtaskInput(subtask_output_url=subtask_output_url, subtask_url=self.subtask_url, task_relation_blueprint_url=self.task_relation_blueprint_url, dataproduct_urls=self.dataproduct_urls, task_relation_selection_template_url=self.task_relation_selection_template_url)
+        sti_test_data = test_data_creator.SubtaskInput(subtask_output_url=subtask_output_url, subtask_url=self.subtask_url, input_role_url=self.task_connector_type_url, dataproduct_urls=self.dataproduct_urls, task_relation_selection_template_url=self.task_relation_selection_template_url)
 
         # POST with dependency
         url = POST_and_assert_expected_response(self, BASE_URL + '/subtask_input/', sti_test_data, 201, sti_test_data)['url']
@@ -611,7 +592,7 @@ class SubtaskInputTestCase(unittest.TestCase):
         task_relation_selection_template_url = test_data_creator.post_data_and_get_url(test_data_creator.TaskRelationSelectionTemplate(), '/task_relation_selection_template/')
         sti_test_data = test_data_creator.SubtaskInput(task_relation_selection_template_url=task_relation_selection_template_url,
                                                        subtask_url=self.subtask_url,
-                                                       task_relation_blueprint_url=self.task_relation_blueprint_url,
+                                                       input_role_url=self.task_connector_type_url,
                                                        dataproduct_urls=self.dataproduct_urls,
                                                        subtask_output_url=self.subtask_output_url)
 
diff --git a/SAS/TMSS/backend/test/tmss_test_data_rest.py b/SAS/TMSS/backend/test/tmss_test_data_rest.py
index 71bf02695ce..2965cb508e6 100644
--- a/SAS/TMSS/backend/test/tmss_test_data_rest.py
+++ b/SAS/TMSS/backend/test/tmss_test_data_rest.py
@@ -765,12 +765,12 @@ class TMSSRESTTestDataCreator():
                 "corrupted_since": datetime.utcnow().isoformat(),
                 "tags": ['tmss', 'testing']}
     
-    def SubtaskInput(self, subtask_url=None, task_relation_blueprint_url=None, dataproduct_urls=None, subtask_output_url=None, task_relation_selection_template_url=None, selection_doc=None):
+    def SubtaskInput(self, subtask_url=None, input_role_url=None, dataproduct_urls=None, subtask_output_url=None, task_relation_selection_template_url=None, selection_doc=None):
         if subtask_url is None:
             subtask_url = self.cached_subtask_url
     
-        if task_relation_blueprint_url is None:
-            task_relation_blueprint_url = self.post_data_and_get_url(self.TaskRelationBlueprint(), '/task_relation_blueprint/')
+        if input_role_url is None:
+            input_role_url = self.post_data_and_get_url(self.TaskConnectorType(), '/task_connector_type/')
     
         if dataproduct_urls is None:
             dataproduct_urls = [self.cached_dataproduct_url]
@@ -785,7 +785,7 @@ class TMSSRESTTestDataCreator():
             selection_doc = self.get_response_as_json_object(task_relation_selection_template_url+'/default')
 
         return {"subtask": subtask_url,
-                "task_relation_blueprint": task_relation_blueprint_url,
+                "input_role": input_role_url,
                 "producer": subtask_output_url,
                 "dataproducts": dataproduct_urls,
                 "selection_doc": selection_doc,
-- 
GitLab