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" ); }