feat(kol): KOL workspace services (profile/package/prompt/asset/renderer)

2026-04-17 09:19:30 +08:00
parent 76f5c73773
commit 6029b0b169
6 changed files with 1066 additions and 0 deletions
@@ -0,0 +1,98 @@
+package app
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestKolVariableRenderPrompt(t *testing.T) {
+	t.Parallel()
+
+	testCases := []struct {
+		name    string
+		content string
+		schema  KolSchemaJSON
+		values  map[string]any
+		want    string
+		wantErr string
+	}{
+		{
+			name:    "success",
+			content: "Hello {{var_a}} from {{var_b}}",
+			schema: KolSchemaJSON{
+				Variables: []KolVariableDefinition{
+					{ID: "var_a", Type: KolVariableTypeInput, Required: true},
+					{ID: "var_b", Type: KolVariableTypeSelect},
+				},
+			},
+			values: map[string]any{
+				"var_a": "宜家",
+				"var_b": "家具",
+			},
+			want: "Hello 宜家 from 家具",
+		},
+		{
+			name:    "missing required",
+			content: "Hello {{var_required}}",
+			schema: KolSchemaJSON{
+				Variables: []KolVariableDefinition{
+					{ID: "var_required", Type: KolVariableTypeInput, Required: true},
+				},
+			},
+			values:  map[string]any{},
+			wantErr: "missing variables",
+		},
+	}
+
+	for _, tc := range testCases {
+		tc := tc
+		t.Run(tc.name, func(t *testing.T) {
+			t.Parallel()
+
+			rendered, err := RenderPrompt(tc.content, tc.schema, tc.values)
+			if tc.wantErr != "" {
+				require.Error(t, err)
+				require.Contains(t, err.Error(), tc.wantErr)
+				return
+			}
+
+			require.NoError(t, err)
+			require.Equal(t, tc.want, rendered)
+		})
+	}
+}
+
+func TestKolVariableHashPromptStable(t *testing.T) {
+	t.Parallel()
+
+	require.Equal(t, HashPrompt("abc"), HashPrompt("abc"))
+	require.NotEqual(t, HashPrompt("abc"), HashPrompt("abd"))
+}
+
+func TestKolVariableValidateSchemaRejectsDuplicateID(t *testing.T) {
+	t.Parallel()
+
+	err := ValidateSchema(KolSchemaJSON{
+		Variables: []KolVariableDefinition{
+			{ID: "var_a", Type: KolVariableTypeInput},
+			{ID: "var_a", Type: KolVariableTypeSelect},
+		},
+	})
+
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "duplicate variable id")
+}
+
+func TestKolVariableValidateSchemaRejectsInvalidID(t *testing.T) {
+	t.Parallel()
+
+	err := ValidateSchema(KolSchemaJSON{
+		Variables: []KolVariableDefinition{
+			{ID: "bad_id", Type: KolVariableTypeInput},
+		},
+	})
+
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "start with var_")
+}