You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
1.6 KiB
98 lines
1.6 KiB
#include <cxxtest/TestSuite.h>
|
|
|
|
// #include <time.h> // time
|
|
|
|
#include "Task.hpp"
|
|
#include "Thread.hpp"
|
|
#include "ThreadPool.hpp"
|
|
#include "Common.hpp"
|
|
|
|
|
|
|
|
|
|
class TestThreadPoolSuite : public CxxTest::TestSuite
|
|
{
|
|
|
|
class DummyTask : public Task
|
|
{
|
|
|
|
public:
|
|
|
|
DummyTask() { m_timeOut = 5; TRACE(this); }
|
|
|
|
void run()
|
|
{
|
|
TRACE(this);
|
|
m_startedToRun = time(NULL);
|
|
TRACE("I'm a task...");
|
|
m_startedToRun = 0;
|
|
}
|
|
|
|
bool isItStucked () const
|
|
{
|
|
TRACE(this);
|
|
return ( m_startedToRun + m_timeOut < time(NULL) );
|
|
}
|
|
};
|
|
|
|
class WorkerThread : public Thread
|
|
{
|
|
|
|
public:
|
|
|
|
WorkerThread( ThreadPool& tp )
|
|
: m_tp(tp)
|
|
{
|
|
TRACE(this);
|
|
}
|
|
|
|
private:
|
|
|
|
void* run()
|
|
{
|
|
TRACE(this);
|
|
while ( m_isRunning )
|
|
{
|
|
Task* task(0);
|
|
try {
|
|
task = m_tp.popTask();
|
|
task->run();
|
|
delete task;
|
|
} catch (CancelledException) {
|
|
TRACE("Now I die.");
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
ThreadPool& m_tp;
|
|
};
|
|
|
|
|
|
public:
|
|
|
|
void testBasic()
|
|
{
|
|
ThreadPool* tp = new ThreadPool();
|
|
|
|
Thread* wt1 = new WorkerThread(*tp);
|
|
Thread* wt2 = new WorkerThread(*tp);
|
|
Thread* wt3 = new WorkerThread(*tp);
|
|
tp->pushWorkerThread(wt1);
|
|
tp->pushWorkerThread(wt2);
|
|
tp->pushWorkerThread(wt3);
|
|
tp->startWorkerThreads();
|
|
|
|
Task* t1 = new DummyTask();
|
|
tp->pushTask(t1);
|
|
Task* t2 = new DummyTask();
|
|
tp->pushTask(t2);
|
|
|
|
sleep(2);
|
|
|
|
tp->stop();
|
|
tp->join();
|
|
delete tp;
|
|
}
|
|
};
|