source: trunk/test/scheduler3.cc @ 3423

// $Id: scheduler3.cc 3423 2015-09-15 23:40:55Z peter $

/*
  Copyright (C) 2015 Peter Johansson

  This file is part of the yat library, http://dev.thep.lu.se/yat

  The yat library is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License as
  published by the Free Software Foundation; either version 3 of the
  License, or (at your option) any later version.

  The yat library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with yat. If not, see <http://www.gnu.org/licenses/>.
*/

// test Scheduler and a case in which a Job throws an exception which
// is caught after Scheduler has been destructed.


#include <config.h>

#include "Suite.h"

#include "yat/utility/Exception.h"
#include "yat/utility/Scheduler.h"

#include <boost/shared_ptr.hpp>
#include <boost/exception/all.hpp>

#include <iostream>

using namespace theplu::yat;
using namespace theplu::yat::utility;

class Sleeper : public Scheduler::Job
{
public:
  Sleeper(int x) : time_(x) {}
  void operator()(void)
  {
    if (time_==3)
      throw utility::runtime_error("some message");
    sleep(time_);
    std::cerr << "sleeping " << time_ << " seconds\n";
  }
private:
  int time_;
};


void run(void)
{
  Scheduler scheduler(2);
  std::vector<boost::shared_ptr<Sleeper> > task;
  task.push_back(boost::shared_ptr<Sleeper>(new Sleeper(1)));
  task.push_back(boost::shared_ptr<Sleeper>(new Sleeper(2)));
  task.push_back(boost::shared_ptr<Sleeper>(new Sleeper(3)));
  task.push_back(boost::shared_ptr<Sleeper>(new Sleeper(1)));
  task.push_back(boost::shared_ptr<Sleeper>(new Sleeper(1)));
  for (size_t i=0; i<task.size(); ++i)
    scheduler.submit(task[i]);
  scheduler.wait();
}


int main(int argc, char* argv[])
{
  test::Suite suite(argc, argv);

  try {
    run();
    suite.add(false);
  }
  catch (std::runtime_error& e) {
    suite.out() << "expected exception caught with what: " << e.what() << "\n";
  }
  return suite.return_value();
}