Function TestPool_List

pkg/core/pool_test.go:174–215

TestPool_List 测试列出 Agent

(t *testing.T)

// TestPool_List 测试列出 Agent
func TestPool_List(t *testing.T) {
	deps := createTestDeps(t)
	pool := NewPool(&PoolOptions{
		Dependencies: deps,
		MaxAgents:    10,
	})
	defer func() {
		if err := pool.Shutdown(); err != nil {
			t.Errorf("Shutdown failed: %v", err)
		}
	}()

	ctx := context.Background()

	// 创建不同前缀的 Agent
	agents := []string{"user-1", "user-2", "admin-1", "admin-2"}
	for _, agentID := range agents {
		config := createTestConfig(agentID)
		_, err := pool.Create(ctx, config)
		if err != nil {
			t.Fatalf("Failed to create agent %s: %v", agentID, err)
		}
	}

	// 列出所有 Agent
	allAgents := pool.List("")
	if len(allAgents) != 4 {
		t.Errorf("Expected 4 agents, got %d", len(allAgents))
	}

	// 列出 user- 前缀的 Agent
	userAgents := pool.List("user-")
	if len(userAgents) != 2 {
		t.Errorf("Expected 2 user agents, got %d", len(userAgents))
	}

	// 列出 admin- 前缀的 Agent
	adminAgents := pool.List("admin-")
	if len(adminAgents) != 2 {
		t.Errorf("Expected 2 admin agents, got %d", len(adminAgents))
	}
}

// TestPool_Remove 测试移除 Agent
func TestPool_Remove(t *testing.T) {