Function TestPool_ConcurrentAccess

pkg/core/pool_test.go:298–351

TestPool_ConcurrentAccess 测试并发访问

(t *testing.T)

// TestPool_ConcurrentAccess 测试并发访问
func TestPool_ConcurrentAccess(t *testing.T) {
	deps := createTestDeps(t)
	pool := NewPool(&PoolOptions{
		Dependencies: deps,
		MaxAgents:    100,
	})
	defer func() {
		if err := pool.Shutdown(); err != nil {
			t.Errorf("Shutdown failed: %v", err)
		}
	}()

	ctx := context.Background()
	concurrency := 50
	var wg sync.WaitGroup

	// 并发创建 Agent
	for i := range concurrency {
		wg.Add(1)
		go func(idx int) {
			defer wg.Done()

			config := createTestConfig("concurrent-agent-" + string(rune('0'+idx)))
			_, err := pool.Create(ctx, config)
			if err != nil {
				t.Logf("Failed to create agent %d: %v", idx, err)
			}
		}(i)
	}

	wg.Wait()

	// 验证池大小
	size := pool.Size()
	if size != concurrency {
		t.Logf("Expected %d agents, got %d (some creates may have failed)", concurrency, size)
	}

	// 并发读取 Agent
	for i := range concurrency {
		wg.Add(1)
		go func(idx int) {
			defer wg.Done()

			agentID := "concurrent-agent-" + string(rune('0'+idx))
			_, exists := pool.Get(agentID)
			if !exists {
				t.Logf("Agent %s not found", agentID)
			}
		}(i)
	}

	wg.Wait()
}

// TestPool_Shutdown 测试关闭池
func TestPool_Shutdown(t *testing.T) {