Fixing the flaky tests. (#15142)

This commit is contained in:
Karan Kumar 2023-10-13 16:20:05 +05:30 committed by GitHub
parent 4deeb7e936
commit f0a70fe3c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1467,22 +1467,21 @@ public class MSQSelectTest extends MSQTestBase
); );
// adding result stage counter checks // adding result stage counter checks
if (isPageSizeLimited()) { if (isPageSizeLimited()) {
selectTester = selectTester.setExpectedCountersForStageWorkerChannel( selectTester.setExpectedCountersForStageWorkerChannel(
CounterSnapshotMatcher CounterSnapshotMatcher
.with().rows(2, 0, 2).frames(1, 0, 1), .with().rows(2, 0, 2),
1, 0, "input0" 1, 0, "input0"
).setExpectedCountersForStageWorkerChannel( ).setExpectedCountersForStageWorkerChannel(
CounterSnapshotMatcher CounterSnapshotMatcher
.with().rows(2, 0, 2).frames(1, 0, 1), .with().rows(2, 0, 2),
1, 0, "output" 1, 0, "output"
); ).setExpectedCountersForStageWorkerChannel(
selectTester = selectTester.setExpectedCountersForStageWorkerChannel(
CounterSnapshotMatcher CounterSnapshotMatcher
.with().rows(0, 2, 0, 4).frames(0, 1, 0, 1), .with().rows(0, 2, 0, 4),
1, 1, "input0" 1, 1, "input0"
).setExpectedCountersForStageWorkerChannel( ).setExpectedCountersForStageWorkerChannel(
CounterSnapshotMatcher CounterSnapshotMatcher
.with().rows(0, 2, 0, 4).frames(0, 1, 0, 1), .with().rows(0, 2, 0, 4),
1, 1, "output" 1, 1, "output"
); );
} }