Function Test_Emitter_CalledOnChanges

x/acpio/acp_conversation_test.go:375–418

(t *testing.T)

}

func Test_Emitter_CalledOnChanges(t *testing.T) {
	mClock := quartz.NewMock(t)
	mock := newMockAgentIO()
	// Block the write so we can simulate chunks during processing
	started, done := mock.BlockWrite()

	emitter := newMockEmitter()

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	conv := acpio.NewACPConversation(ctx, mock, nil, nil, emitter, mClock)
	conv.Start(ctx)

	// Send blocks, so run in a goroutine
	errCh := make(chan error, 1)
	go func() { errCh <- conv.Send(screentracker.MessagePartText{Content: "test"}) }()

	// Wait for write to start
	<-started

	// Simulate chunks - each should trigger emitter calls
	mock.SimulateChunks("chunk1")
	mock.SimulateChunks("chunk2")

	emitter.mu.Lock()
	messagesCallsBeforeComplete := emitter.messagesCalls
	emitter.mu.Unlock()

	// Should have emit calls from chunks (each chunk emits messages, status, and screen)
	assert.Equal(t, 2, messagesCallsBeforeComplete)

	// Unblock the write to complete processing
	close(done)
	require.NoError(t, <-errCh)

	// Wait for completion emit
	emitter.WaitForStatus(ctx, t, screentracker.ConversationStatusStable)

	emitter.mu.Lock()
	finalMessagesCalls := emitter.messagesCalls
	emitter.mu.Unlock()
	assert.GreaterOrEqual(t, finalMessagesCalls, 3, "2 from chunks + 1 from completion")
}

func Test_InitialPrompt_SentOnStart(t *testing.T) {
	mClock := quartz.NewMock(t)