Oops, removed one flag too many. Adding it back.

andrewor14 · andrewor14 · commit c4714902a7d9 · 2014-05-07T09:58:51.000-07:00
diff --git a/core/src/test/scala/org/apache/spark/CacheManagerSuite.scala b/core/src/test/scala/org/apache/spark/CacheManagerSuite.scala
@@ -23,7 +23,6 @@ import org.scalatest.{BeforeAndAfter, FunSuite}
 import org.scalatest.mock.EasyMockSugar
 
 import org.apache.spark.rdd.RDD
-import org.apache.spark.executor.TaskMetrics
 import org.apache.spark.storage._
 
 // TODO: Test the CacheManager's thread-safety aspects
@@ -84,7 +83,7 @@ class CacheManagerSuite extends FunSuite with BeforeAndAfter with EasyMockSugar
     }
 
     whenExecuting(blockManager) {
-      val context = new TaskContext(0, 0, 0)
+      val context = new TaskContext(0, 0, 0, runningLocally = true)
       val value = cacheManager.getOrCompute(rdd, split, context, StorageLevel.MEMORY_ONLY)
       assert(value.toList === List(1, 2, 3, 4))
     }

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,6 @@ import org.scalatest.{BeforeAndAfter, FunSuite}`
`23`	`23`	`import org.scalatest.mock.EasyMockSugar`
`24`	`24`
`25`	`25`	`import org.apache.spark.rdd.RDD`
`26`		`-import org.apache.spark.executor.TaskMetrics`
`27`	`26`	`import org.apache.spark.storage._`
`28`	`27`
`29`	`28`	`// TODO: Test the CacheManager's thread-safety aspects`
`@@ -84,7 +83,7 @@ class CacheManagerSuite extends FunSuite with BeforeAndAfter with EasyMockSugar`
`84`	`83`	`}`
`85`	`84`
`86`	`85`	`whenExecuting(blockManager) {`
`87`		`- val context = new TaskContext(0, 0, 0)`
	`86`	`+ val context = new TaskContext(0, 0, 0, runningLocally = true)`
`88`	`87`	`val value = cacheManager.getOrCompute(rdd, split, context, StorageLevel.MEMORY_ONLY)`
`89`	`88`	`assert(value.toList === List(1, 2, 3, 4))`
`90`	`89`	`}`