Function TestGetTimelineFiltered

pkg/events/bus_test.go:219–238

TestGetTimelineFiltered 测试过滤获取时间线

(t *testing.T)

// TestGetTimelineFiltered 测试过滤获取时间线
func TestGetTimelineFiltered(t *testing.T) {
	eb := NewEventBus()
	defer eb.Close()

	// 添加不同类型的事件
	for i := range 5 {
		eb.EmitProgress(&types.ProgressTextChunkEvent{Step: i, Delta: "test"})
	}
	eb.EmitProgress(&types.ProgressDoneEvent{})

	// 过滤只获取 ProgressDoneEvent
	events := eb.GetTimelineFiltered(func(env types.AgentEventEnvelope) bool {
		_, ok := env.Event.(*types.ProgressDoneEvent)
		return ok
	})

	if len(events) != 1 {
		t.Errorf("expected 1 done event, got %d", len(events))
	}
}

// TestGetTimelineCount 测试获取事件数量
func TestGetTimelineCount(t *testing.T) {