[DEVOPS-646] Fixed wait condition

- Failed tasks are also complete!
salsadigitalauorg · Oct 22, 2024 · 0ca5558 · 0ca5558
1 parent c24c6d9
commit 0ca5558
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/api/plugins/action/mutation.py b/api/plugins/action/mutation.py
@@ -29,7 +29,7 @@ def run(self, tmp=None, task_vars=None):
         wait = self._task.args.get('wait', False)
         waitCondition = self._task.args.get('waitCondition', {
             'field': 'status',
-            'value': 'complete',
+            'value': ['complete', 'failed'],
         })
 
         with self.client:
@@ -69,5 +69,8 @@ def waitQuery(self, qryName, waitQry, waitCondition) -> Dict[str, Any]:
 
     def evaluateWaitCondition(self, qryName, waitResult, waitCondition) -> bool:
         if waitCondition["field"] in waitResult[qryName]:
-            return waitResult[qryName][waitCondition["field"]] == waitCondition["value"]
+            if isinstance(waitCondition["value"], list):
+                return waitResult[qryName][waitCondition["field"]] in waitCondition["value"]
+            else:
+                return waitResult[qryName][waitCondition["field"]] == waitCondition["value"]
         return False