From fdd1573c5ec715ac7331d1f3af34ca7f023c451c Mon Sep 17 00:00:00 2001 From: CodingCat Date: Thu, 31 Jul 2014 10:12:19 -0400 Subject: [PATCH] fix failed test cases --- .../scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala | 2 ++ 1 file changed, 2 insertions(+) diff --git a/core/src/test/scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala index 5c3d0520da341..076afa26a42ed 100644 --- a/core/src/test/scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala +++ b/core/src/test/scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala @@ -245,6 +245,7 @@ class TaskSetManagerSuite extends FunSuite with LocalSparkContext with Logging { // Offer host2, exec3 again, at NODE_LOCAL level: we should get noPref task // after failing to find a node_Local task assert(manager.resourceOffer("exec3", "host2", NODE_LOCAL) == None) + clock.advance(LOCALITY_WAIT) assert(manager.resourceOffer("exec3", "host2", NO_PREF).get.index == 3) } @@ -577,6 +578,7 @@ class TaskSetManagerSuite extends FunSuite with LocalSparkContext with Logging { // schedule a node local task assert(manager.resourceOffer("execA", "host1", NODE_LOCAL).get.index === 1) manager.speculatableTasks += 1 + clock.advance(LOCALITY_WAIT) // schedule the nonPref task assert(manager.resourceOffer("execA", "host1", NO_PREF).get.index === 2) // schedule the speculative task