[hotfix][tests] Add tests timeouts

fhalim · Sep 17, 2019 · a7102b7 · a7102b7
1 parent 15a7f97
commit a7102b7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/...t/src/test/java/org/apache/flink/formats/parquet/avro/ParquetStreamingFileSinkITCase.java b/...t/src/test/java/org/apache/flink/formats/parquet/avro/ParquetStreamingFileSinkITCase.java
@@ -39,7 +39,9 @@
 import org.apache.parquet.hadoop.ParquetReader;
 import org.apache.parquet.hadoop.util.HadoopInputFile;
 import org.apache.parquet.io.InputFile;
+import org.junit.Rule;
 import org.junit.Test;
+import org.junit.rules.Timeout;
 
 import java.io.File;
 import java.io.IOException;
@@ -62,6 +64,9 @@
 @SuppressWarnings("serial")
 public class ParquetStreamingFileSinkITCase extends AbstractTestBase {
 
+	@Rule
+	public final Timeout timeoutPerTest = Timeout.seconds(20);
+
 	@Test
 	public void testWriteParquetAvroSpecific() throws Exception {
 

diff --git a/.../src/test/java/org/apache/flink/formats/sequencefile/SequenceStreamingFileSinkITCase.java b/.../src/test/java/org/apache/flink/formats/sequencefile/SequenceStreamingFileSinkITCase.java
@@ -33,7 +33,9 @@
 import org.apache.hadoop.io.LongWritable;
 import org.apache.hadoop.io.SequenceFile;
 import org.apache.hadoop.io.Text;
+import org.junit.Rule;
 import org.junit.Test;
+import org.junit.rules.Timeout;
 
 import java.io.File;
 import java.io.IOException;
@@ -59,6 +61,9 @@ public class SequenceStreamingFileSinkITCase extends AbstractTestBase {
 			new Tuple2<>(3L, "c")
 	);
 
+	@Rule
+	public final Timeout timeoutPerTest = Timeout.seconds(20);
+
 	@Test
 	public void testWriteSequenceFile() throws Exception {
 		final File folder = TEMPORARY_FOLDER.newFolder();