53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
|
package catalog
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/descrybe/descrybe-v2/apps/api/internal/aiprompts"
|
||
|
|
"github.com/descrybe/descrybe-v2/apps/api/internal/company"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRepairedCategoryEnhancePromptMap(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
want, err := repairedCategoryEnhancePromptMap()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
tpl := strings.TrimSpace(aiprompts.CategoryEnhanceUserTemplate)
|
||
|
|
if want["sl"] != tpl {
|
||
|
|
t.Fatalf("want prompt->>'sl' = shared template")
|
||
|
|
}
|
||
|
|
if want[company.LangPromptAny] != tpl {
|
||
|
|
t.Fatalf("want * = shared template")
|
||
|
|
}
|
||
|
|
if !categoryEnhancePromptMapOK(want, want) {
|
||
|
|
t.Fatal("canonical map should be OK")
|
||
|
|
}
|
||
|
|
legacy := company.LangPromptMap{"sl": "<H2>legacy HTML marketing</H2>"}
|
||
|
|
if categoryEnhancePromptMapOK(legacy, want) {
|
||
|
|
t.Fatal("legacy HTML must need repair")
|
||
|
|
}
|
||
|
|
slOnly := company.LangPromptMap{"sl": tpl}
|
||
|
|
if !categoryEnhancePromptMapOK(slOnly, want) {
|
||
|
|
t.Fatal("sl-only repaired map should be OK (idempotent)")
|
||
|
|
}
|
||
|
|
starOnly := company.LangPromptMap{company.LangPromptAny: tpl}
|
||
|
|
if !categoryEnhancePromptMapOK(starOnly, want) {
|
||
|
|
t.Fatal("*-only repaired map should be OK (idempotent)")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRepairTargetsUseSharedTemplate(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
tpl := strings.TrimSpace(aiprompts.CategoryEnhanceUserTemplate)
|
||
|
|
for _, v := range []string{"name", "description", "attrs", "category", "language"} {
|
||
|
|
if !strings.Contains(tpl, "{{"+v+"}}") {
|
||
|
|
t.Fatalf("shared template missing {{%s}}", v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if a1CompanyID == "" || platformDemoCompanyName == "" {
|
||
|
|
t.Fatal("missing company target constants")
|
||
|
|
}
|
||
|
|
}
|