Reduced time taken by graceful shutdown unit test.

witgo · Apr 7, 2014 · d0b8d65 · d0b8d65
1 parent f55bc67
commit d0b8d65
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/streaming/src/test/scala/org/apache/spark/streaming/StreamingContextSuite.scala b/streaming/src/test/scala/org/apache/spark/streaming/StreamingContextSuite.scala
@@ -178,7 +178,7 @@ class StreamingContextSuite extends FunSuite with BeforeAndAfter with Timeouts w
     val conf = new SparkConf().setMaster(master).setAppName(appName)
     conf.set("spark.cleaner.ttl", "3600")
     sc = new SparkContext(conf)
-    for (i <- 1 to 3) {
+    for (i <- 1 to 4) {
       logInfo("==================================")
       ssc = new StreamingContext(sc, batchDuration)
       var runningCount = 0
@@ -190,7 +190,7 @@ class StreamingContextSuite extends FunSuite with BeforeAndAfter with Timeouts w
         runningCount += count.toInt
       })
       ssc.start()
-      ssc.awaitTermination(2000)
+      ssc.awaitTermination(500)
       ssc.stop(stopSparkContext = false, stopGracefully = true)
       logInfo("Running count = " + runningCount)
       logInfo("TestReceiver.counter = " + TestReceiver.counter.get())