[SPARK-10454] [SPARK CORE] wait for empty event queue

robbins · Andrew Or · commit 09e08dbc15d4 · 2015-09-04T15:23:36.000-07:00
Author: robbins <robbins@uk.ibm.com> Closes #8605 from robbinspg/DAGSchedulerSuite-fix. (cherry picked from commit 2e1c175) Signed-off-by: Andrew Or <andrew@databricks.com>
diff --git a/core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala
@@ -527,6 +527,7 @@ class DAGSchedulerSuite
     }
 
     // The map stage should have been submitted.
+    sc.listenerBus.waitUntilEmpty(WAIT_TIMEOUT_MILLIS)
     assert(countSubmittedMapStageAttempts() === 1)
 
     complete(taskSets(0), Seq(

Original file line number	Diff line number	Diff line change
`@@ -527,6 +527,7 @@ class DAGSchedulerSuite`
`527`	`527`	`}`
`528`	`528`
`529`	`529`	`// The map stage should have been submitted.`
	`530`	`+ sc.listenerBus.waitUntilEmpty(WAIT_TIMEOUT_MILLIS)`
`530`	`531`	`assert(countSubmittedMapStageAttempts() === 1)`
`531`	`532`
`532`	`533`	`complete(taskSets(0), Seq(`