9278c909d3
Because the fake clock is getting out of sync with the real clock for the nonce flood tests, add a call to TestSleep::SyncFakeClock() in Clock::GetCurrentTime() to force a sync. Merged from http://go/wvgerrit/133223 Test: duration_use_case_test and nonce flood tests on bonito Bug: 198329759 Change-Id: Idfd92c6cc57750a52477189461220c682b302082
44 lines
1.1 KiB
C++
44 lines
1.1 KiB
C++
// Copyright 2019 Google LLC. All Rights Reserved. This file and proprietary
|
|
// source code may only be used and distributed under the Widevine License
|
|
// Agreement.
|
|
//
|
|
// Clock - A fake clock just for running tests.
|
|
|
|
#include "clock.h"
|
|
|
|
#include <chrono>
|
|
|
|
#include "test_sleep.h"
|
|
|
|
namespace wvcdm {
|
|
|
|
namespace {
|
|
// A fake clock that only advances when TestSleep::Sleep is called.
|
|
class FakeClock : public wvcdm::TestSleep::CallBack {
|
|
public:
|
|
FakeClock() {
|
|
auto now = std::chrono::system_clock().now();
|
|
now_ = now.time_since_epoch() / std::chrono::milliseconds(1);
|
|
TestSleep::set_callback(this);
|
|
}
|
|
~FakeClock() { TestSleep::set_callback(nullptr); }
|
|
void ElapseTime(int64_t milliseconds) { now_ += milliseconds; }
|
|
|
|
int64_t now() const { return now_; }
|
|
|
|
private:
|
|
int64_t now_;
|
|
};
|
|
|
|
FakeClock* g_fake_clock = nullptr;
|
|
} // namespace
|
|
|
|
// On devices running a fake OEMCrypto, we can use a fake sleep and fake time.
|
|
int64_t Clock::GetCurrentTime() {
|
|
wvcdm::TestSleep::SyncFakeClock();
|
|
if (g_fake_clock == nullptr) g_fake_clock = new FakeClock();
|
|
return g_fake_clock->now() / 1000;
|
|
}
|
|
|
|
} // namespace wvcdm
|