#include "absl/time/clock_interface.h"
#include <cstdint>
#include <functional>
#include <thread>
#include <vector>
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "absl/functional/bind_front.h"
#include "absl/synchronization/mutex.h"
#include "absl/synchronization/notification.h"
#include "absl/time/clock.h"
#include "absl/time/time.h"
namespace {
constexpr absl::Duration kLongPause = absl::Milliseconds(150);
#ifdef _MSC_VER
const char* kSkipFlakyReason =
"Skipping this timing test because it is too flaky";
#else
const char* kSkipFlakyReason = nullptr;
#endif
void AwaitWithDeadlineAndNotify(absl::Clock* clock, absl::Mutex* mu,
absl::Condition* cond, absl::Time wakeup_time,
absl::Notification* note, bool* return_val) {
mu->lock_shared();
*return_val = clock->AwaitWithDeadline(mu, *cond, wakeup_time);
mu->unlock_shared();
note->Notify();
}
TEST(RealClockTest, AwaitWithVeryLargeDeadline) {
if (kSkipFlakyReason != nullptr) {
GTEST_SKIP() << kSkipFlakyReason;
}
absl::Clock& clock = absl::Clock::GetRealClock();
absl::Mutex mu;
bool f = false;
absl::Condition cond(&f);
absl::Notification note;
bool return_val;
std::thread thread(AwaitWithDeadlineAndNotify, &clock, &mu, &cond,
absl::InfiniteFuture(), ¬e, &return_val);
EXPECT_FALSE(note.HasBeenNotified());
mu.lock();
f = true;
mu.unlock();
absl::SleepFor(kLongPause);
EXPECT_TRUE(note.HasBeenNotified());
EXPECT_TRUE(return_val);
note.WaitForNotification(); thread.join();
}
TEST(RealClockTest, AwaitWithPastDeadline) {
if (kSkipFlakyReason != nullptr) {
GTEST_SKIP() << kSkipFlakyReason;
}
absl::Clock& clock = absl::Clock::GetRealClock();
absl::Mutex mu;
bool f = false;
absl::Condition cond(&f);
absl::Notification note;
bool return_val;
std::thread thread(AwaitWithDeadlineAndNotify, &clock, &mu, &cond,
clock.TimeNow() - absl::Seconds(10), ¬e, &return_val);
absl::Time start = absl::Now();
note.WaitForNotification(); EXPECT_GE(kLongPause, absl::Now() - start);
EXPECT_FALSE(return_val);
thread.join();
}
TEST(RealClockTest, AwaitWithSmallDeadline) {
if (kSkipFlakyReason != nullptr) {
GTEST_SKIP() << kSkipFlakyReason;
}
absl::Clock& clock = absl::Clock::GetRealClock();
absl::Mutex mu;
bool f = false;
absl::Condition cond(&f);
for (int i = 0; i < 5; ++i) {
absl::Duration wait_time = absl::Milliseconds(i);
absl::Duration elapsed_time;
{
absl::MutexLock lock(mu);
absl::Time start = clock.TimeNow();
absl::Time wakeup_time = start + wait_time;
EXPECT_FALSE(clock.AwaitWithDeadline(&mu, cond, wakeup_time));
elapsed_time = clock.TimeNow() - start;
}
if (elapsed_time >= absl::ZeroDuration()) { EXPECT_GE(elapsed_time, wait_time - absl::Microseconds(60));
}
}
}
}