Fix: Update

Author: vr-varad

dbos/dbos.go

@@ -360,6 +360,7 @@ func NewDBOSContext(ctx context.Context, inputConfig Config) (DBOSContext, error
 	var ws []WorkflowStatus
 	safeGobRegister(ws, initExecutor.logger)
 	var si []StepInfo
+	safeGobRegister(si, initExecutor.logger)
 	var h workflowHandle[any]
 	safeGobRegister(h, initExecutor.logger)
 	var ph workflowPollingHandle[any]

dbos/dbos_test.go

@@ -735,22 +735,38 @@ func TestWorkflowHandlerRegistration(t *testing.T) {
 
 	require.NotNil(t, ctx)
 
-	wf := func(ctx DBOSContext, input string) (string, error) {
-		return input, nil
+	childWf := func(ctx DBOSContext, input string) (string, error) {
+		return "child-" + input, nil
+	}
+
+	wf := func(ctx DBOSContext, input string) (WorkflowHandle[string], error) {
+		childHandle, err := RunWorkflow(ctx, childWf, input)
+		if err != nil {
+			var emptyHandle WorkflowHandle[string]
+			return emptyHandle, err
+		}
+		return childHandle, nil
 	}
 
+	RegisterWorkflow(ctx, childWf)
 	RegisterWorkflow(ctx, wf)
 
 	// Launch a workflow
 	err = Launch(ctx)
 	require.NoError(t, err)
 	defer Shutdown(ctx, 1*time.Minute)
 
-	// Run a workflow
-	handle, err := RunWorkflow(ctx, wf, "test-input")
+	parentHandle, err := RunWorkflow(ctx, wf, "test-input")
 	require.NoError(t, err)
 
-	result, err := handle.GetResult()
-	require.NoError(t, err, "failed to get result from workflow")
-	assert.Equal(t, "test-input", result)
+	// The result of the parent workflow should itself be a workflow handle
+	childHandle, err := parentHandle.GetResult()
+	require.NoError(t, err, "failed to get child handle from parent workflow")
+
+	// Now we can use the child handle to get the final result
+	result, err := childHandle.GetResult()
+	require.NoError(t, err, "failed to get result from child workflow")
+
+	// Assert final child workflow result
+	assert.Equal(t, "child-test-input", result)
 }