Code review

apache · Oct 9, 2018 · 16b23a8 · 16b23a8
1 parent 6a36250
commit 16b23a8
Showing 1 changed file with 2 additions and 3 deletions.
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,9 +332,7 @@ class KafkaSinkSuite extends StreamTest with SharedSQLContext with KafkaTest {
     var ex: Exception = null
     try {
       ex = intercept[StreamingQueryException] {
-        writer = createKafkaWriter(input.toDF(),
-          withTopic = Some(topic),
-          withOptions = Map("kafka.max.block.ms" -> "10000"))()
+        writer = createKafkaWriter(input.toDF(), withTopic = Some(topic))()
         input.addData("1", "2", "3", "4", "5")
         writer.processAllAvailable()
       }
@@ -429,6 +427,7 @@ class KafkaSinkSuite extends StreamTest with SharedSQLContext with KafkaTest {
         .format("kafka")
         .option("checkpointLocation", checkpointDir.getCanonicalPath)
         .option("kafka.bootstrap.servers", testUtils.brokerAddress)
+        .option("kafka.max.block.ms", "5000")
         .queryName("kafkaStream")
       withTopic.foreach(stream.option("topic", _))
       withOutputMode.foreach(stream.outputMode(_))