Function TestFullStack_Integration

pkg/middleware/integration_test.go:121–178

TestFullStack_Integration 集成测试: 完整 Middleware Stack

(t *testing.T)

// TestFullStack_Integration 集成测试: 完整 Middleware Stack
func TestFullStack_Integration(t *testing.T) {
	ctx := context.Background()

	// 1. 创建 Backend
	backend := backends.NewStateBackend()

	// 2. 创建 FilesystemMiddleware
	fsMiddleware := NewFilesystemMiddleware(&FilesystemMiddlewareConfig{
		Backend: backend,
	})

	// 3. 创建 SubAgentMiddleware
	subagentMiddleware, err := NewSubAgentMiddleware(&SubAgentMiddlewareConfig{
		Specs: []SubAgentSpec{
			{Name: "researcher", Description: "Research agent"},
			{Name: "coder", Description: "Coding agent"},
		},
		Factory: func(ctx context.Context, spec SubAgentSpec) (SubAgent, error) {
			return NewSimpleSubAgent(spec.Name, spec.Prompt, nil), nil
		},
	})
	if err != nil {
		t.Fatalf("Failed to create SubAgentMiddleware: %v", err)
	}

	// 4. 创建 Stack
	stack := NewStack([]Middleware{
		fsMiddleware,
		subagentMiddleware,
	})

	// 5. 验证工具总数 (6 个文件工具 + 1 个 task 工具)
	tools := stack.Tools()
	if len(tools) != 7 {
		t.Errorf("Expected 7 tools, got %d", len(tools))
	}

	// 6. 验证中间件顺序(按优先级)
	middlewares := stack.Middlewares()
	if len(middlewares) != 2 {
		t.Fatalf("Expected 2 middlewares, got %d", len(middlewares))
	}

	// FilesystemMiddleware (priority: 100) 应该在前
	if middlewares[0].Name() != "filesystem" {
		t.Errorf("First middleware should be 'filesystem', got '%s'", middlewares[0].Name())
	}

	// SubAgentMiddleware (priority: 200) 应该在后
	if middlewares[1].Name() != "subagent" {
		t.Errorf("Second middleware should be 'subagent', got '%s'", middlewares[1].Name())
	}

	// 7. 清理
	if err := stack.OnAgentStop(ctx, "test"); err != nil {
		t.Errorf("Stack cleanup failed: %v", err)
	}
}