fix test

zsxwing · zsxwing · commit 3fade9526fda · 2016-12-09T17:10:23.000-08:00
diff --git a/core/src/test/scala/org/apache/spark/rdd/AsyncRDDActionsSuite.scala b/core/src/test/scala/org/apache/spark/rdd/AsyncRDDActionsSuite.scala
@@ -199,10 +199,9 @@ class AsyncRDDActionsSuite extends SparkFunSuite with BeforeAndAfterAll with Tim
     val f = sc.parallelize(1 to 100, 4)
               .mapPartitions(itr => { Thread.sleep(20); itr })
               .countAsync()
-    val e = intercept[SparkException] {
+    intercept[TimeoutException] {
       ThreadUtils.awaitResult(f, Duration(20, "milliseconds"))
     }
-    assert(e.getCause.isInstanceOf[TimeoutException])
   }
 
   private def testAsyncAction[R](action: RDD[Int] => FutureAction[R]): Unit = {

Original file line number	Diff line number	Diff line change
`@@ -199,10 +199,9 @@ class AsyncRDDActionsSuite extends SparkFunSuite with BeforeAndAfterAll with Tim`
`199`	`199`	`val f = sc.parallelize(1 to 100, 4)`
`200`	`200`	`.mapPartitions(itr => { Thread.sleep(20); itr })`
`201`	`201`	`.countAsync()`
`202`		`- val e = intercept[SparkException] {`
	`202`	`+ intercept[TimeoutException] {`
`203`	`203`	`ThreadUtils.awaitResult(f, Duration(20, "milliseconds"))`
`204`	`204`	`}`
`205`		`- assert(e.getCause.isInstanceOf[TimeoutException])`
`206`	`205`	`}`
`207`	`206`
`208`	`207`	`private def testAsyncAction[R](action: RDD[Int] => FutureAction[R]): Unit = {`