kami
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package store
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"orchestra/internal/domain"
|
|
)
|
|
|
|
func created(id string) domain.Event {
|
|
b, _ := json.Marshal(map[string]any{"source": "jsonl", "external_id": "42", "project": "demo", "capability": []string{"mechanical"}})
|
|
return domain.Event{ID: id, Type: "TaskCreated", TaskID: "task-1", Version: 1, Payload: b}
|
|
}
|
|
|
|
func TestAppendReplayAndDeduplicate(t *testing.T) {
|
|
dir := t.TempDir()
|
|
s, err := Open(dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := s.Append(created("e1")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := s.Append(created("e2")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(s.Events(0)); got != 1 {
|
|
t.Fatalf("duplicate ingest appended %d events", got)
|
|
}
|
|
ref, err := s.PutArtifact([]byte("report"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
completion, _ := json.Marshal(map[string]string{"report_ref": ref})
|
|
if err := s.Append(domain.Event{Type: "TaskCompleted", TaskID: "task-1", Version: 2, Payload: completion}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := s.Append(domain.Event{Type: "TaskReleased", TaskID: "task-1", Version: 2, Payload: json.RawMessage(`{"handoff_ref":"x"}`)}); err != domain.ErrConflict {
|
|
t.Fatalf("expected conflict, got %v", err)
|
|
}
|
|
s2, err := Open(dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := s2.Tasks()[0].State; got != domain.StateCompleted {
|
|
t.Fatalf("replay state = %s", got)
|
|
}
|
|
if _, err := os.Stat(filepath.Join(dir, "snapshot.json")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestArtifactIsContentAddressed(t *testing.T) {
|
|
s, err := Open(t.TempDir())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
h1, err := s.PutArtifact([]byte("proof"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
h2, err := s.PutArtifact([]byte("proof"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if h1 != h2 {
|
|
t.Fatal("same artifact received different hashes")
|
|
}
|
|
if _, err := os.Stat(filepath.Join(s.cas, h1)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|