fix(test): TestWorkflowTemplateRefWithShutdownAndSuspend flaky (#5381)

Signed-off-by: Saravanan Balasubramanian <[email protected]>
argoproj · Mar 12, 2021 · 982e5e9 · 982e5e9
1 parent 57c05df
commit 982e5e9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/workflow/controller/operator_workflow_template_ref_test.go b/workflow/controller/operator_workflow_template_ref_test.go
@@ -333,7 +333,8 @@ func TestWorkflowTemplateRefWithShutdownAndSuspend(t *testing.T) {
  assert.Equal(t, wfv1.ShutdownStrategyTerminate, woc1.wf.Status.StoredWorkflowSpec.Shutdown)
  for _, node := range woc1.wf.Status.Nodes {
  if assert.NotNil(t, node) {
- assert.Contains(t, node.Message, "workflow shutdown with strategy: Terminate")
+ assert.Contains(t, node.Message, "workflow shutdown with strategy")
+ assert.Contains(t, node.Message, "Terminate")
  }
  }
  })
@@ -352,7 +353,8 @@ func TestWorkflowTemplateRefWithShutdownAndSuspend(t *testing.T) {
  assert.Equal(t, wfv1.ShutdownStrategyStop, woc1.wf.Status.StoredWorkflowSpec.Shutdown)
  for _, node := range woc1.wf.Status.Nodes {
  if assert.NotNil(t, node) {
- assert.Contains(t, node.Message, "workflow shutdown with strategy: Stop")
+ assert.Contains(t, node.Message, "workflow shutdown with strategy")
+ assert.Contains(t, node.Message, "Stop")
  }
  }
  })