lanyuanxiaoyao/hudi
1
0
Fork 0
Code Activity

[HUDI-2515] Add close when producing records failed (#3746)

Browse Source
This commit is contained in:
董可伦
2021-11-02 19:43:20 +08:00
committed by GitHub
parent 2eaf0fd939
commit f239187da8
3 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
hudi-common/src/test/java/org/apache/hudi/common/util/TestParquetReaderIterator.java
View File

@@ -18,7 +18,7 @@
package org.apache.hudi.common.util;
import org.apache.hudi.exception.HoodieIOException;
import org.apache.hudi.exception.HoodieException;
import org.apache.parquet.hadoop.ParquetReader;
import org.junit.jupiter.api.Test;
@@ -30,6 +30,8 @@ import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
public class TestParquetReaderIterator {
@@ -59,6 +61,7 @@ public class TestParquetReaderIterator {
assertEquals(1, iterator.next());
// no more entries to iterate on
assertFalse(iterator.hasNext());
assertThrows(HoodieIOException.class, iterator::next, "should throw an exception since there is only 1 record");
assertThrows(HoodieException.class, iterator::next, "should throw an exception since there is only 1 record");
verify(reader, times(1)).close();
}
}