Function TestPool_Remove

pkg/core/pool_test.go:218–259

TestPool_Remove 测试移除 Agent

(t *testing.T)

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

	ctx := context.Background()
	config := createTestConfig("test-agent")

	// 创建 Agent
	_, err := pool.Create(ctx, config)
	if err != nil {
		t.Fatalf("Failed to create agent: %v", err)
	}

	// 验证存在
	if pool.Size() != 1 {
		t.Error("Agent not in pool")
	}

	// 移除 Agent
	err = pool.Remove("test-agent")
	if err != nil {
		t.Fatalf("Failed to remove agent: %v", err)
	}

	// 验证已移除
	if pool.Size() != 0 {
		t.Error("Agent still in pool after removal")
	}

	_, exists := pool.Get("test-agent")
	if exists {
		t.Error("Agent still retrievable after removal")
	}
}

// TestPool_Status 测试获取 Agent 状态
func TestPool_Status(t *testing.T) {