From c5f9c1ecb6cd7e39c43d40be23189c03b24a7f4f Mon Sep 17 00:00:00 2001 From: Dongjoon Hyun Date: Fri, 16 Oct 2020 17:37:53 -0700 Subject: [PATCH] [SPARK-33173][CORE][TESTS] Use `eventually` to check `numOnTaskFailed` --- .../spark/internal/plugin/PluginContainerSuite.scala | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/core/src/test/scala/org/apache/spark/internal/plugin/PluginContainerSuite.scala b/core/src/test/scala/org/apache/spark/internal/plugin/PluginContainerSuite.scala index e7fbe5b998a88..15966e2744491 100644 --- a/core/src/test/scala/org/apache/spark/internal/plugin/PluginContainerSuite.scala +++ b/core/src/test/scala/org/apache/spark/internal/plugin/PluginContainerSuite.scala @@ -156,9 +156,11 @@ class PluginContainerSuite extends SparkFunSuite with BeforeAndAfterEach with Lo case t: Throwable => // ignore exception } - assert(TestSparkPlugin.executorPlugin.numOnTaskStart == 2) - assert(TestSparkPlugin.executorPlugin.numOnTaskSucceeded == 0) - assert(TestSparkPlugin.executorPlugin.numOnTaskFailed == 2) + eventually(timeout(10.seconds), interval(100.millis)) { + assert(TestSparkPlugin.executorPlugin.numOnTaskStart == 2) + assert(TestSparkPlugin.executorPlugin.numOnTaskSucceeded == 0) + assert(TestSparkPlugin.executorPlugin.numOnTaskFailed == 2) + } } test("plugin initialization in non-local mode") {