fix failing test case (flyteorg#91)

eapolinario · Mar 16, 2020 · 1f23878 · 1f23878
1 parent f0f2237
commit 1f23878
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/flytepropeller/pkg/controller/nodes/subworkflow/subworkflow_test.go b/flytepropeller/pkg/controller/nodes/subworkflow/subworkflow_test.go
@@ -53,7 +53,10 @@ func Test_subworkflowHandler_HandleAbort(t *testing.T) {
 		nCtx.OnNodeID().Return("n1")
 		n := &coreMocks.ExecutableNode{}
 		swf.OnStartNode().Return(n)
-		nodeExec.OnAbortHandler(ctx, wf, n, "reason").Return(fmt.Errorf("err"))
+		swf.OnGetID().Return("swf")
+		nodeExec.OnAbortHandlerMatch(mock.Anything, mock.MatchedBy(func(wf v1alpha1.ExecutableWorkflow) bool {
+			return wf.GetID() == swf.GetID()
+		}), n, mock.Anything).Return(fmt.Errorf("err"))
 		assert.Error(t, s.HandleAbort(ctx, nCtx, wf, "x", "reason"))
 	})