parent
0866c80c1a
commit
4a13d359fa
@ -1,103 +0,0 @@
|
||||
#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);
|
||||
std::cout << "I'm a task..." << std::endl;
|
||||
// other stuff
|
||||
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) {
|
||||
std::cout << "Now I die." << std::endl;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
ThreadPool& m_tp;
|
||||
bool m_isRunning;
|
||||
|
||||
};
|
||||
|
||||
|
||||
public:
|
||||
|
||||
void testBasic()
|
||||
{
|
||||
TRACE("testBasic begin");
|
||||
|
||||
ThreadPool* tp = new ThreadPool(5);
|
||||
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);
|
||||
|
||||
sleep(2);
|
||||
|
||||
tp->stop();
|
||||
tp->join();
|
||||
delete tp;
|
||||
|
||||
TRACE("testBasic end");
|
||||
}
|
||||
};
|
Loading…
Reference in new issue