Function TestDrain_Train_CacheEvictionIntegration

pkg/drain/drain_test.go:162–194

(t *testing.T)

}

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

	config := DefaultConfig()
	config.MaxClusters = 3 // Small cache size to trigger evictions
	config.StaleClusterAge = 1 * time.Hour
	drain := New("test-tenant", config)

	// Fill cache beyond max size
	spanNames := []string{
		"GET /api/users/1",
		"GET /api/users/2",
		"GET /api/posts/3",
		"GET /api/posts/4",
		"GET /api/comments/5",
	}

	for _, spanName := range spanNames {
		cluster := drain.Train(spanName)
		require.NotNil(t, cluster)
	}

	// After filling cache, some clusters may have been evicted
	allClusters := drain.Clusters()
	require.LessOrEqual(t, len(allClusters), config.MaxClusters, "cache should respect max size")

	// Prune should clean up references to evicted clusters
	drain.Prune()

	// After pruning, clusters count should still be <= max
	allClustersAfterPrune := drain.Clusters()
	require.LessOrEqual(t, len(allClustersAfterPrune), config.MaxClusters)
}

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