Skip to content

Commit 75cdd1e

Browse files
committed
fix tests
1 parent 6239a99 commit 75cdd1e

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

core/src/test/scala/org/apache/spark/scheduler/dynalloc/ExecutorMonitorSuite.scala

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -435,6 +435,8 @@ class ExecutorMonitorSuite extends SparkFunSuite {
435435

436436
test("SPARK-33799: ExecutorMonitor should handle " +
437437
"SparkListenerExecutorExcluded/SparkListenerExecutorUnexcluded") {
438+
conf.set(EXCLUDE_ON_FAILURE_ENABLED, true)
439+
.set(EXCLUDE_ON_FAILURE_KILL_ENABLED, false)
438440
monitor = new ExecutorMonitor(conf, client, null, clock)
439441
monitor.onExecutorAdded(SparkListenerExecutorAdded(clock.getTimeMillis(), "1", execInfo))
440442
assert(monitor.executorCountWithResourceProfile(execInfo.resourceProfileId) === 1)
@@ -450,6 +452,8 @@ class ExecutorMonitorSuite extends SparkFunSuite {
450452

451453
test("SPARK-33799: ExecutorMonitor should handle " +
452454
"SparkListenerNodeExcluded/SparkListenerNodeUnexcluded") {
455+
conf.set(EXCLUDE_ON_FAILURE_ENABLED, true)
456+
.set(EXCLUDE_ON_FAILURE_KILL_ENABLED, false)
453457
monitor = new ExecutorMonitor(conf, client, null, clock)
454458
monitor.onExecutorAdded(SparkListenerExecutorAdded(clock.getTimeMillis(), "1", execInfo))
455459
monitor.onExecutorAdded(SparkListenerExecutorAdded(clock.getTimeMillis(), "2", execInfo))
@@ -468,6 +472,8 @@ class ExecutorMonitorSuite extends SparkFunSuite {
468472
}
469473

470474
test("SPARK-33799: Excluded executor can still be timedout") {
475+
conf.set(EXCLUDE_ON_FAILURE_ENABLED, true)
476+
.set(EXCLUDE_ON_FAILURE_KILL_ENABLED, false)
471477
monitor = new ExecutorMonitor(conf, client, null, clock)
472478
monitor.onExecutorAdded(SparkListenerExecutorAdded(clock.getTimeMillis(), "1", execInfo))
473479
assert(monitor.executorCount === 1)

0 commit comments

Comments
 (0)