feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/domain/workflow/entity/vo/node_test.go

@@ -0,0 +1,147 @@
+/*
+ * 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 vo
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestTypeInfoToJSONSchema(t *testing.T) {
+	tests := []struct {
+		name     string
+		typeInfo map[string]*TypeInfo
+		validate func(t *testing.T, schema string)
+	}{
+		{
+			name: "Basic Data Types",
+			typeInfo: map[string]*TypeInfo{
+				"stringField": {Type: DataTypeString},
+				"intField":    {Type: DataTypeInteger},
+				"numField":    {Type: DataTypeNumber},
+				"boolField":   {Type: DataTypeBoolean},
+				"timeField":   {Type: DataTypeTime},
+			},
+			validate: func(t *testing.T, schema string) {
+				var schemaObj map[string]any
+				err := json.Unmarshal([]byte(schema), &schemaObj)
+				assert.NoError(t, err)
+
+				props := schemaObj["properties"].(map[string]any)
+
+				// 验证字符串字段
+				stringProp := props["stringField"].(map[string]any)
+				assert.Equal(t, "string", stringProp["type"])
+
+				// 验证整数字段
+				intProp := props["intField"].(map[string]any)
+				assert.Equal(t, "integer", intProp["type"])
+
+				// 验证数字字段
+				numProp := props["numField"].(map[string]any)
+				assert.Equal(t, "number", numProp["type"])
+
+				// 验证布尔字段
+				boolProp := props["boolField"].(map[string]any)
+				assert.Equal(t, "boolean", boolProp["type"])
+
+				// 验证时间字段
+				timeProp := props["timeField"].(map[string]any)
+				assert.Equal(t, "string", timeProp["type"])
+				assert.Equal(t, "date-time", timeProp["format"])
+			},
+		},
+		{
+			name: "Complex Data Types",
+			typeInfo: map[string]*TypeInfo{
+				"objectField": {Type: DataTypeObject},
+				"arrayField": {
+					Type:         DataTypeArray,
+					ElemTypeInfo: &TypeInfo{Type: DataTypeString},
+				},
+				"fileField": {
+					Type:     DataTypeFile,
+					FileType: fileSubTypePtr(FileTypeImage),
+				},
+			},
+			validate: func(t *testing.T, schema string) {
+				var schemaObj map[string]any
+				err := json.Unmarshal([]byte(schema), &schemaObj)
+				assert.NoError(t, err)
+
+				props := schemaObj["properties"].(map[string]any)
+
+				// 验证对象字段
+				objProp := props["objectField"].(map[string]any)
+				assert.Equal(t, "object", objProp["type"])
+
+				// 验证数组字段
+				arrProp := props["arrayField"].(map[string]any)
+				assert.Equal(t, "array", arrProp["type"])
+				items := arrProp["items"].(map[string]any)
+				assert.Equal(t, "string", items["type"])
+
+				// 验证文件字段
+				fileProp := props["fileField"].(map[string]any)
+				assert.Equal(t, "string", fileProp["type"])
+				assert.Equal(t, "image", fileProp["contentMediaType"])
+			},
+		},
+		{
+			name: "Nested Array",
+			typeInfo: map[string]*TypeInfo{
+				"nestedArray": {
+					Type:         DataTypeArray,
+					ElemTypeInfo: &TypeInfo{Type: DataTypeObject},
+				},
+			},
+			validate: func(t *testing.T, schema string) {
+				var schemaObj map[string]any
+				err := json.Unmarshal([]byte(schema), &schemaObj)
+				assert.NoError(t, err)
+
+				props := schemaObj["properties"].(map[string]any)
+
+				// 验证嵌套数组字段
+				arrProp := props["nestedArray"].(map[string]any)
+				assert.Equal(t, "array", arrProp["type"])
+				items := arrProp["items"].(map[string]any)
+				assert.Equal(t, "object", items["type"])
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			schema, err := TypeInfoToJSONSchema(tt.typeInfo, nil)
+			assert.NoError(t, err)
+			tt.validate(t, schema)
+		})
+	}
+}
+
+// 辅助函数，用于创建 DataType 指针
+func stringPtr(dt DataType) *DataType {
+	return &dt
+}
+
+// 辅助函数，用于创建 FileSubType 指针
+func fileSubTypePtr(fst FileSubType) *FileSubType {
+	return &fst
+}