Function TestDrain_PruneTreeClearsOldBranches

pkg/drain/drain_test.go:10–41

(t *testing.T)

)

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

	inputLines := []string{
		"test test test 123",
		"test test test 456",
		"test test test 789",
		"test test test 101",
		"my name is 104",
		"my name is 105",
		"my name is 106",
		"my name is 107",
	}

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

	for _, line := range inputLines {
		drain.Train(line)
	}

	require.Len(t, drain.Clusters(), 2)
	require.Equal(t, 16, countNodes(drain.rootNode))

	// Delete a cluster manually
	drain.idToCluster.Remove(1)
	require.Len(t, drain.Clusters(), 1)
	require.Equal(t, 16, countNodes(drain.rootNode), "expected same number of nodes before pruning")

	drain.Prune()
	require.Len(t, drain.Clusters(), 1)
	require.Equal(t, 9, countNodes(drain.rootNode), "expected fewer nodes after pruning")
}

func countNodes(node *Node) int {
	total := 1