cpp_utils/test/test_ThreadPool.hpp

#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;
  }
};