kami
59 lines
2.1 KiB
Go
59 lines
2.1 KiB
Go
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")
|
|
}
|
|
}
|