Fix: Update

Author: vr-varad

dbos/workflow.go

@@ -549,6 +549,27 @@ func withWorkflowName(name string) WorkflowOption {
 	}
 }
 
+// Sets the authenticated user for the workflow
+func WithAutheticatedUser(user string) WorkflowOption {
+	return func(p *workflowOptions) {
+		p.authenticated_user = user
+	}
+}
+
+// Sets the assumed role for the workflow
+func WithAssumedRole(role string) WorkflowOption {
+	return func(p *workflowOptions) {
+		p.assumed_role = role
+	}
+}
+
+// Sets the authenticated role for the workflow
+func WithAuthenticatedRoles(roles []string) WorkflowOption {
+	return func(p *workflowOptions) {
+		p.authenticated_roles = roles
+	}
+}
+
 // RunWorkflow executes a workflow function with type safety and durability guarantees.
 // The workflow can be executed immediately or enqueued for later execution based on options.
 // Returns a typed handle that can be used to wait for completion and retrieve results.

dbos/workflows_test.go

@@ -4131,3 +4131,31 @@ func TestSpecialSteps(t *testing.T) {
 		require.Equal(t, "success", result, "workflow should return success")
 	})
 }
+func TestWorkflowIdentity(t *testing.T) {
+	dbosCtx := setupDBOS(t, true, true)
+	handle, err := RunWorkflow(
+		dbosCtx,
+		simpleWorkflow,
+		"test",
+		WithWorkflowID("my-workflow-id"),
+		WithAutheticatedUser("user123"),
+		WithAssumedRole("admin"),
+		WithAuthenticatedRoles([]string{"reader", "writer"}))
+	require.NoError(t, err, "failed to start workflow")
+
+	// Retrieve the workflow's status.
+	status, err := handle.GetStatus()
+	require.NoError(t, err)
+
+	t.Run("CheckAuthenticatedUser", func(t *testing.T) {
+		assert.Equal(t, "user123", status.AuthenticatedUser)
+	})
+
+	t.Run("CheckAssumedRole", func(t *testing.T) {
+		assert.Equal(t, "admin", status.AssumedRole)
+	})
+
+	t.Run("CheckAuthenticatedRoles", func(t *testing.T) {
+		assert.Equal(t, []string{"reader", "writer"}, status.AuthenticatedRoles)
+	})
+}