ddnet/src/test/jobs.cpp

108 lines
2 KiB
C++
Raw Normal View History

2017-11-24 09:33:42 +00:00
#include "test.h"
#include <gtest/gtest.h>
#include <base/system.h>
#include <engine/shared/jobs.h>
#include <engine/engine.h>
#include <functional>
static const int TEST_NUM_THREADS = 4;
class Jobs : public ::testing::Test
{
protected:
CJobPool m_Pool;
Jobs()
{
m_Pool.Init(TEST_NUM_THREADS);
}
void Add(std::shared_ptr<IJob> pJob)
{
m_Pool.Add(std::move(pJob));
}
};
class CJob : public IJob
{
std::function<void()> m_JobFunction;
void Run() { m_JobFunction(); }
public:
CJob(std::function<void()>&& JobFunction) : m_JobFunction(JobFunction) {}
};
TEST_F(Jobs, Constructor)
{
}
TEST_F(Jobs, Simple)
{
Add(std::make_shared<CJob>([] {}));
}
TEST_F(Jobs, Wait)
{
SEMAPHORE sphore;
sphore_init(&sphore);
Add(std::make_shared<CJob>([&] { sphore_signal(&sphore); }));
sphore_wait(&sphore);
sphore_destroy(&sphore);
}
TEST_F(Jobs, LookupHost)
{
static const char *HOST = "example.com";
static const int NETTYPE = NETTYPE_ALL;
auto pJob = std::make_shared<CHostLookup>(HOST, NETTYPE);
EXPECT_STREQ(pJob->m_aHostname, HOST);
EXPECT_EQ(pJob->m_Nettype, NETTYPE);
Add(pJob);
while(pJob->Status() != IJob::STATE_DONE)
{
// yay, busy loop...
thread_yield();
}
EXPECT_STREQ(pJob->m_aHostname, HOST);
EXPECT_EQ(pJob->m_Nettype, NETTYPE);
ASSERT_EQ(pJob->m_Result, 0);
EXPECT_EQ(pJob->m_Addr.type & NETTYPE, pJob->m_Addr.type);
}
TEST_F(Jobs, Many)
{
std::atomic<int> ThreadsRunning(0);
std::vector<std::shared_ptr<IJob>> apJobs;
SEMAPHORE sphore;
sphore_init(&sphore);
for(int i = 0; i < TEST_NUM_THREADS; i++)
{
std::shared_ptr<IJob> pJob = std::make_shared<CJob>([&]
{
int Prev = ThreadsRunning.fetch_add(1);
if(Prev == TEST_NUM_THREADS - 1)
{
sphore_signal(&sphore);
}
});
EXPECT_EQ(pJob->Status(), IJob::STATE_PENDING);
apJobs.push_back(pJob);
}
for(auto &pJob: apJobs)
{
Add(pJob);
}
sphore_wait(&sphore);
sphore_destroy(&sphore);
m_Pool.~CJobPool();
for(auto &pJob: apJobs)
{
EXPECT_EQ(pJob->Status(), IJob::STATE_DONE);
}
new(&m_Pool) CJobPool();
}