Use withSystemProperty in FileServerSuite

apache · Dec 19, 2014 · 14a92e4 · 14a92e4
1 parent 628f46c
commit 14a92e4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/core/src/test/scala/org/apache/spark/FileServerSuite.scala b/core/src/test/scala/org/apache/spark/FileServerSuite.scala
@@ -31,10 +31,11 @@ class FileServerSuite extends FunSuite with LocalSparkContext {
   @transient var tmpFile: File = _
   @transient var tmpJarUrl: String = _
 
-  override def beforeEach() {
-    super.beforeEach()
+  override def withFixture(test: NoArgTest) = {
     resetSparkContext()
-    System.setProperty("spark.authenticate", "false")
+    TestUtils.withSystemProperty("spark.authenticate", "false") {
+      super.withFixture(test)
+    }
   }
 
   override def beforeAll() {
@@ -52,7 +53,6 @@ class FileServerSuite extends FunSuite with LocalSparkContext {
     val jarFile = new File(testTempDir, "test.jar")
     val jarStream = new FileOutputStream(jarFile)
     val jar = new JarOutputStream(jarStream, new java.util.jar.Manifest())
-    System.setProperty("spark.authenticate", "false")
 
     val jarEntry = new JarEntry(textFile.getName)
     jar.putNextEntry(jarEntry)