Fixing the flaky tests. (#15142) (#15155)

Co-authored-by: Karan Kumar <[email protected]>
apache · Oct 14, 2023 · 6f6fc62 · 6f6fc62
1 parent 9cfba45
commit 6f6fc62
Showing 1 changed file with 6 additions and 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
@@ -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"
       );
     }