Function TestAgentEventBus

pkg/agent/agent_test.go:83–132

(t *testing.T)

}

func TestAgentEventBus(t *testing.T) {
	deps := setupTestDeps(t)

	config := &types.AgentConfig{
		TemplateID: "test-template",
		ModelConfig: &types.ModelConfig{
			Provider: "anthropic",
			Model:    "claude-sonnet-4-5",
			APIKey:   "test-key",
		},
		Sandbox: &types.SandboxConfig{
			Kind:    types.SandboxKindMock,
			WorkDir: "/tmp/test",
		},
	}

	ag, err := Create(context.Background(), config, deps)
	if err != nil {
		t.Fatalf("Failed to create agent: %v", err)
	}
	defer func() { _ = ag.Close() }()

	// 订阅事件
	eventCh := ag.Subscribe([]types.AgentChannel{types.ChannelMonitor}, nil)

	// 发送一个测试事件
	go func() {
		time.Sleep(100 * time.Millisecond)
		ag.eventBus.EmitMonitor(&types.MonitorStateChangedEvent{
			State: types.AgentStateWorking,
		})
	}()

	// 等待事件
	select {
	case envelope := <-eventCh:
		if envelope.Event == nil {
			t.Error("Event should not be nil")
		}
		if evt, ok := envelope.Event.(*types.MonitorStateChangedEvent); ok {
			if evt.State != types.AgentStateWorking {
				t.Errorf("Expected state Working, got %s", evt.State)
			}
		} else {
			t.Error("Event should be MonitorStateChangedEvent")
		}
	case <-time.After(1 * time.Second):
		t.Error("Timeout waiting for event")
	}
}

// setupTestDeps 创建测试依赖
func setupTestDeps(t *testing.T) *Dependencies {