Function TestLoop_ChiefDoneEvent

internal/loop/loop_test.go:335–379

TestLoop_ChiefDoneEvent tests detection of event.

(t *testing.T)

// TestLoop_ChiefDoneEvent tests detection of <chief-done/> event.
func TestLoop_ChiefDoneEvent(t *testing.T) {
	l := NewLoop("/test/prd.json", "test", 5, testProvider)
	l.iteration = 1

	done := make(chan bool)
	var events []Event
	go func() {
		for event := range l.Events() {
			events = append(events, event)
			if event.Type == EventStoryDone {
				break
			}
		}
		done <- true
	}()

	// Simulate processing a line with chief-done
	r, w, _ := os.Pipe()
	go func() {
		w.WriteString(`{"type":"assistant","message":{"content":[{"type":"text","text":"All criteria pass! <chief-done/>"}]}}` + "\n")
		w.Close()
	}()

	l.processOutput(r)
	close(l.events)
	<-done

	// Check that we got a StoryDone event and sawStoryDone was set
	hasStoryDone := false
	for _, e := range events {
		if e.Type == EventStoryDone {
			hasStoryDone = true
		}
	}

	if !hasStoryDone {
		t.Error("Expected StoryDone event for <chief-done/>")
	}

	l.mu.Lock()
	if !l.sawStoryDone {
		t.Error("Expected sawStoryDone to be true after processing <chief-done/>")
	}
	l.mu.Unlock()
}

// TestLoop_SetMaxIterations tests setting max iterations at runtime.
func TestLoop_SetMaxIterations(t *testing.T) {