Function TestLogClusterCache_Values

pkg/drain/log_cluster_cache_test.go:138–172

(t *testing.T)

}

func TestLogClusterCache_Values(t *testing.T) {
	t.Parallel()

	evictions := prometheus.NewCounter(prometheus.CounterOpts{Name: "evictions"})
	expired := prometheus.NewCounter(prometheus.CounterOpts{Name: "expired"})

	cache := newLogClusterCache(1*time.Hour, 100, evictions, expired)

	// Add multiple clusters
	clusters := []*LogCluster{
		{id: 1, Tokens: []string{"GET", "/users"}, ParamString: "<_>"},
		{id: 2, Tokens: []string{"POST", "/users"}, ParamString: "<_>"},
		{id: 3, Tokens: []string{"PUT", "/users"}, ParamString: "<_>"},
	}

	for _, cluster := range clusters {
		cache.Put(cluster)
	}

	// Values() should return all clusters
	values := []*LogCluster{}
	for cluster := range cache.Values() {
		values = append(values, cluster)
	}

	assert.Len(t, values, 3)
	// Verify all clusters are present
	ids := make(map[int]bool)
	for _, cluster := range values {
		ids[cluster.id] = true
	}
	assert.True(t, ids[1])
	assert.True(t, ids[2])
	assert.True(t, ids[3])
}