ops: migrate legacy event sequences safely

internal/store/migration_test.go

@@ -0,0 +1,58 @@
+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")
+	}
+}