diff --git a/core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/CoarseMesosSchedulerBackendSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/CoarseMesosSchedulerBackendSuite.scala index 5e01d95fc3bbbf912ec074e737e653087fd48bf7..5db7535d36e18efb7d752ca0d75f5f38d7353388 100644 --- a/core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/CoarseMesosSchedulerBackendSuite.scala +++ b/core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/CoarseMesosSchedulerBackendSuite.scala @@ -300,6 +300,11 @@ class CoarseMesosSchedulerBackendSuite extends SparkFunSuite override protected def createDriverEndpointRef( properties: ArrayBuffer[(String, String)]): RpcEndpointRef = endpoint + // override to avoid race condition with the driver thread on `mesosDriver` + override def startScheduler(newDriver: SchedulerDriver): Unit = { + mesosDriver = newDriver + } + markRegistered() } backend.start()