From 312131cde7b4a3c86686b9cfd06b2d80e4ea214c Mon Sep 17 00:00:00 2001
From: Suraj Nath <9503187+electron0zero@users.noreply.github.com>
Date: Wed, 25 Sep 2024 23:55:06 +0530
Subject: [PATCH] test collect return and Exceeded

---
 pkg/collector/distinct_value_collector_test.go | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/pkg/collector/distinct_value_collector_test.go b/pkg/collector/distinct_value_collector_test.go
index 77b8f176bd0e..c56dcf930682 100644
--- a/pkg/collector/distinct_value_collector_test.go
+++ b/pkg/collector/distinct_value_collector_test.go
@@ -13,11 +13,16 @@ import (
 func TestDistinctValueCollector(t *testing.T) {
 	d := NewDistinctValue[string](10, func(s string) int { return len(s) })
 
-	d.Collect("123")
-	d.Collect("4567")
-	d.Collect("890")
+	var stop bool
+	stop = d.Collect("123")
+	require.False(t, stop)
+	stop = d.Collect("4567")
+	require.False(t, stop)
+	stop = d.Collect("890")
+	require.True(t, stop)
 
 	require.True(t, d.Exceeded())
+	require.Equal(t, stop, d.Exceeded()) // final stop should be same as Exceeded
 	require.Equal(t, []string{"123", "4567"}, d.Values())
 
 	// diff fails when diff is not enabled