107 lines
3.2 KiB
Go
107 lines
3.2 KiB
Go
package promptdef
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/scriptorium/internal/domain"
|
|
)
|
|
|
|
func TestFilesystemRepository_GetPromptDefinition(t *testing.T) {
|
|
tmpDir, err := os.MkdirTemp("", "promptdef_test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(tmpDir)
|
|
|
|
testDataDir := "testdata"
|
|
files, err := os.ReadDir(testDataDir)
|
|
if err != nil {
|
|
t.Fatalf("failed to read testdata: %v", err)
|
|
}
|
|
|
|
for _, f := range files {
|
|
src := filepath.Join(testDataDir, f.Name())
|
|
dst := filepath.Join(tmpDir, f.Name())
|
|
data, err := os.ReadFile(src)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(dst, data, 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
repo := NewFilesystemRepository(tmpDir)
|
|
ctx := context.Background()
|
|
|
|
t.Run("valid prompt definition", func(t *testing.T) {
|
|
p, err := repo.GetPromptDefinition(ctx, "test-profile", "")
|
|
if err != nil {
|
|
t.Fatalf("expected no error, got %v", err)
|
|
}
|
|
if p == nil || p.ID != "test-profile" {
|
|
t.Errorf("expected prompt definition test-profile, got %v", p)
|
|
}
|
|
if p.Version != "1.0.0" {
|
|
t.Fatalf("expected version 1.0.0, got %q", p.Version)
|
|
}
|
|
if len(p.Inputs) != 2 || p.Inputs[0].Name != "transcript" || p.Inputs[1].Name != "glossary" {
|
|
t.Fatalf("unexpected inputs: %#v", p.Inputs)
|
|
}
|
|
if len(p.Templates) != 2 {
|
|
t.Fatalf("expected 2 templates, got %d", len(p.Templates))
|
|
}
|
|
if p.Templates[0].Role != "system" || p.Templates[1].Role != "user" {
|
|
t.Fatalf("unexpected template roles: %#v", p.Templates)
|
|
}
|
|
if p.OutputFormat != domain.FormatMarkdown {
|
|
t.Fatalf("expected output format markdown, got %q", p.OutputFormat)
|
|
}
|
|
if p.Validation.ValidationMode != domain.ValidationBasic {
|
|
t.Fatalf("expected validation mode basic, got %q", p.Validation.ValidationMode)
|
|
}
|
|
if p.DefaultProfile != "test-exec" {
|
|
t.Fatalf("expected default profile test-exec, got %q", p.DefaultProfile)
|
|
}
|
|
})
|
|
|
|
t.Run("invalid YAML", func(t *testing.T) {
|
|
_, err := repo.GetPromptDefinition(ctx, "invalid_yaml", "")
|
|
if !errors.Is(err, ErrInvalidYAML) {
|
|
t.Errorf("expected ErrInvalidYAML, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("missing ID", func(t *testing.T) {
|
|
_, err := repo.GetPromptDefinition(ctx, "missing-id", "")
|
|
if !errors.Is(err, ErrPromptDefinitionNotFound) {
|
|
t.Errorf("expected ErrPromptDefinitionNotFound for profile with missing ID, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("no templates", func(t *testing.T) {
|
|
_, err := repo.GetPromptDefinition(ctx, "no-templates", "")
|
|
if !errors.Is(err, ErrInvalidPromptDefinition) {
|
|
t.Errorf("expected ErrInvalidPromptDefinition for profile with no templates, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("json schema mode missing schema path", func(t *testing.T) {
|
|
_, err := repo.GetPromptDefinition(ctx, "json-schema-missing-path", "")
|
|
if !errors.Is(err, ErrInvalidPromptDefinition) {
|
|
t.Errorf("expected ErrInvalidPromptDefinition for json_schema profile without schema_path, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("prompt definition not found", func(t *testing.T) {
|
|
_, err := repo.GetPromptDefinition(ctx, "unknown", "")
|
|
if !errors.Is(err, ErrPromptDefinitionNotFound) {
|
|
t.Errorf("expected ErrPromptDefinitionNotFound, got %v", err)
|
|
}
|
|
})
|
|
}
|