83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
package consumers_test
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
configkit "gitea.maximumdirect.net/eric/ConfigKit"
|
|
"gitea.maximumdirect.net/eric/ConfigKit/examples/consumers"
|
|
)
|
|
|
|
func Example_identifiedPromptCatalog() {
|
|
result, err := consumers.LoadPromptCatalog(fixture("prompt-catalog", "config.yml"))
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
ids := make([]string, 0, len(result.Config.Prompts))
|
|
for _, prompt := range result.Config.Prompts {
|
|
ids = append(ids, prompt.ID)
|
|
}
|
|
fmt.Println(ids, err)
|
|
// Output: [main curated weather welcome] <nil>
|
|
}
|
|
|
|
func TestPromptCatalogImportsRegisteredItemsAndReplacesDefaults(t *testing.T) {
|
|
result, err := consumers.LoadPromptCatalog(fixture("prompt-catalog", "config.yml"))
|
|
if err != nil {
|
|
var loadError *configkit.LoadError
|
|
if errors.As(err, &loadError) {
|
|
t.Fatalf("load prompt catalog: %v (%#v)", err, loadError.Diagnostics())
|
|
}
|
|
t.Fatal(err)
|
|
}
|
|
if len(result.Config.Prompts) != 4 {
|
|
t.Fatalf("prompt records = %#v", result.Config.Prompts)
|
|
}
|
|
ids := []string{"main", "curated", "weather", "welcome"}
|
|
for index, id := range ids {
|
|
if result.Config.Prompts[index].ID != id {
|
|
t.Fatalf("prompt %d ID = %q, want %q", index, result.Config.Prompts[index].ID, id)
|
|
}
|
|
}
|
|
for _, prompt := range result.Config.Prompts {
|
|
if prompt.ID == "default" {
|
|
t.Fatalf("lower-priority default survived main sequence replacement: %#v", result.Config.Prompts)
|
|
}
|
|
}
|
|
|
|
var itemImports, helperImports int
|
|
for _, source := range result.Sources() {
|
|
switch filepath.Base(source.Path) {
|
|
case "curated.yml", "generated-weather.yml", "generated-welcome.yml":
|
|
if source.PlacementMode != configkit.PlacementModeSequenceItem || source.Destination.String() != "/prompts" {
|
|
t.Fatalf("item source placement = %#v", source)
|
|
}
|
|
itemImports++
|
|
case "curated-id.yml":
|
|
if source.PlacementMode != configkit.PlacementModeMapping || source.ImportedFrom == nil {
|
|
t.Fatalf("helper source = %#v", source)
|
|
}
|
|
helperImports++
|
|
}
|
|
}
|
|
if itemImports != 3 || helperImports != 1 {
|
|
t.Fatalf("item imports = %d, helper imports = %d, sources = %#v", itemImports, helperImports, result.Sources())
|
|
}
|
|
if result.Config.Prompts[1].ID != "curated" {
|
|
t.Fatalf("helper did not provide prompt ID: %#v", result.Config.Prompts[1])
|
|
}
|
|
}
|
|
|
|
func TestPromptCatalogLeavesDomainValidationWithConsumer(t *testing.T) {
|
|
_, err := consumers.LoadPromptCatalog(fixture("prompt-catalog", "invalid.yml"))
|
|
var loadError *configkit.LoadError
|
|
if !errors.As(err, &loadError) || len(loadError.Diagnostics()) != 1 ||
|
|
loadError.Diagnostics()[0].Code != configkit.CodeValidateError ||
|
|
loadError.Diagnostics()[0].Path.String() != "/prompts/0/messages" {
|
|
t.Fatalf("consumer domain validation error = %v", err)
|
|
}
|
|
}
|