geo/server/internal/tenant/app/kol_assist_runtime_test.go

package app

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestFinalizeKolAssistResultRepairsDuplicateVariableIDs(t *testing.T) {
	t.Parallel()

	result, err := FinalizeKolAssistResult(
		AssistRequest{Mode: kolAssistModeGenerate},
		&KolAssistResult{
			Content: "围绕 {{城市}} 和 {{目标词}} 输出内容",
			Schema: KolSchemaJSON{
				Variables: []KolVariableDefinition{
					{ID: "var_dup", Key: "城市", Label: "城市", Type: KolVariableTypeInput, Required: true},
					{ID: "var_dup", Key: "目标词", Label: "目标词", Type: KolVariableTypeInput, Required: true},
				},
			},
		},
	)

	require.NoError(t, err)
	require.Len(t, result.Schema.Variables, 2)
	require.NotEqual(t, result.Schema.Variables[0].ID, result.Schema.Variables[1].ID)
	require.Equal(t, "城市", result.Schema.Variables[0].Key)
	require.Equal(t, "目标词", result.Schema.Variables[1].Key)
}

func TestFinalizeKolAssistResultOptimizePreservesExistingSchemaByKey(t *testing.T) {
	t.Parallel()

	originalMaxLength := 80

	result, err := FinalizeKolAssistResult(
		AssistRequest{
			Mode: kolAssistModeOptimize,
			Schema: &KolSchemaJSON{
				Variables: []KolVariableDefinition{
					{ID: "var_city", Key: "城市", Label: "城市", Type: KolVariableTypeInput, Required: true, MaxLength: intPtr(originalMaxLength)},
				},
			},
		},
		&KolAssistResult{
			Content: "围绕 {{城市}} 优化提示词",
			Schema: KolSchemaJSON{
				Variables: []KolVariableDefinition{
					{ID: "var_dup", Key: "城市", Label: "城市", Type: KolVariableTypeTextarea, Required: false},
				},
			},
		},
	)

	require.NoError(t, err)
	require.Len(t, result.Schema.Variables, 1)
	require.Equal(t, "var_city", result.Schema.Variables[0].ID)
	require.Equal(t, KolVariableTypeInput, result.Schema.Variables[0].Type)
	require.NotNil(t, result.Schema.Variables[0].MaxLength)
	require.Equal(t, originalMaxLength, *result.Schema.Variables[0].MaxLength)
}

func TestParseKolAssistResponseSupportsStringVariables(t *testing.T) {
	t.Parallel()

	result, err := ParseKolAssistResponse(`{"content":"围绕 {{城市}} 写内容","schema":{"variables":["城市"]}}`)

	require.NoError(t, err)
	require.Equal(t, "围绕 {{城市}} 写内容", result.Content)
	require.Len(t, result.Schema.Variables, 1)
	require.Equal(t, "城市", result.Schema.Variables[0].Key)
}

func TestBuildKolAssistUserPromptIncludesOptimizeInstruction(t *testing.T) {
	t.Parallel()

	prompt, err := BuildKolAssistUserPrompt(AssistRequest{
		Mode:           kolAssistModeOptimize,
		Description:    "让表达更清晰，但保留变量",
		CurrentContent: "围绕 {{城市}} 写一段内容",
	})

	require.NoError(t, err)
	require.Contains(t, prompt, "Instruction:\n让表达更清晰，但保留变量")
	require.Contains(t, prompt, "PROMPT:\n围绕 {{城市}} 写一段内容")
}

func TestExtractKolAssistContentFromPartialJSON(t *testing.T) {
	t.Parallel()

	content := extractKolAssistContentFromPartialJSON("{\"content\":\"第一段\\n第二")

	require.Equal(t, "第一段\n第二", content)
}