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