Function TestDrain_Train_MultipleClusters

pkg/drain/drain_test.go:88–119

(t *testing.T)

}

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

	drain := New("test-tenant", DefaultConfig())

	// Train with different patterns
	spanNames := []string{
		"GET /api/users/123",
		"GET /api/users/456",
		"POST /api/posts/789",
		"POST /api/posts/101",
		"PUT /api/comments/202",
		"DELETE /api/comments/303",
	}

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

	allClusters := drain.Clusters()
	// Should have multiple distinct clusters (at least 2-3 different patterns)
	require.GreaterOrEqual(t, len(allClusters), 2, "should have multiple clusters")

	// Verify clusters have different patterns
	patterns := make(map[string]bool)
	for _, cluster := range allClusters {
		pattern := cluster.String()
		patterns[pattern] = true
	}
	require.GreaterOrEqual(t, len(patterns), 2, "should have multiple distinct patterns")
}

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