Function TestGetMiddlewareForSubAgent

pkg/middleware/subagent_enhanced_test.go:279–312

TestGetMiddlewareForSubAgent 测试获取子代理中间件栈

(t *testing.T)

// TestGetMiddlewareForSubAgent 测试获取子代理中间件栈
func TestGetMiddlewareForSubAgent(t *testing.T) {
	factory := func(ctx context.Context, spec SubAgentSpec) (SubAgent, error) {
		return NewSimpleSubAgent(spec.Name, spec.Prompt, nil), nil
	}

	middleware, err := NewSubAgentMiddleware(&SubAgentMiddlewareConfig{
		Specs:                []SubAgentSpec{},
		Factory:              factory,
		EnableGeneralPurpose: true,
	})
	if err != nil {
		t.Fatalf("Failed to create middleware: %v", err)
	}

	parentMiddlewares := []Middleware{
		NewBaseMiddleware("parent1", 100),
		NewBaseMiddleware("parent2", 200),
	}

	// 测试获取中间件栈
	spec := SubAgentSpec{
		Name:               "test",
		InheritMiddlewares: true,
		MiddlewareOverrides: []Middleware{
			NewBaseMiddleware("custom", 300),
		},
	}

	stack := middleware.GetMiddlewareForSubAgent(spec, parentMiddlewares)

	if len(stack) != 3 {
		t.Errorf("Expected 3 middlewares, got %d", len(stack))
	}
}