Fix for SparkContext stop behavior

marmbrus · Jun 26, 2015 · c97839a · c97839a
1 parent 1aff39c
commit c97839a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/src/test/scala/org/apache/spark/SparkContextSuite.scala b/core/src/test/scala/org/apache/spark/SparkContextSuite.scala
@@ -275,13 +275,16 @@ class SparkContextSuite extends SparkFunSuite with LocalSparkContext {
   }
 
   test("calling multiple sc.stop() must not throw any exception") {
-    noException should be thrownBy {
+    try {
       sc = new SparkContext(new SparkConf().setAppName("test").setMaster("local"))
       val cnt = sc.parallelize(1 to 4).count()
       sc.cancelAllJobs()
       sc.stop()
       // call stop second time
       sc.stop()
+    } catch {
+      case e: Exception =>
+        fail("calling multiple sc.stop() must not have thrown any exception", e);
     }
   }
 }