105 lines
3.2 KiB
Go
105 lines
3.2 KiB
Go
package spells
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/santhosh-tekuri/jsonschema/v6"
|
|
)
|
|
|
|
func TestLoadResponseSchemaUsesExtractorOwnedLLMSchema(t *testing.T) {
|
|
schema, err := loadResponseSchema()
|
|
if err != nil {
|
|
t.Fatalf("loadResponseSchema() error = %v, want nil", err)
|
|
}
|
|
if schema.Key != ResponseSchemaKey || schema.ID != ResponseSchemaID || schema.Version != SchemaVersion || schema.Name != ResponseSchemaName {
|
|
t.Fatalf("schema identity = %#v, want maintained response identity", schema)
|
|
}
|
|
if !strings.HasPrefix(schema.SHA256, "sha256:") || !json.Valid(schema.JSONSchema) {
|
|
t.Fatalf("schema metadata = %#v, want valid hashed JSON", schema)
|
|
}
|
|
|
|
valid := validSpellsResponse()
|
|
validJSON, err := json.Marshal(valid)
|
|
if err != nil {
|
|
t.Fatalf("Marshal(validSpellsResponse()) error = %v, want nil", err)
|
|
}
|
|
if err := validateJSONSchema(validJSON, schema.JSONSchema); err != nil {
|
|
t.Fatalf("valid private spells response rejected: %v", err)
|
|
}
|
|
|
|
withCanonicalSourceID := validSpellsResponse()
|
|
withCanonicalSourceID["spell_casts"].([]any)[0].(map[string]any)["source_refs"].([]any)[0].(map[string]any)["source_id"] = "session-alpha"
|
|
content, err := json.Marshal(withCanonicalSourceID)
|
|
if err != nil {
|
|
t.Fatalf("Marshal(response with source_id) error = %v, want nil", err)
|
|
}
|
|
if err := validateJSONSchema(content, schema.JSONSchema); err == nil {
|
|
t.Fatal("validateJSONSchema() error = nil, want canonical source_id rejected")
|
|
}
|
|
}
|
|
|
|
func TestResponseSchemaJSONIsMutationSafe(t *testing.T) {
|
|
first, err := loadResponseSchema()
|
|
if err != nil {
|
|
t.Fatalf("loadResponseSchema() error = %v", err)
|
|
}
|
|
first.JSONSchema[0] = '['
|
|
second, err := loadResponseSchema()
|
|
if err != nil || !json.Valid(second.JSONSchema) || bytes.Equal(first.JSONSchema, second.JSONSchema) {
|
|
t.Fatalf("second schema = %s, %v; want defensive valid copy", second.JSONSchema, err)
|
|
}
|
|
}
|
|
|
|
func TestResponseSchemaDiagnosticsOmitRawSchema(t *testing.T) {
|
|
schema, err := loadResponseSchema()
|
|
if err != nil {
|
|
t.Fatalf("loadResponseSchema() error = %v", err)
|
|
}
|
|
diagnostics := schema.DiagnosticsMap()
|
|
if diagnostics["key"] != ResponseSchemaKey {
|
|
t.Fatalf("diagnostics = %#v, want response key", diagnostics)
|
|
}
|
|
if _, ok := diagnostics["json_schema"]; ok {
|
|
t.Fatalf("diagnostics include raw schema: %#v", diagnostics)
|
|
}
|
|
}
|
|
|
|
func validSpellsResponse() map[string]any {
|
|
return map[string]any{
|
|
"spell_casts": []any{
|
|
map[string]any{
|
|
"caster": "Aria",
|
|
"spell": "Cure Wounds",
|
|
"effect": "The wounds close.",
|
|
"narrative_description": "Aria casts Cure Wounds.",
|
|
"source_refs": []any{
|
|
map[string]any{"start_unit_id": 1, "end_unit_id": 2},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func validateJSONSchema(instanceContent, schemaContent []byte) error {
|
|
instance, err := jsonschema.UnmarshalJSON(bytes.NewReader(instanceContent))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
schemaDocument, err := jsonschema.UnmarshalJSON(bytes.NewReader(schemaContent))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
compiler := jsonschema.NewCompiler()
|
|
if err := compiler.AddResource("schema.json", schemaDocument); err != nil {
|
|
return err
|
|
}
|
|
schema, err := compiler.Compile("schema.json")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return schema.Validate(instance)
|
|
}
|