aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Jonathan Hseu <jhseu@google.com>2018-04-11 18:57:49 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-11 18:59:55 -0700
commitb52b5a47148b6f05ed9439840dff9e3f189b3b19 (patch)
tree59f57c9d5bc0b213a4b47fb36e2b1fdbef82d0d5
parent4e29ebd67cd4409cbdfa6510b06acd780166aa9d (diff)
Switch to WaitForNotification to fix the flaky test.
-rw-r--r--tensorflow/core/platform/cloud/ram_file_block_cache_test.cc3
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/core/platform/cloud/ram_file_block_cache_test.cc b/tensorflow/core/platform/cloud/ram_file_block_cache_test.cc
index d555b682a6..10203783fc 100644
--- a/tensorflow/core/platform/cloud/ram_file_block_cache_test.cc
+++ b/tensorflow/core/platform/cloud/ram_file_block_cache_test.cc
@@ -487,8 +487,7 @@ TEST(RamFileBlockCacheTest, CoalesceConcurrentReads) {
TF_EXPECT_OK(ReadCache(&cache, "", 0, block_size / 2, &out));
EXPECT_EQ(out.size(), block_size / 2);
}));
- EXPECT_TRUE(WaitForNotificationWithTimeout(&notification, 10000))
- << "Timeout waiting for concurrent thread to start.";
+ notification.WaitForNotification();
std::vector<char> out;
TF_EXPECT_OK(ReadCache(&cache, "", block_size / 2, block_size / 2, &out));
EXPECT_EQ(out.size(), block_size / 2);