Error ACKs should trigger IOExceptions, so catch only those exception…

…s in the test.
xiliu82 · Aug 6, 2014 · 83673de · 83673de
1 parent b8bb4d4
commit 83673de
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/core/src/test/scala/org/apache/spark/network/ConnectionManagerSuite.scala b/core/src/test/scala/org/apache/spark/network/ConnectionManagerSuite.scala
@@ -246,8 +246,9 @@ class ConnectionManagerSuite extends FunSuite {
 
     val future = manager.sendMessageReliably(managerServer.id, bufferMessage)
 
-    val message = Try(Await.result(future, 1 second))
-    assert(message.isFailure)
+    intercept[IOException] {
+      Await.result(future, 1 second)
+    }
 
     manager.stop()
     managerServer.stop()