Fixed test hanging.

apache · Oct 10, 2014 · f0a16e9 · f0a16e9
1 parent 14323a5
commit f0a16e9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/core/src/test/scala/org/apache/spark/storage/ShuffleBlockFetcherIteratorSuite.scala b/core/src/test/scala/org/apache/spark/storage/ShuffleBlockFetcherIteratorSuite.scala
@@ -207,6 +207,8 @@ class ShuffleBlockFetcherIteratorSuite extends FunSuite {
             ShuffleBlockId(0, 0, 0).toString, blocks(ShuffleBlockId(0, 0, 0)))
           listener.onBlockFetchFailure(
             ShuffleBlockId(0, 1, 0).toString, new BlockNotFoundException("blah"))
+          listener.onBlockFetchFailure(
+            ShuffleBlockId(0, 2, 0).toString, new BlockNotFoundException("blah"))
           sem.release()
         }
       }