106 lines
2.5 KiB
C++
106 lines
2.5 KiB
C++
#include <gtest/gtest.h>
|
|
|
|
#include <EC/ThreadPool.hpp>
|
|
|
|
using OneThreadPool = EC::ThreadPool<1>;
|
|
using ThreeThreadPool = EC::ThreadPool<3>;
|
|
|
|
TEST(ECThreadPool, OneThread) {
|
|
OneThreadPool p;
|
|
std::atomic_int data;
|
|
data.store(0);
|
|
const auto fn = [](void *ud) {
|
|
auto *data = static_cast<std::atomic_int*>(ud);
|
|
data->fetch_add(1);
|
|
};
|
|
|
|
p.queueFn(fn, &data);
|
|
|
|
p.startThreads();
|
|
|
|
do {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(10));
|
|
} while(!p.isQueueEmpty() || !p.isNotRunning());
|
|
|
|
ASSERT_EQ(data.load(), 1);
|
|
|
|
for(unsigned int i = 0; i < 10; ++i) {
|
|
p.queueFn(fn, &data);
|
|
}
|
|
p.startThreads();
|
|
|
|
do {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(10));
|
|
} while(!p.isQueueEmpty() || !p.isNotRunning());
|
|
|
|
ASSERT_EQ(data.load(), 11);
|
|
}
|
|
|
|
TEST(ECThreadPool, Simple) {
|
|
ThreeThreadPool p;
|
|
std::atomic_int data;
|
|
data.store(0);
|
|
const auto fn = [](void *ud) {
|
|
auto *data = static_cast<std::atomic_int*>(ud);
|
|
data->fetch_add(1);
|
|
};
|
|
|
|
p.queueFn(fn, &data);
|
|
|
|
p.startThreads();
|
|
|
|
do {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(10));
|
|
} while(!p.isQueueEmpty() || !p.isNotRunning());
|
|
|
|
ASSERT_EQ(data.load(), 1);
|
|
|
|
for(unsigned int i = 0; i < 10; ++i) {
|
|
p.queueFn(fn, &data);
|
|
}
|
|
p.startThreads();
|
|
|
|
do {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(10));
|
|
} while(!p.isQueueEmpty() || !p.isNotRunning());
|
|
|
|
ASSERT_EQ(data.load(), 11);
|
|
}
|
|
|
|
TEST(ECThreadPool, QueryCount) {
|
|
{
|
|
OneThreadPool oneP;
|
|
ASSERT_EQ(1, oneP.getMaxThreadCount());
|
|
}
|
|
{
|
|
ThreeThreadPool threeP;
|
|
ASSERT_EQ(3, threeP.getMaxThreadCount());
|
|
}
|
|
}
|
|
|
|
TEST(ECThreadPool, easyStartAndWait) {
|
|
std::atomic_int data;
|
|
data.store(0);
|
|
{
|
|
OneThreadPool oneP;
|
|
for(unsigned int i = 0; i < 20; ++i) {
|
|
oneP.queueFn([] (void *ud) {
|
|
auto *atomicInt = static_cast<std::atomic_int*>(ud);
|
|
atomicInt->fetch_add(1);
|
|
}, &data);
|
|
}
|
|
oneP.easyStartAndWait();
|
|
EXPECT_EQ(20, data.load());
|
|
}
|
|
{
|
|
ThreeThreadPool threeP;
|
|
for(unsigned int i = 0; i < 20; ++i) {
|
|
threeP.queueFn([] (void *ud) {
|
|
auto *atomicInt = static_cast<std::atomic_int*>(ud);
|
|
atomicInt->fetch_add(1);
|
|
}, &data);
|
|
}
|
|
threeP.easyStartAndWait();
|
|
EXPECT_EQ(40, data.load());
|
|
}
|
|
}
|