fix tests

marmbrus · Nov 16, 2014 · 38c42f1 · 38c42f1
1 parent ab87958
commit 38c42f1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/core/src/main/scala/org/apache/spark/ui/ConsoleProgressBar.scala b/core/src/main/scala/org/apache/spark/ui/ConsoleProgressBar.scala
@@ -36,7 +36,11 @@ private[spark] class ConsoleProgressBar(sc: SparkContext) extends Logging {
   // Delay to show up a progress bar, in milli seconds
   val DELAY_SHOW_UP = 500L
   // The width of terminal
-  val TerminalWidth = sys.env.getOrElse("COLUMNS", "80").toInt
+  val TerminalWidth = if (!sys.env.getOrElse("COLUMNS", "").isEmpty) {
+    sys.env.get("COLUMNS").get.toInt
+  } else {
+    80
+  }
 
   @volatile var hasShowed = false
 

diff --git a/project/SparkBuild.scala b/project/SparkBuild.scala
@@ -377,7 +377,7 @@ object TestSettings {
     javaOptions in Test += "-Dspark.testing=1",
     javaOptions in Test += "-Dspark.port.maxRetries=100",
     javaOptions in Test += "-Dspark.ui.enabled=false",
-    javaOptions in Test += "-Dspark.ui.showProgressBar=false",
+    javaOptions in Test += "-Dspark.ui.showConsoleProgress=false",
     javaOptions in Test += "-Dsun.io.serialization.extendedDebugInfo=true",
     javaOptions in Test ++= System.getProperties.filter(_._1 startsWith "spark")
       .map { case (k,v) => s"-D$k=$v" }.toSeq,