Fix flaky controller test (#4930)

kanisterio · Feb 9, 2019 · 4c38a71 · 4c38a71
1 parent 4c945f0
commit 4c38a71
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/controller/controller_test.go b/pkg/controller/controller_test.go
@@ -133,11 +133,14 @@ func (s *ControllerSuite) waitOnActionSetState(c *C, as *crv1alpha1.ActionSet, s
 		if as.Status.State == state {
 			return true, nil
 		}
+		if state == crv1alpha1.StateRunning && as.Status.State == crv1alpha1.StateComplete {
+			return true, nil
+		}
 		// These are non-terminal states.
 		if as.Status.State == crv1alpha1.StatePending || as.Status.State == crv1alpha1.StateRunning {
 			return false, nil
 		}
-		return false, errors.New(fmt.Sprintf("Unexpected state: %s", state))
+		return false, errors.New(fmt.Sprintf("Unexpected state: %s", as.Status.State))
 
 	})
 	if err == nil {