Use ResetSystemProperties in JobCancellationSuite

apache · Dec 19, 2014 · 60a63a1 · 60a63a1
1 parent 14a92e4
commit 60a63a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/test/scala/org/apache/spark/JobCancellationSuite.scala b/core/src/test/scala/org/apache/spark/JobCancellationSuite.scala
@@ -28,19 +28,19 @@ import org.scalatest.{BeforeAndAfter, FunSuite}
 import org.scalatest.Matchers
 
 import org.apache.spark.scheduler.{SparkListener, SparkListenerTaskStart}
+import org.apache.spark.util.ResetSystemProperties
 
 /**
  * Test suite for cancelling running jobs. We run the cancellation tasks for single job action
  * (e.g. count) as well as multi-job action (e.g. take). We test the local and cluster schedulers
  * in both FIFO and fair scheduling modes.
  */
 class JobCancellationSuite extends FunSuite with Matchers with BeforeAndAfter
-  with LocalSparkContext {
+  with ResetSystemProperties with LocalSparkContext {
 
   override def afterEach() {
     super.afterEach()
     resetSparkContext()
-    System.clearProperty("spark.scheduler.mode")
   }
 
   test("local mode, FIFO scheduler") {