Function TestSubAgentMiddleware_Integration

pkg/middleware/integration_test.go:52–118

TestSubAgentMiddleware_Integration 集成测试: SubAgent Middleware

(t *testing.T)

// TestSubAgentMiddleware_Integration 集成测试: SubAgent Middleware
func TestSubAgentMiddleware_Integration(t *testing.T) {
	ctx := context.Background()

	// 1. 定义子代理规格
	specs := []SubAgentSpec{
		{
			Name:        "test-agent",
			Description: "测试用子代理",
			Prompt:      "Test prompt",
		},
	}

	// 2. 创建工厂
	factory := func(ctx context.Context, spec SubAgentSpec) (SubAgent, error) {
		execFn := func(ctx context.Context, description string, parentContext map[string]any) (string, error) {
			return "Test result: " + description, nil
		}
		return NewSimpleSubAgent(spec.Name, spec.Prompt, execFn), nil
	}

	// 3. 创建 SubAgentMiddleware
	middleware, err := NewSubAgentMiddleware(&SubAgentMiddlewareConfig{
		Specs:   specs,
		Factory: factory,
	})
	if err != nil {
		t.Fatalf("Failed to create SubAgentMiddleware: %v", err)
	}

	// 4. 测试工具
	tools := middleware.Tools()
	if len(tools) != 1 {
		t.Fatalf("Expected 1 tool, got %d", len(tools))
	}

	taskTool := tools[0]
	if taskTool.Name() != "task" {
		t.Errorf("Expected tool name 'task', got %s", taskTool.Name())
	}

	// 5. 测试 task 执行
	result, err := taskTool.Execute(ctx, map[string]any{
		"description":   "Test task",
		"subagent_type": "test-agent",
	}, nil)
	if err != nil {
		t.Fatalf("Task execution failed: %v", err)
	}

	resultMap, ok := result.(map[string]any)
	if !ok {
		t.Fatal("Result is not a map")
	}

	if !resultMap["ok"].(bool) {
		t.Errorf("Task should succeed, got ok=false")
	}