Skip to content

Commit 8e2a969

Browse files
committed
formatting of runEvent(CompletionEvent(...))
1 parent 3e3ea39 commit 8e2a969

File tree

1 file changed

+28
-8
lines changed

1 file changed

+28
-8
lines changed

core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala

Lines changed: 28 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -659,21 +659,41 @@ class DAGSchedulerSuite
659659
val shuffleStage = scheduler.stageIdToStage(taskSet.stageId).asInstanceOf[ShuffleMapStage]
660660
assert(shuffleStage.numAvailableOutputs === 0)
661661
// should be ignored for being too old
662-
runEvent(CompletionEvent(taskSet.tasks(0), Success, makeMapStatus("hostA",
663-
reduceRdd.partitions.size), null, createFakeTaskInfo(), null))
662+
runEvent(CompletionEvent(
663+
taskSet.tasks(0),
664+
Success,
665+
makeMapStatus("hostA", reduceRdd.partitions.size),
666+
null,
667+
createFakeTaskInfo(),
668+
null))
664669
assert(shuffleStage.numAvailableOutputs === 0)
665670
// should work because it's a non-failed host (so the available map outputs will increase)
666-
runEvent(CompletionEvent(taskSet.tasks(0), Success, makeMapStatus("hostB",
667-
reduceRdd.partitions.size), null, createFakeTaskInfo(), null))
671+
runEvent(CompletionEvent(
672+
taskSet.tasks(0),
673+
Success,
674+
makeMapStatus("hostB", reduceRdd.partitions.size),
675+
null,
676+
createFakeTaskInfo(),
677+
null))
668678
assert(shuffleStage.numAvailableOutputs === 1)
669679
// should be ignored for being too old
670-
runEvent(CompletionEvent(taskSet.tasks(0), Success, makeMapStatus("hostA",
671-
reduceRdd.partitions.size), null, createFakeTaskInfo(), null))
680+
runEvent(CompletionEvent(
681+
taskSet.tasks(0),
682+
Success,
683+
makeMapStatus("hostA", reduceRdd.partitions.size),
684+
null,
685+
createFakeTaskInfo(),
686+
null))
672687
assert(shuffleStage.numAvailableOutputs === 1)
673688
// should work because it's a new epoch
674689
taskSet.tasks(1).epoch = newEpoch
675-
runEvent(CompletionEvent(taskSet.tasks(1), Success, makeMapStatus("hostA",
676-
reduceRdd.partitions.size), null, createFakeTaskInfo(), null))
690+
runEvent(CompletionEvent(
691+
taskSet.tasks(1),
692+
Success,
693+
makeMapStatus("hostA", reduceRdd.partitions.size),
694+
null,
695+
createFakeTaskInfo(),
696+
null))
677697
assert(shuffleStage.numAvailableOutputs === 2)
678698
assert(mapOutputTracker.getMapSizesByExecutorId(shuffleId, 0).map(_._1).toSet ===
679699
HashSet(makeBlockManagerId("hostB"), makeBlockManagerId("hostA")))

0 commit comments

Comments
 (0)