114 lines
3.3 KiB
Go
114 lines
3.3 KiB
Go
package profile
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/scriptorium/internal/domain"
|
|
)
|
|
|
|
func TestFilesystemRepository_GetProfile(t *testing.T) {
|
|
tmpDir, err := os.MkdirTemp("", "profile_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 profile", func(t *testing.T) {
|
|
p, err := repo.GetProfile(ctx, "test-profile", "")
|
|
if err != nil {
|
|
t.Fatalf("expected no error, got %v", err)
|
|
}
|
|
if p == nil || p.ID != "test-profile" {
|
|
t.Errorf("expected profile 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.ExpectedInputs) != 2 || p.ExpectedInputs[0] != "transcript" || p.ExpectedInputs[1] != "glossary" {
|
|
t.Fatalf("unexpected expected_inputs: %#v", p.ExpectedInputs)
|
|
}
|
|
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.ModelDefaults.TimeoutSeconds != 120 {
|
|
t.Fatalf("expected timeout_seconds 120, got %d", p.ModelDefaults.TimeoutSeconds)
|
|
}
|
|
})
|
|
|
|
t.Run("invalid YAML", func(t *testing.T) {
|
|
_, err := repo.GetProfile(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.GetProfile(ctx, "missing-id", "")
|
|
if !errors.Is(err, ErrProfileNotFound) {
|
|
t.Errorf("expected ErrProfileNotFound for profile with missing ID, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("no templates", func(t *testing.T) {
|
|
_, err := repo.GetProfile(ctx, "no-templates", "")
|
|
if !errors.Is(err, ErrInvalidProfile) {
|
|
t.Errorf("expected ErrInvalidProfile for profile with no templates, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("json schema mode missing schema path", func(t *testing.T) {
|
|
_, err := repo.GetProfile(ctx, "json-schema-missing-path", "")
|
|
if !errors.Is(err, ErrInvalidProfile) {
|
|
t.Errorf("expected ErrInvalidProfile for json_schema profile without schema_path, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("negative timeout seconds", func(t *testing.T) {
|
|
_, err := repo.GetProfile(ctx, "negative-timeout", "")
|
|
if !errors.Is(err, ErrInvalidProfile) {
|
|
t.Errorf("expected ErrInvalidProfile for negative timeout_seconds, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("profile not found", func(t *testing.T) {
|
|
_, err := repo.GetProfile(ctx, "unknown", "")
|
|
if !errors.Is(err, ErrProfileNotFound) {
|
|
t.Errorf("expected ErrProfileNotFound, got %v", err)
|
|
}
|
|
})
|
|
}
|