Skip to content

Commit 60a63a1

Browse files
committed
Use ResetSystemProperties in JobCancellationSuite
1 parent 14a92e4 commit 60a63a1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

core/src/test/scala/org/apache/spark/JobCancellationSuite.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,19 +28,19 @@ import org.scalatest.{BeforeAndAfter, FunSuite}
2828
import org.scalatest.Matchers
2929

3030
import org.apache.spark.scheduler.{SparkListener, SparkListenerTaskStart}
31+
import org.apache.spark.util.ResetSystemProperties
3132

3233
/**
3334
* Test suite for cancelling running jobs. We run the cancellation tasks for single job action
3435
* (e.g. count) as well as multi-job action (e.g. take). We test the local and cluster schedulers
3536
* in both FIFO and fair scheduling modes.
3637
*/
3738
class JobCancellationSuite extends FunSuite with Matchers with BeforeAndAfter
38-
with LocalSparkContext {
39+
with ResetSystemProperties with LocalSparkContext {
3940

4041
override def afterEach() {
4142
super.afterEach()
4243
resetSparkContext()
43-
System.clearProperty("spark.scheduler.mode")
4444
}
4545

4646
test("local mode, FIFO scheduler") {

0 commit comments

Comments
 (0)