Roll back the case which wait timeout is reduced (My bad)

apache · Sep 20, 2019 · 8e661f7 · 8e661f7
1 parent 769c1ee
commit 8e661f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/test/scala/org/apache/spark/scheduler/CoarseGrainedSchedulerBackendSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/CoarseGrainedSchedulerBackendSuite.scala
@@ -180,7 +180,7 @@ class CoarseGrainedSchedulerBackendSuite extends SparkFunSuite with LocalSparkCo
     backend.driverEndpoint.askSync[Boolean](
       RegisterExecutor("3", mockEndpointRef, mockAddress.host, 1, logUrls, attributes, Map.empty))
 
-    sc.listenerBus.waitUntilEmpty()
+    sc.listenerBus.waitUntilEmpty(executorUpTimeout.toMillis)
     assert(executorAddedCount === 3)
   }
 
@@ -251,7 +251,7 @@ class CoarseGrainedSchedulerBackendSuite extends SparkFunSuite with LocalSparkCo
       assert(execResources(GPU).availableAddrs.sorted === Array("0", "1", "3"))
       assert(execResources(GPU).assignedAddrs.isEmpty)
     }
-    sc.listenerBus.waitUntilEmpty()
+    sc.listenerBus.waitUntilEmpty(executorUpTimeout.toMillis)
     assert(executorAddedCount === 3)
   }