package spells

import (
	"encoding/json"
	"strings"
	"testing"
)

func TestLoadResponseSchemaForSpells(t *testing.T) {
	schema, err := loadResponseSchema()
	if err != nil {
		t.Fatalf("loadResponseSchema() error = %v, want nil", err)
	}
	if schema.Key != ResponseSchemaKey {
		t.Fatalf("schema.Key = %q, want %q", schema.Key, ResponseSchemaKey)
	}
	if schema.ID != ResponseSchemaID {
		t.Fatalf("schema.ID = %q, want %q", schema.ID, ResponseSchemaID)
	}
	if schema.Version != SchemaVersion {
		t.Fatalf("schema.Version = %q, want %q", schema.Version, SchemaVersion)
	}
	if schema.Name != ResponseSchemaName {
		t.Fatalf("schema.Name = %q, want %q", schema.Name, ResponseSchemaName)
	}
	if !strings.HasPrefix(schema.SHA256, "sha256:") {
		t.Fatalf("schema.SHA256 = %q, want sha256 prefix", schema.SHA256)
	}
	if !json.Valid(schema.JSONSchema) {
		t.Fatalf("schema.JSONSchema is invalid JSON: %s", schema.JSONSchema)
	}
}

func TestResponseSchemaJSONIsMutationSafe(t *testing.T) {
	first, err := loadResponseSchema()
	if err != nil {
		t.Fatalf("loadResponseSchema() error = %v, want nil", err)
	}
	first.JSONSchema[0] = '['

	second, err := loadResponseSchema()
	if err != nil {
		t.Fatalf("loadResponseSchema() error = %v, want nil", err)
	}
	if !json.Valid(second.JSONSchema) {
		t.Fatalf("schema JSON was mutated: %s", second.JSONSchema)
	}
	if len(second.JSONSchema) > 0 && second.JSONSchema[0] == '[' {
		t.Fatalf("schema JSON did not use defensive copy")
	}
}

func TestResponseSchemaDiagnosticsOmitRawSchema(t *testing.T) {
	schema, err := loadResponseSchema()
	if err != nil {
		t.Fatalf("loadResponseSchema() error = %v, want nil", err)
	}
	diagnostics := schema.DiagnosticsMap()

	if diagnostics["key"] != ResponseSchemaKey {
		t.Fatalf("diagnostics[key] = %#v, want %q", diagnostics["key"], ResponseSchemaKey)
	}
	for _, key := range []string{"id", "version", "name", "sha256"} {
		if diagnostics[key] == "" {
			t.Fatalf("diagnostics[%q] = %#v, want value", key, diagnostics[key])
		}
	}
	if _, ok := diagnostics["json_schema"]; ok {
		t.Fatalf("diagnostics should omit raw schema content: %#v", diagnostics)
	}
	if _, ok := diagnostics["JSONSchema"]; ok {
		t.Fatalf("diagnostics should omit raw schema content: %#v", diagnostics)
	}
}