From 8f39124047ef017ef4cd6e161a595744e7a0595b Mon Sep 17 00:00:00 2001 From: cryptoe Date: Thu, 12 Oct 2023 21:10:49 +0530 Subject: [PATCH] Fixing the flaky tests. --- .../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 31ea64375199..30bbb850fbcd 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 @@ -1467,22 +1467,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" ); }