Make propeller tests pass

Signed-off-by: Fabio Graetz <[email protected]>
flyteorg · Apr 7, 2024 · c6d0802 · c6d0802
1 parent 8a8bdf1
commit c6d0802
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/flytepropeller/pkg/controller/nodes/task/k8s/plugin_manager_test.go b/flytepropeller/pkg/controller/nodes/task/k8s/plugin_manager_test.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"reflect"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
@@ -762,13 +763,15 @@ func TestPluginManager_Handle_PluginState(t *testing.T) {
 		},
 	}
 
-	phaseInfoQueued := pluginsCore.PhaseInfoQueuedWithTaskInfo(pluginStateQueued.K8sPluginState.PhaseVersion, pluginStateQueued.K8sPluginState.Reason, nil)
+	phaseInfoQueued := pluginsCore.PhaseInfoQueuedWithTaskInfo(time.Now(), pluginStateQueued.K8sPluginState.PhaseVersion, pluginStateQueued.K8sPluginState.Reason, nil)
 	phaseInfoQueuedVersion1 := pluginsCore.PhaseInfoQueuedWithTaskInfo(
+		time.Now(),
 		pluginStateQueuedVersion1.K8sPluginState.PhaseVersion,
 		pluginStateQueuedVersion1.K8sPluginState.Reason,
 		nil,
 	)
 	phaseInfoQueuedReasonBar := pluginsCore.PhaseInfoQueuedWithTaskInfo(
+		time.Now(),
 		pluginStateQueuedReasonBar.K8sPluginState.PhaseVersion,
 		pluginStateQueuedReasonBar.K8sPluginState.Reason,
 		nil,