add struct test

Author: maxdml

dbos/system_database.go

@@ -1014,7 +1014,7 @@ func (s *systemDatabase) notificationListenerLoop(ctx context.Context) {
 const DBOS_NULL_TOPIC = "__null__topic__"
 
 // Send is a special type of step that sends a message to another workflow.
-// Two differences with a normal steps: durability and the function run in the same transaction, and we forbid nested step execution
+// Three differences with a normal steps: durability and the function run in the same transaction, and we forbid nested step execution
 func (s *systemDatabase) Send(ctx context.Context, input WorkflowSendInput) error {
 	functionName := "DBOS.send"
 

dbos/workflows_test.go

@@ -980,8 +980,10 @@ func TestScheduledWorkflows(t *testing.T) {
 }
 
 var (
-	sendWf    = WithWorkflow(sendWorkflow)
-	receiveWf = WithWorkflow(receiveWorkflow)
+	sendWf       = WithWorkflow(sendWorkflow)
+	receiveWf    = WithWorkflow(receiveWorkflow)
+	sendStructWf = WithWorkflow(sendStructWorkflow)
+	receiveStructWf = WithWorkflow(receiveStructWorkflow)
 )
 
 type sendWorkflowInput struct {
@@ -1023,6 +1025,20 @@ func receiveWorkflow(ctx context.Context, topic string) (string, error) {
 	return msg1 + "-" + msg2 + "-" + msg3, nil
 }
 
+func sendStructWorkflow(ctx context.Context, input sendWorkflowInput) (string, error) {
+	testStruct := sendRecvType{Value: "test-struct-value"}
+	err := Send(ctx, WorkflowSendInput{DestinationID: input.DestinationID, Topic: input.Topic, Message: testStruct})
+	return "", err
+}
+
+func receiveStructWorkflow(ctx context.Context, topic string) (sendRecvType, error) {
+	return Recv[sendRecvType](ctx, WorkflowRecvInput{Topic: topic, Timeout: 3 * time.Second})
+}
+
+type sendRecvType struct {
+	Value string
+}
+
 func TestSendRecv(t *testing.T) {
 	setupDBOS(t)
 
@@ -1102,6 +1118,39 @@ func TestSendRecv(t *testing.T) {
 		}
 	})
 
+	t.Run("SendRecvCustomStruct", func(t *testing.T) {
+		// Start the receive workflow
+		receiveHandle, err := receiveStructWf(context.Background(), "struct-topic")
+		if err != nil {
+			t.Fatalf("failed to start receive workflow: %v", err)
+		}
+
+		// Send the struct to the receive workflow
+		sendHandle, err := sendStructWf(context.Background(), sendWorkflowInput{
+			DestinationID: receiveHandle.GetWorkflowID(),
+			Topic:         "struct-topic",
+		})
+		if err != nil {
+			t.Fatalf("failed to send struct: %v", err)
+		}
+
+		_, err = sendHandle.GetResult(context.Background())
+		if err != nil {
+			t.Fatalf("failed to get result from send workflow: %v", err)
+		}
+
+		// Get the result from receive workflow
+		result, err := receiveHandle.GetResult(context.Background())
+		if err != nil {
+			t.Fatalf("failed to get result from receive workflow: %v", err)
+		}
+
+		// Verify the struct was received correctly
+		if result.Value != "test-struct-value" {
+			t.Fatalf("expected received struct value to be 'test-struct-value', got '%s'", result.Value)
+		}
+	})
+
 	t.Run("SendToNonExistentUUID", func(t *testing.T) {
 		// Generate a non-existent UUID
 		destUUID := uuid.NewString()