1package subagents
2
3import "testing"
4
5func TestAgentAttemptValueAndTranscriptAccounting(t *testing.T) {
6 a := &Agent{maxOutputBytes: 6, maxOutputLines: 10}
7 if got := a.AttemptValue(); got != 0 {
8 t.Fatalf("initial attempt = %d, want 0", got)
9 }
10 if got := a.incrementAttempt(); got != 1 || a.AttemptValue() != 1 {
11 t.Fatalf("attempt after increment = %d, value = %d; want 1", got, a.AttemptValue())
12 }
13
14 a.appendTranscript("a\nbb")
15 if got := a.outputBytes; got != 5 {
16 t.Fatalf("output bytes after append = %d, want 5", got)
17 }
18 if got := a.outputLines; got != 2 {
19 t.Fatalf("output lines after append = %d, want 2", got)
20 }
21 a.appendTranscript("ccc")
22 if got, want := a.Transcript(), []string{"ccc"}; len(got) != 1 || got[0] != want[0] {
23 t.Fatalf("transcript after byte eviction = %q, want %q", got, want)
24 }
25 if a.outputBytes != 4 || a.outputLines != 1 {
26 t.Fatalf("accounting after byte eviction = (%d bytes, %d lines), want (4, 1)", a.outputBytes, a.outputLines)
27 }
28
29 a.maxOutputBytes = 1
30 a.appendTranscript("z")
31 if got := a.Transcript(); len(got) != 0 {
32 t.Fatalf("transcript after evicting every line = %q, want empty", got)
33 }
34 if a.outputBytes != 0 || a.outputLines != 0 {
35 t.Fatalf("empty accounting = (%d bytes, %d lines), want (0, 0)", a.outputBytes, a.outputLines)
36 }
37}
38
39func TestAgentTranscriptLineCapAccounting(t *testing.T) {
40 a := &Agent{maxOutputBytes: 100, maxOutputLines: 2}
41 a.appendTranscript("one\ntwo\nthree")
42
43 if got, want := a.Transcript(), []string{"two", "three"}; len(got) != len(want) || got[0] != want[0] || got[1] != want[1] {
44 t.Fatalf("transcript = %q, want %q", got, want)
45 }
46 if a.outputBytes != len("two")+1+len("three")+1 || a.outputLines != 2 {
47 t.Fatalf("line-cap accounting = (%d bytes, %d lines), want (10, 2)", a.outputBytes, a.outputLines)
48 }
49}