@@ -1289,11 +1289,11 @@ class AppStatusListenerSuite extends SparkFunSuite with BeforeAndAfter {
1289
1289
1290
1290
time += 1
1291
1291
tasks(0 ).markFinished(TaskState .FINISHED , time)
1292
- listener.onTaskEnd(SparkListenerTaskEnd (stage.stageId, stage.attemptId , " taskType" ,
1292
+ listener.onTaskEnd(SparkListenerTaskEnd (stage.stageId, stage.attemptNumber , " taskType" ,
1293
1293
Success , tasks(0 ), null ))
1294
1294
time += 1
1295
1295
tasks(1 ).markFinished(TaskState .FINISHED , time)
1296
- listener.onTaskEnd(SparkListenerTaskEnd (stage.stageId, stage.attemptId , " taskType" ,
1296
+ listener.onTaskEnd(SparkListenerTaskEnd (stage.stageId, stage.attemptNumber , " taskType" ,
1297
1297
Success , tasks(1 ), null ))
1298
1298
1299
1299
stage.failureReason = Some (" Failed" )
@@ -1303,11 +1303,11 @@ class AppStatusListenerSuite extends SparkFunSuite with BeforeAndAfter {
1303
1303
1304
1304
time += 1
1305
1305
tasks(2 ).markFinished(TaskState .FAILED , time)
1306
- listener.onTaskEnd(SparkListenerTaskEnd (stage.stageId, stage.attemptId , " taskType" ,
1306
+ listener.onTaskEnd(SparkListenerTaskEnd (stage.stageId, stage.attemptNumber , " taskType" ,
1307
1307
ExecutorLostFailure (" 1" , true , Some (" Lost executor" )), tasks(2 ), null ))
1308
1308
time += 1
1309
1309
tasks(3 ).markFinished(TaskState .FAILED , time)
1310
- listener.onTaskEnd(SparkListenerTaskEnd (stage.stageId, stage.attemptId , " taskType" ,
1310
+ listener.onTaskEnd(SparkListenerTaskEnd (stage.stageId, stage.attemptNumber , " taskType" ,
1311
1311
ExecutorLostFailure (" 2" , true , Some (" Lost executor" )), tasks(3 ), null ))
1312
1312
1313
1313
val esummary = store.view(classOf [ExecutorStageSummaryWrapper ]).asScala.map(_.info)
0 commit comments