Change query fuzz test for bottomk, topk: take 2 (#6420)

Signed-off-by: alanprot <[email protected]>
cortexproject · Dec 12, 2024 · e2c0fce · e2c0fce
1 parent 75525a9
commit e2c0fce
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/integration/query_fuzz_test.go b/integration/query_fuzz_test.go
@@ -312,6 +312,11 @@ func TestDisableChunkTrimmingFuzz(t *testing.T) {
 				t.Logf("case %d error mismatch.\n%s: %s\nerr1: %v\nerr2: %v\n", i, qt, tc.query, tc.err1, tc.err2)
 				failures++
 			}
+		} else if shouldUseSampleNumComparer(tc.query) {
+			if !cmp.Equal(tc.res1, tc.res2, sampleNumComparer) {
+				t.Logf("case %d # of samples mismatch.\n%s: %s\nres1: %s\nres2: %s\n", i, qt, tc.query, tc.res1.String(), tc.res2.String())
+				failures++
+			}
 		} else if !cmp.Equal(tc.res1, tc.res2, comparer) {
 			t.Logf("case %d results mismatch.\n%s: %s\nres1: %s\nres2: %s\n", i, qt, tc.query, tc.res1.String(), tc.res2.String())
 			failures++
@@ -508,6 +513,11 @@ func TestExpandedPostingsCacheFuzz(t *testing.T) {
 					t.Logf("case %d error mismatch.\n%s: %s\nerr1: %v\nerr2: %v\n", i, qt, tc.query, tc.err1, tc.err2)
 					failures++
 				}
+			} else if shouldUseSampleNumComparer(tc.query) {
+				if !cmp.Equal(tc.res1, tc.res2, sampleNumComparer) {
+					t.Logf("case %d # of samples mismatch.\n%s: %s\nres1: %s\nres2: %s\n", i, qt, tc.query, tc.res1.String(), tc.res2.String())
+					failures++
+				}
 			} else if !cmp.Equal(tc.res1, tc.res2, comparer) {
 				t.Logf("case %d results mismatch.\n%s: %s\nres1: %s\nres2: %s\n", i, qt, tc.query, tc.res1.String(), tc.res2.String())
 				failures++