From 8df712dff2bcfffc0d0a37d57fdba5d2712172d3 Mon Sep 17 00:00:00 2001
From: Jorrit Schaap <schaap@astron.nl>
Date: Thu, 30 Nov 2023 15:14:32 +0100
Subject: [PATCH] TMSS-2836: fixed test

---
 SAS/TMSS/backend/test/t_scheduling_units.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/SAS/TMSS/backend/test/t_scheduling_units.py b/SAS/TMSS/backend/test/t_scheduling_units.py
index f3b00444cad..1f998c43e6a 100644
--- a/SAS/TMSS/backend/test/t_scheduling_units.py
+++ b/SAS/TMSS/backend/test/t_scheduling_units.py
@@ -1317,7 +1317,7 @@ class SchedulingUnitBlueprintIndirectModificationsTestCase(unittest.TestCase):
             client.create_cleanuptask_for_scheduling_unit_blueprint(scheduling_unit_blueprint['id'])
 
             # check the blueprint graph
-            scheduling_unit_blueprint = client.get_schedulingunit_blueprint(scheduling_unit_blueprint['id'])
+            scheduling_unit_blueprint = client.get_schedulingunit_blueprint(scheduling_unit_blueprint['id'], extended=True)
             self.assertEqual(num_task_blueprints_before + 1, len(scheduling_unit_blueprint['task_blueprints']))
             self.assertIn("Cleanup", [task['name'] for task in scheduling_unit_blueprint['task_blueprints']])
             for task in scheduling_unit_blueprint['task_blueprints']:
@@ -1336,7 +1336,7 @@ class SchedulingUnitBlueprintIndirectModificationsTestCase(unittest.TestCase):
 
             # create another cleanup task for the scheduling unit and check name is as expected
             client.create_cleanuptask_for_scheduling_unit_blueprint(scheduling_unit_blueprint['id'])
-            scheduling_unit_blueprint = client.get_schedulingunit_blueprint(scheduling_unit_blueprint['id'])
+            scheduling_unit_blueprint = client.get_schedulingunit_blueprint(scheduling_unit_blueprint['id'], extended=True)
             self.assertIn("Cleanup2", [task['name'] for task in scheduling_unit_blueprint['task_blueprints']])
 
     def test_TMSS_1585_bugfix_check_cleanup_starts(self):
-- 
GitLab