0eb30db8db
Reviewers: teon.banek Reviewed By: teon.banek Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1062
46 lines
1.2 KiB
C++
46 lines
1.2 KiB
C++
#include <atomic>
|
|
#include <memory>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "utils/executor.hpp"
|
|
|
|
TEST(Executor, Run) {
|
|
std::atomic<int> count{0};
|
|
{
|
|
Executor exec(std::chrono::milliseconds(500));
|
|
// Be sure executor is sleeping.
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
|
|
|
exec.RegisterJob([&count]() { ++count; });
|
|
exec.RegisterJob([&count]() { ++count; });
|
|
exec.RegisterJob([&count]() { ++count; });
|
|
|
|
// Be sure executor execute thread is triggered
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(500));
|
|
}
|
|
EXPECT_EQ(count, 3);
|
|
}
|
|
|
|
TEST(Executor, RunUnregister) {
|
|
std::atomic<int> count1{0};
|
|
std::atomic<int> count2{0};
|
|
{
|
|
Executor exec(std::chrono::milliseconds(500));
|
|
// Be sure executor is sleeping.
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
|
auto job = exec.RegisterJob([&count1]() { ++count1; });
|
|
exec.RegisterJob([&count2]() { ++count2; });
|
|
|
|
// Be sure executor execute thread is triggered
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(500));
|
|
|
|
exec.UnRegisterJob(job);
|
|
|
|
// Be sure executor execute thread is triggered
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(500));
|
|
}
|
|
EXPECT_EQ(count1, 1);
|
|
EXPECT_EQ(count2, 2);
|
|
}
|