package subagents

import "testing"

func TestAgentAttemptValueAndTranscriptAccounting(t *testing.T) {
	a := &Agent{maxOutputBytes: 6, maxOutputLines: 10}
	if got := a.AttemptValue(); got != 0 {
		t.Fatalf("initial attempt = %d, want 0", got)
	}
	if got := a.incrementAttempt(); got != 1 || a.AttemptValue() != 1 {
		t.Fatalf("attempt after increment = %d, value = %d; want 1", got, a.AttemptValue())
	}

	a.appendTranscript("a\nbb")
	if got := a.outputBytes; got != 5 {
		t.Fatalf("output bytes after append = %d, want 5", got)
	}
	if got := a.outputLines; got != 2 {
		t.Fatalf("output lines after append = %d, want 2", got)
	}
	a.appendTranscript("ccc")
	if got, want := a.Transcript(), []string{"ccc"}; len(got) != 1 || got[0] != want[0] {
		t.Fatalf("transcript after byte eviction = %q, want %q", got, want)
	}
	if a.outputBytes != 4 || a.outputLines != 1 {
		t.Fatalf("accounting after byte eviction = (%d bytes, %d lines), want (4, 1)", a.outputBytes, a.outputLines)
	}

	a.maxOutputBytes = 1
	a.appendTranscript("z")
	if got := a.Transcript(); len(got) != 0 {
		t.Fatalf("transcript after evicting every line = %q, want empty", got)
	}
	if a.outputBytes != 0 || a.outputLines != 0 {
		t.Fatalf("empty accounting = (%d bytes, %d lines), want (0, 0)", a.outputBytes, a.outputLines)
	}
}

func TestAgentTranscriptLineCapAccounting(t *testing.T) {
	a := &Agent{maxOutputBytes: 100, maxOutputLines: 2}
	a.appendTranscript("one\ntwo\nthree")

	if got, want := a.Transcript(), []string{"two", "three"}; len(got) != len(want) || got[0] != want[0] || got[1] != want[1] {
		t.Fatalf("transcript = %q, want %q", got, want)
	}
	if a.outputBytes != len("two")+1+len("three")+1 || a.outputLines != 2 {
		t.Fatalf("line-cap accounting = (%d bytes, %d lines), want (10, 2)", a.outputBytes, a.outputLines)
	}
}