Function TestDrain_Train_BasicPatternDetection

pkg/drain/drain_test.go:51–86

(t *testing.T)

}

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

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

	// Train with similar span names that should form a pattern
	spanNames := []string{
		"GET /api/users/123",
		"GET /api/users/456",
		"GET /api/users/789",
		"GET /api/users/101",
	}

	var clusters []*LogCluster
	for _, spanName := range spanNames {
		cluster := drain.Train(spanName)
		require.NotNil(t, cluster)
		clusters = append(clusters, cluster)
	}

	// After training multiple similar patterns, they should cluster together
	// All clusters should have the same pattern (with parameter markers)
	uniquePatterns := make(map[string]bool)
	for _, cluster := range clusters {
		pattern := cluster.String()
		uniquePatterns[pattern] = true
	}

	// Should have fewer unique patterns than input span names (clustering occurred)
	require.Less(t, len(uniquePatterns), len(spanNames), "patterns should be clustered")

	// Verify pattern contains parameter marker
	for pattern := range uniquePatterns {
		require.Contains(t, pattern, "<_>", "pattern should contain parameter marker")
	}
}

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