feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/domain/workflow/internal/compose/field_fill_test.go

@@ -0,0 +1,300 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package compose
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/coze-dev/coze-studio/backend/domain/workflow/entity/vo"
+)
+
+func TestNodeSchema_OutputValueFiller(t *testing.T) {
+	type fields struct {
+		In      map[string]any
+		Outputs map[string]*vo.TypeInfo
+	}
+	tests := []struct {
+		name    string
+		fields  fields
+		want    map[string]any
+		wantErr string
+	}{
+		{
+			name: "string field",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeString,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": nil,
+			},
+		},
+		{
+			name: "integer field",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeInteger,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": nil,
+			},
+		},
+		{
+			name: "number field",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeNumber,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": nil,
+			},
+		},
+		{
+			name: "boolean field",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeBoolean,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": nil,
+			},
+		},
+		{
+			name: "time field",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeTime,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": nil,
+			},
+		},
+		{
+			name: "object field",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeObject,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": nil,
+			},
+		},
+		{
+			name: "array field",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeArray,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": nil,
+			},
+		},
+		{
+			name: "file field",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeFile,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": nil,
+			},
+		},
+		{
+			name: "required field not present",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type:     vo.DataTypeString,
+						Required: true,
+					},
+				},
+			},
+			wantErr: "is required but not present",
+		},
+		{
+			name: "layered: object.string",
+			fields: fields{
+				In: map[string]any{
+					"key": map[string]any{},
+				},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeObject,
+						Properties: map[string]*vo.TypeInfo{
+							"sub_key": {
+								Type: vo.DataTypeString,
+							},
+						},
+					},
+				},
+			},
+			want: map[string]any{
+				"key": map[string]any{
+					"sub_key": nil,
+				},
+			},
+		},
+		{
+			name: "layered: object.object",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeObject,
+						Properties: map[string]*vo.TypeInfo{
+							"sub_key": {
+								Type: vo.DataTypeObject,
+							},
+						},
+					},
+				},
+			},
+			want: map[string]any{
+				"key": map[string]any{
+					"sub_key": nil,
+				},
+			},
+		},
+		{
+			name: "layered: object.object.array",
+			fields: fields{
+				In: map[string]any{},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeObject,
+						Properties: map[string]*vo.TypeInfo{
+							"sub_key": {
+								Type: vo.DataTypeObject,
+								Properties: map[string]*vo.TypeInfo{
+									"sub_key2": {
+										Type: vo.DataTypeArray,
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+			want: map[string]any{
+				"key": map[string]any{
+					"sub_key": map[string]any{
+						"sub_key2": nil,
+					},
+				},
+			},
+		},
+		{
+			name: "key present",
+			fields: fields{
+				In: map[string]any{
+					"key": "value",
+				},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeString,
+					},
+				},
+			},
+			want: map[string]any{
+				"key": "value",
+			},
+		},
+		{
+			name: "layered key present",
+			fields: fields{
+				In: map[string]any{
+					"key": map[string]any{},
+				},
+				Outputs: map[string]*vo.TypeInfo{
+					"key": {
+						Type: vo.DataTypeObject,
+						Properties: map[string]*vo.TypeInfo{
+							"sub_key": {
+								Type: vo.DataTypeObject,
+								Properties: map[string]*vo.TypeInfo{
+									"sub_key2": {
+										Type: vo.DataTypeArray,
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+			want: map[string]any{
+				"key": map[string]any{
+					"sub_key": map[string]any{
+						"sub_key2": nil,
+					},
+				},
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			s := &NodeSchema{
+				OutputTypes: tt.fields.Outputs,
+			}
+
+			got, err := s.outputValueFiller()(context.Background(), tt.fields.In)
+
+			if len(tt.wantErr) > 0 {
+				assert.Error(t, err)
+				assert.ErrorContains(t, err, tt.wantErr)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.want, got)
+			}
+		})
+	}
+}