address comment

Ngone51 · Ngone51 · commit 021c0a50d0f6 · 2020-05-22T11:43:03.000+08:00
diff --git a/core/src/test/scala/org/apache/spark/scheduler/BarrierTaskContextSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/BarrierTaskContextSuite.scala
@@ -71,10 +71,10 @@ class BarrierTaskContextSuite extends SparkFunSuite with LocalSparkContext with
       val messages: Array[String] = context.allGather(message)
       Iterator.single(messages.toList)
     }
-    // Take a sorted list of all the partitionId messages
-    val messages = rdd2.collect().head
-    // All the task partitionIds are shared
-    assert(messages === List("0", "1", "2", "3"))
+    val messages = rdd2.collect()
+    // All the task partitionIds are shared across all tasks
+    assert(messages.length === 4)
+    assert(messages.forall(_ == List("0", "1", "2", "3")))
   }
 
   test("throw exception if we attempt to synchronize with different blocking calls") {

Original file line number	Diff line number	Diff line change
`@@ -71,10 +71,10 @@ class BarrierTaskContextSuite extends SparkFunSuite with LocalSparkContext with`
`71`	`71`	`val messages: Array[String] = context.allGather(message)`
`72`	`72`	`Iterator.single(messages.toList)`
`73`	`73`	`}`
`74`		`- // Take a sorted list of all the partitionId messages`
`75`		`- val messages = rdd2.collect().head`
`76`		`- // All the task partitionIds are shared`
`77`		`- assert(messages === List("0", "1", "2", "3"))`
	`74`	`+ val messages = rdd2.collect()`
	`75`	`+ // All the task partitionIds are shared across all tasks`
	`76`	`+ assert(messages.length === 4)`
	`77`	`+ assert(messages.forall(_ == List("0", "1", "2", "3")))`
`78`	`78`	`}`
`79`	`79`
`80`	`80`	`test("throw exception if we attempt to synchronize with different blocking calls") {`