mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-14 22:56:12 -04:00
80 lines
2.6 KiB
C++
80 lines
2.6 KiB
C++
// Copyright 2020 Google LLC
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "internal/platform/cancelable_alarm.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "absl/time/time.h"
|
|
#include "internal/platform/atomic_boolean.h"
|
|
#include "internal/platform/atomic_reference.h"
|
|
#include "internal/platform/implementation/system_clock.h"
|
|
#include "internal/platform/scheduled_executor.h"
|
|
|
|
namespace nearby {
|
|
namespace {
|
|
|
|
TEST(CancelableAlarmTest, CanCreateDefault) { CancelableAlarm alarm; }
|
|
|
|
TEST(CancelableAlarmTest, CancelDefaultFails) {
|
|
CancelableAlarm alarm;
|
|
EXPECT_FALSE(alarm.Cancel());
|
|
}
|
|
|
|
TEST(CancelableAlarmTest, CanCreateAndFireAlarm) {
|
|
ScheduledExecutor alarm_executor;
|
|
AtomicBoolean done{false};
|
|
CancelableAlarm alarm(
|
|
"test_alarm", [&done]() { done.Set(true); }, absl::Milliseconds(100),
|
|
&alarm_executor);
|
|
SystemClock::Sleep(absl::Milliseconds(1000));
|
|
EXPECT_TRUE(done.Get());
|
|
}
|
|
|
|
TEST(CancelableAlarmTest, CanCreateAndCancelAlarm) {
|
|
ScheduledExecutor alarm_executor;
|
|
AtomicBoolean done{false};
|
|
CancelableAlarm alarm(
|
|
"test_alarm", [&done]() { done.Set(true); }, absl::Milliseconds(100),
|
|
&alarm_executor);
|
|
EXPECT_TRUE(alarm.Cancel());
|
|
SystemClock::Sleep(absl::Milliseconds(1000));
|
|
EXPECT_FALSE(done.Get());
|
|
}
|
|
|
|
TEST(CancelableAlarmTest, CancelExpiredAlarmFails) {
|
|
ScheduledExecutor alarm_executor;
|
|
AtomicBoolean done{false};
|
|
CancelableAlarm alarm(
|
|
"test_alarm", [&done]() { done.Set(true); }, absl::Milliseconds(100),
|
|
&alarm_executor);
|
|
SystemClock::Sleep(absl::Milliseconds(1000));
|
|
EXPECT_TRUE(done.Get());
|
|
EXPECT_FALSE(alarm.Cancel());
|
|
}
|
|
|
|
TEST(CancelableAlarmTest, CanCreateRecurringAlarm) {
|
|
ScheduledExecutor alarm_executor;
|
|
AtomicReference<int> count(0);
|
|
CancelableAlarm alarm(
|
|
"test_alarm", [&count]() { count.Set(count.Get() + 1); },
|
|
absl::Milliseconds(1000), &alarm_executor, /*is_recurring=*/true);
|
|
// Wait for 2 rounds (>1000ms * 2) and expect the `count` = 2.
|
|
SystemClock::Sleep(absl::Milliseconds(2800));
|
|
alarm.Cancel();
|
|
EXPECT_EQ(count.Get(), 2);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace nearby
|