package store

import (
	"encoding/json"
	"os"
	"path/filepath"
	"testing"
	"time"

	"orchestra/internal/domain"
)

func TestNormalizeLegacyEventSequence(t *testing.T) {
	dir := t.TempDir()
	events := []domain.Event{
		{SchemaVersion: 2, Seq: 1, ID: "created", Type: "TaskCreated", TaskID: "t", Version: 1, At: time.Now().UTC(), Payload: json.RawMessage(`{"source":"test","external_id":"1","project":"p"}`), Surface: "system"},
		{SchemaVersion: 2, Seq: 1, ID: "leased", Type: "TaskLeased", TaskID: "t", Version: 2, At: time.Now().UTC(), Payload: json.RawMessage(`{"harness_id":"h","ttl":60,"until_ns":2000000000000000000,"expected_version":1}`), Surface: "system"},
		{SchemaVersion: 2, Seq: 2, ID: "recovered-lease", Type: "TaskLeased", TaskID: "t", Version: 2, At: time.Now().UTC(), Payload: json.RawMessage(`{"harness_id":"h","ttl":60,"until_ns":2000000000000000000,"expected_version":1}`), Surface: "system"},
	}
	var raw []byte
	for _, e := range events {
		b, err := json.Marshal(e)
		if err != nil {
			t.Fatal(err)
		}
		raw = append(raw, append(b, '\n')...)
	}
	if err := os.WriteFile(filepath.Join(dir, "events.jsonl"), raw, 0644); err != nil {
		t.Fatal(err)
	}
	changed, err := NormalizeLegacyEventSequence(dir)
	if err != nil || !changed {
		t.Fatalf("NormalizeLegacyEventSequence = %v, %v", changed, err)
	}
	if backups, err := filepath.Glob(filepath.Join(dir, "events.jsonl.legacy-*")); err != nil || len(backups) != 1 {
		t.Fatalf("backups = %v, %v", backups, err)
	}
	s, err := Open(dir)
	if err != nil {
		t.Fatal(err)
	}
	if got := len(s.Events(0)); got != 2 {
		t.Fatalf("events = %d, want 2", got)
	}
	if got := s.Events(0)[1].ID; got != "recovered-lease" {
		t.Fatalf("second event = %q, want recovered suffix", got)
	}
}

func TestNormalizeLegacyEventSequenceRejectsOtherCorruption(t *testing.T) {
	dir := t.TempDir()
	if err := os.WriteFile(filepath.Join(dir, "events.jsonl"), []byte(`{"seq":2}`+"\n"), 0644); err != nil {
		t.Fatal(err)
	}
	if _, err := NormalizeLegacyEventSequence(dir); err == nil {
		t.Fatal("NormalizeLegacyEventSequence accepted non-legacy corruption")
	}
}