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

robbins · Marcelo Vanzin · commit f64e01b4a97e · 2015-09-29T13:19:45.000-07:00
Author: robbins <robbins@uk.ibm.com> Closes apache#8605 from robbinspg/DAGSchedulerSuite-fix. (cherry picked from commit 2e1c175) Signed-off-by: Andrew Or <andrew@databricks.com> (cherry picked from commit 09e08db)
diff --git a/core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala
@@ -803,6 +803,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
`@@ -803,6 +803,7 @@ class DAGSchedulerSuite`
`803`	`803`	`}`
`804`	`804`
`805`	`805`	`// The map stage should have been submitted.`
	`806`	`+ sc.listenerBus.waitUntilEmpty(WAIT_TIMEOUT_MILLIS)`
`806`	`807`	`assert(countSubmittedMapStageAttempts() === 1)`
`807`	`808`
`808`	`809`	`complete(taskSets(0), Seq(`