dddfe52a45
Summary: see documentation Reviewers: mislav.bradac, buda, teon.banek, dgleich Reviewed By: dgleich Subscribers: teon.banek, dgleich, pullbot Differential Revision: https://phabricator.memgraph.io/D951
60 lines
1.1 KiB
C++
60 lines
1.1 KiB
C++
#include <atomic>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "utils/watchdog.hpp"
|
|
|
|
using namespace std::chrono_literals;
|
|
|
|
TEST(Watchdog, Run) {
|
|
std::atomic<int> count(0);
|
|
Watchdog dog(200ms, 200ms, [&count]() { ++count; });
|
|
|
|
std::this_thread::sleep_for(250ms);
|
|
EXPECT_EQ(count, 1);
|
|
|
|
std::this_thread::sleep_for(200ms);
|
|
EXPECT_EQ(count, 2);
|
|
|
|
std::this_thread::sleep_for(50ms);
|
|
dog.Notify();
|
|
|
|
std::this_thread::sleep_for(150ms);
|
|
EXPECT_EQ(count, 2);
|
|
dog.Notify();
|
|
|
|
std::this_thread::sleep_for(250ms);
|
|
EXPECT_EQ(count, 3);
|
|
}
|
|
|
|
TEST(Watchdog, Blocker) {
|
|
std::atomic<int> count(0);
|
|
Watchdog dog(200ms, 200ms, [&count]() { ++count; });
|
|
|
|
std::this_thread::sleep_for(250ms);
|
|
EXPECT_EQ(count, 1);
|
|
|
|
dog.Block();
|
|
|
|
std::this_thread::sleep_for(200ms);
|
|
EXPECT_EQ(count, 1);
|
|
|
|
std::this_thread::sleep_for(200ms);
|
|
EXPECT_EQ(count, 1);
|
|
|
|
dog.Unblock();
|
|
|
|
std::this_thread::sleep_for(150ms);
|
|
EXPECT_EQ(count, 1);
|
|
|
|
std::this_thread::sleep_for(100ms);
|
|
EXPECT_EQ(count, 2);
|
|
dog.Notify();
|
|
|
|
std::this_thread::sleep_for(100ms);
|
|
dog.Unblock();
|
|
|
|
std::this_thread::sleep_for(150ms);
|
|
EXPECT_EQ(count, 3);
|
|
}
|