Skip to content

Commit 5b3cb54

Browse files
committed
Use ResetSystemProperties in SparkListenerSuite
1 parent 0995c4b commit 5b3cb54

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

core/src/test/scala/org/apache/spark/scheduler/SparkListenerSuite.scala

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,10 @@ import org.scalatest.Matchers
2626

2727
import org.apache.spark.{LocalSparkContext, SparkContext}
2828
import org.apache.spark.executor.TaskMetrics
29+
import org.apache.spark.util.ResetSystemProperties
2930

30-
class SparkListenerSuite extends FunSuite with LocalSparkContext with Matchers
31-
with BeforeAndAfter with BeforeAndAfterAll {
31+
class SparkListenerSuite extends FunSuite with ResetSystemProperties with LocalSparkContext
32+
with Matchers with BeforeAndAfter with BeforeAndAfterAll {
3233

3334
/** Length of time to wait while draining listener events. */
3435
val WAIT_TIMEOUT_MILLIS = 10000
@@ -37,10 +38,6 @@ class SparkListenerSuite extends FunSuite with LocalSparkContext with Matchers
3738
sc = new SparkContext("local", "SparkListenerSuite")
3839
}
3940

40-
override def afterAll() {
41-
System.clearProperty("spark.akka.frameSize")
42-
}
43-
4441
test("basic creation and shutdown of LiveListenerBus") {
4542
val counter = new BasicJobCounter
4643
val bus = new LiveListenerBus

0 commit comments

Comments
 (0)