From e2c0fceabee7fa30909ae7d1e57ac69708299edc Mon Sep 17 00:00:00 2001 From: Alan Protasio Date: Wed, 11 Dec 2024 17:42:07 -0800 Subject: [PATCH] Change query fuzz test for bottomk, topk: take 2 (#6420) Signed-off-by: alanprot --- integration/query_fuzz_test.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/integration/query_fuzz_test.go b/integration/query_fuzz_test.go index 72ae8ce330..1e1485f219 100644 --- 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++