completion_test.go RawHistoryFolder

1package subagents
2
3import (
4	"context"
5	"strings"
6	"testing"
7)
8
9func TestFormatCompletionUpdateIncludesFinalSummary(t *testing.T) {
10	got := FormatCompletionUpdate([]Completion{{AgentID: "child", Status: "completed", Task: "review", Summary: "found the regression"}}, "")
11	if !strings.Contains(got, "final: found the regression") {
12		t.Fatalf("completion update = %q", got)
13	}
14}
15
16func TestCompletionTrackerDropsReportsAfterCancellationReset(t *testing.T) {
17	tracker := NewCompletionTracker()
18	if !tracker.TrackResident("child", "turn-1") {
19		t.Fatal("TrackResident returned false")
20	}
21	tracker.Reset()
22	if tracker.Report(Completion{AgentID: "child", TurnID: "turn-1"}) {
23		t.Fatal("Report returned true after reset")
24	}
25	got, err := tracker.WaitIdle(context.Background())
26	if err != nil {
27		t.Fatal(err)
28	}
29	if len(got) != 0 {
30		t.Fatalf("stale completions = %#v", got)
31	}
32}
33
34func TestCompletionTrackerMatchesTerminalReportsToAcceptedTurns(t *testing.T) {
35	tracker := NewCompletionTracker()
36	if !tracker.TrackResident("child", "turn-1") || !tracker.TrackResident("child", "turn-2") {
37		t.Fatal("TrackResident returned false")
38	}
39	if tracker.Report(Completion{AgentID: "child", TurnID: "other"}) {
40		t.Fatal("Report accepted an unknown turn")
41	}
42	if !tracker.Report(Completion{AgentID: "child", TurnID: "turn-1"}) {
43		t.Fatal("Report rejected an accepted turn")
44	}
45	if tracker.Report(Completion{AgentID: "child", TurnID: "turn-1"}) {
46		t.Fatal("Report accepted a duplicate terminal result")
47	}
48	if got := tracker.Pending(); got != 1 {
49		t.Fatalf("Pending = %d, want 1", got)
50	}
51	if !tracker.Report(Completion{AgentID: "child", TurnID: "turn-2"}) {
52		t.Fatal("Report rejected second accepted turn")
53	}
54	got, err := tracker.WaitIdle(context.Background())
55	if err != nil {
56		t.Fatal(err)
57	}
58	if len(got) != 2 {
59		t.Fatalf("completions = %#v, want 2", got)
60	}
61}