test: fix flaky test in comsumerTest (#5846)

Co-authored-by: Cai, Zhouxiang <zc31@illinois.edu>
Co-authored-by: Zhenxu Ke <kezhenxu94@apache.org>
This commit is contained in:
zcai2 2020-11-16 02:19:47 -06:00 committed by GitHub
parent 832582427a
commit 48479b74df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -35,6 +35,7 @@ public class ConsumerTest {
@Test
public void testConsumerLessThanChannel() throws IllegalAccessException {
final DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(2, 100);
for (int i = 0; i < 100; i++) {
@ -56,6 +57,7 @@ public class ConsumerTest {
@Test
public void testConsumerMoreThanChannel() throws IllegalAccessException, InterruptedException {
BUFFER.drainTo(new ArrayList<SampleData>());
final DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(2, 100);
for (int i = 0; i < 200; i++) {