From e1b56ad6a5e8f7ffcb28e9fb0563564b091c9eda Mon Sep 17 00:00:00 2001 From: Karan Kumar Date: Fri, 13 Oct 2023 16:20:05 +0530 Subject: [PATCH] Fixing the flaky tests. (#15142) --- .../org/apache/druid/msq/exec/MSQSelectTest.java | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/exec/MSQSelectTest.java b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/exec/MSQSelectTest.java index 000e3e7ebe79..030f352633f8 100644 --- a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/exec/MSQSelectTest.java +++ b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/exec/MSQSelectTest.java @@ -1463,22 +1463,21 @@ public void testExternSelectWithMultipleWorkers() throws IOException ); // adding result stage counter checks if (isPageSizeLimited()) { - selectTester = selectTester.setExpectedCountersForStageWorkerChannel( + selectTester.setExpectedCountersForStageWorkerChannel( CounterSnapshotMatcher - .with().rows(2, 0, 2).frames(1, 0, 1), + .with().rows(2, 0, 2), 1, 0, "input0" ).setExpectedCountersForStageWorkerChannel( CounterSnapshotMatcher - .with().rows(2, 0, 2).frames(1, 0, 1), + .with().rows(2, 0, 2), 1, 0, "output" - ); - selectTester = selectTester.setExpectedCountersForStageWorkerChannel( + ).setExpectedCountersForStageWorkerChannel( CounterSnapshotMatcher - .with().rows(0, 2, 0, 4).frames(0, 1, 0, 1), + .with().rows(0, 2, 0, 4), 1, 1, "input0" ).setExpectedCountersForStageWorkerChannel( CounterSnapshotMatcher - .with().rows(0, 2, 0, 4).frames(0, 1, 0, 1), + .with().rows(0, 2, 0, 4), 1, 1, "output" ); }