From 6ce8b63bb6329a3cad00ffca8fc5bc659eccd383 Mon Sep 17 00:00:00 2001
From: fangfengbin <869218239a@zju.edu.cn>
Date: Sun, 29 Mar 2020 11:18:47 +0800
Subject: [PATCH] fix TestTaskLeaseRenewal test failure (#7765)

---
 src/ray/raylet/task_dependency_manager_test.cc | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/ray/raylet/task_dependency_manager_test.cc b/src/ray/raylet/task_dependency_manager_test.cc
index aef9915c8..1b3c67bbe 100644
--- a/src/ray/raylet/task_dependency_manager_test.cc
+++ b/src/ray/raylet/task_dependency_manager_test.cc
@@ -449,7 +449,12 @@ TEST_F(TaskDependencyManagerTest, TestTaskLeaseRenewal) {
   for (int i = 1; i <= num_expected_calls; i++) {
     sleep_time += i * initial_lease_period_ms_;
   }
-  EXPECT_CALL(*task_accessor_mock_, AsyncAddTaskLease(_, _)).Times(num_expected_calls);
+  // When sleep_time = 10 * initial_lease_period_ms_, test case fails, because the
+  // AsyncAddTaskLease function is expected to be called four times, but only three times.
+  // It's hard to determine the sleep_time value, so let's double it for now.
+  sleep_time = sleep_time * 2;
+  EXPECT_CALL(*task_accessor_mock_, AsyncAddTaskLease(_, _))
+      .Times(testing::AtLeast(num_expected_calls));
   Run(sleep_time);
 }