Update security context mock in test

Signed-off-by: noahjax <[email protected]>
flyteorg · Mar 27, 2024 · 81f8753 · 81f8753
1 parent bce172e
commit 81f8753
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/flyteplugins/go/tasks/plugins/webapi/agent/integration_test.go b/flyteplugins/go/tasks/plugins/webapi/agent/integration_test.go
@@ -227,6 +227,9 @@ func getTaskContext(t *testing.T) *pluginCoreMocks.TaskExecutionContext {
 	tMeta.OnGetAnnotations().Return(map[string]string{"foo": "bar"})
 	tMeta.OnGetK8sServiceAccount().Return("k8s-account")
 	tMeta.OnGetEnvironmentVariables().Return(map[string]string{"foo": "bar"})
+	tMeta.OnGetSecurityContext().Return(flyteIdlCore.SecurityContext{
+		RunAs: &flyteIdlCore.Identity{ExecutionIdentity: "execution-identity"},
+	})
 	resourceManager := &pluginCoreMocks.ResourceManager{}
 	resourceManager.OnAllocateResourceMatch(mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(pluginCore.AllocationStatusGranted, nil)
 	resourceManager.OnReleaseResourceMatch(mock.Anything, mock.Anything, mock.Anything).Return(nil)