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
1 changed files with 6 additions and 7 deletions

View File

@ -1467,22 +1467,21 @@ public class MSQSelectTest extends MSQTestBase
);
// 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"
);
}