agent-orchestrator/backend/internal/review/prompt_test.go

38 lines
1.1 KiB
Go

package review
import (
"strings"
"testing"
"github.com/aoagents/agent-orchestrator/backend/internal/ports"
)
func TestReviewTextsIncludesMultiPRQueue(t *testing.T) {
spec := launchSpec()
spec.RunID = "run-2"
spec.PRURL = "https://github.com/o/r/pull/2"
spec.TargetSHA = "sha2"
spec.ReviewIndex = 1
spec.ReviewQueue = []ports.ReviewTask{
{RunID: "run-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1"},
{RunID: "run-2", PRURL: "https://github.com/o/r/pull/2", TargetSHA: "sha2"},
}
prompt, _ := reviewTexts(spec)
for _, want := range []string{
"AO created 2 review tasks",
"Review every queued PR, then submit all results together",
"Complete every review task in the queue autonomously",
"Do not ask the user whether to continue to the next PR",
"* 1. https://github.com/o/r/pull/1 (head commit sha1, run run-1)",
"* 2. https://github.com/o/r/pull/2 (head commit sha2, run run-2)",
"After every PR has its own GitHub review from step 1",
"ao review submit --session mer-1 --reviews -",
`"reviews": [`,
} {
if !strings.Contains(prompt, want) {
t.Fatalf("prompt missing %q:\n%s", want, prompt)
}
}
}