From f85e63652cbe2272ae05acdc3d8ad2b747972031 Mon Sep 17 00:00:00 2001
From: Jorrit Schaap <schaap@astron.nl>
Date: Fri, 5 Jan 2024 11:33:17 +0100
Subject: [PATCH] TMSS-2879: handle 'on hold' path to 'started'

---
 SAS/TMSS/backend/test/test_utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/SAS/TMSS/backend/test/test_utils.py b/SAS/TMSS/backend/test/test_utils.py
index 7034d425593..4412e642347 100644
--- a/SAS/TMSS/backend/test/test_utils.py
+++ b/SAS/TMSS/backend/test/test_utils.py
@@ -135,8 +135,10 @@ def set_subtask_state_following_allowed_transitions(subtask: typing.Union[models
         elif desired_state_value in (models.SubtaskState.Choices.ON_HOLD.value, models.SubtaskState.Choices.RESTARTING.value) and subtask.state.value == models.SubtaskState.Choices.STARTED.value:
             subtask.state = models.SubtaskState.objects.get(value=models.SubtaskState.Choices.ON_HOLD.value)
         # handle "partially successfull path" to restarting state
-        elif desired_state_value == models.SubtaskState.Choices.RESTARTING.value and subtask.state.value == models.SubtaskState.Choices.ON_HOLD.value:
+        elif desired_state_value in (models.SubtaskState.Choices.RESTARTING.value, models.SubtaskState.Choices.STARTED.value) and subtask.state.value == models.SubtaskState.Choices.ON_HOLD.value:
             subtask.state = models.SubtaskState.objects.get(value=models.SubtaskState.Choices.RESTARTING.value)
+        elif desired_state_value == models.SubtaskState.Choices.STARTED.value and subtask.state.value == models.SubtaskState.Choices.RESTARTING.value:
+            subtask.state = models.SubtaskState.objects.get(value=models.SubtaskState.Choices.STARTING.value)
         # handle reverse path to defined
         elif desired_state_value == models.SubtaskState.Choices.DEFINED.value and subtask.state.value == models.SubtaskState.Choices.UNSCHEDULABLE.value:
             subtask.state = models.SubtaskState.objects.get(value=models.SubtaskState.Choices.DEFINED.value)
-- 
GitLab