[SPARK-25615] Improve the test runtime of KafkaSinkSuite: streaming -…

… write to non-existing topic
apache · Oct 8, 2018 · 6a36250 · 6a36250
1 parent ebd899b
commit 6a36250
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/external/kafka-0-10-sql/src/test/scala/org/apache/spark/sql/kafka010/KafkaSinkSuite.scala b/external/kafka-0-10-sql/src/test/scala/org/apache/spark/sql/kafka010/KafkaSinkSuite.scala
@@ -332,7 +332,9 @@ class KafkaSinkSuite extends StreamTest with SharedSQLContext with KafkaTest {
     var ex: Exception = null
     try {
       ex = intercept[StreamingQueryException] {
-        writer = createKafkaWriter(input.toDF(), withTopic = Some(topic))()
+        writer = createKafkaWriter(input.toDF(),
+          withTopic = Some(topic),
+          withOptions = Map("kafka.max.block.ms" -> "10000"))()
         input.addData("1", "2", "3", "4", "5")
         writer.processAllAvailable()
       }