Rebase with master

apache · Jul 26, 2021 · 1d40d59 · 1d40d59
1 parent f1db4dd
commit 1d40d59
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/test/scala/org/apache/spark/shuffle/ShuffleBlockPusherSuite.scala b/core/src/test/scala/org/apache/spark/shuffle/ShuffleBlockPusherSuite.scala
@@ -332,9 +332,9 @@ class ShuffleBlockPusherSuite extends SparkFunSuite with BeforeAndAfterEach {
     when(shuffleClient.pushBlocks(any(), any(), any(), any(), any()))
       .thenAnswer((invocation: InvocationOnMock) => {
         val pushedBlocks = invocation.getArguments()(2).asInstanceOf[Array[String]]
-        val blockFetchListener = invocation.getArguments()(4).asInstanceOf[BlockFetchingListener]
+        val blockPushListener = invocation.getArguments()(4).asInstanceOf[BlockPushingListener]
         pushedBlocks.foreach(blockId => {
-          blockFetchListener.onBlockFetchFailure(
+          blockPushListener.onBlockPushFailure(
             blockId, new IOException("Failed to send RPC",
               new FileNotFoundException("file not found")))
         })