kami
5fb88724bd
Brief.Git was a single GitSync read from ORCHESTRA_DATA (never a git checkout), and completions were counted but discarded their report_ref/ receipt. Brief.Git is now keyed by project ID and built from each project's real repo; GitSync gained Ahead/Behind vs upstream; Brief now carries Receipts pulled from each TaskCompleted payload. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01W1rkJ2hBMybnJctPbcy4tT
70 lines
2.8 KiB
Go
70 lines
2.8 KiB
Go
package operations
|
|
|
|
import (
|
|
"encoding/json"
|
|
"orchestra/internal/authz"
|
|
"orchestra/internal/domain"
|
|
"orchestra/internal/store"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestBuildBrief(t *testing.T) {
|
|
now := time.Now()
|
|
p, _ := json.Marshal(map[string]any{"harness_id": "cc", "consumed": 12})
|
|
completed, _ := json.Marshal(map[string]any{"report_ref": "sha256:abc", "receipt": map[string]any{"numerator": 42.0}})
|
|
es := []domain.Event{{TaskID: "t1", Type: "TaskCompleted", At: now, Payload: completed}, {Type: "TaskFailed", At: now}, {Type: "ApprovalRequested", At: now}, {Type: "QuotaReported", At: now, Payload: p}}
|
|
b := BuildBrief(es, now.Add(-time.Minute), now.Add(time.Minute), map[string]GitSync{"proj": {Status: "clean"}})
|
|
if b.Completed != 1 || b.Failed != 1 || len(b.NeedsAttention) != 2 || b.Quota["cc"] != 12 {
|
|
t.Fatalf("unexpected brief: %+v", b)
|
|
}
|
|
if len(b.Receipts) != 1 || b.Receipts[0].TaskID != "t1" || b.Receipts[0].ReportRef != "sha256:abc" || b.Receipts[0].Receipt["numerator"] != 42.0 {
|
|
t.Fatalf("unexpected receipts: %+v", b.Receipts)
|
|
}
|
|
if b.Git["proj"].Status != "clean" {
|
|
t.Fatalf("unexpected git: %+v", b.Git)
|
|
}
|
|
}
|
|
|
|
func TestAggregateQuotaSumsRotationsAndWindows(t *testing.T) {
|
|
now := time.Now().UTC()
|
|
enc := func(at time.Time, n float64) domain.Event {
|
|
p, _ := json.Marshal(map[string]any{"harness_id": "codex", "consumed": n})
|
|
return domain.Event{Type: "QuotaReported", At: at, Payload: p, Surface: string(authz.System)}
|
|
}
|
|
es := []domain.Event{enc(now.Add(-2*time.Hour), 4), enc(now.Add(-time.Hour), 6), enc(now.Add(-48*time.Hour), 100)}
|
|
got := AggregateQuota(es, now.Add(-3*time.Hour), now)
|
|
if got["codex"] != 10 {
|
|
t.Fatalf("quota=%v, want 10", got)
|
|
}
|
|
}
|
|
|
|
func TestApplyAdvisoryRequiresApproval(t *testing.T) {
|
|
s, err := store.Open(t.TempDir())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
p, _ := json.Marshal(map[string]any{"source": "test", "external_id": "1", "project": "p", "title": "old"})
|
|
if err := s.Append(domain.Event{ID: "task", TaskID: "task", Type: "TaskCreated", Version: 1, Payload: p, Surface: string(authz.System)}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
ap, _ := json.Marshal(map[string]any{"items": []StandupItem{{TaskID: "task", Title: "new"}}})
|
|
adv := domain.Event{ID: "adv", TaskID: "system", Type: "StandupAdvisory", Payload: ap, Surface: string(authz.System)}
|
|
if err := s.Append(adv); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := ApplyAdvisory(s, "adv"); err == nil {
|
|
t.Fatal("unapproved advisory applied")
|
|
}
|
|
grant, _ := json.Marshal(map[string]any{"subject_ref": "adv"})
|
|
if err := s.Append(domain.Event{ID: "grant", TaskID: "system", Type: "ApprovalGranted", Payload: grant, Surface: string(authz.System)}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := ApplyAdvisory(s, "adv"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if task, _ := s.Task("task"); task.Title != "new" {
|
|
t.Fatalf("title=%q", task.Title)
|
|
}
|
|
}
|