test/job_test.cc

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include <boost/test/unit_test.hpp>
  21 #include "lib/job.h"
  22 #include "lib/job_manager.h"
  23 #include "lib/cross.h"
  24
  25 using std::string;
  26 using boost::shared_ptr;
  27
  28 class TestJob : public Job
  29 {
  30 public:
  31         TestJob (shared_ptr<Film> f)
  32                 : Job (f)
  33         {
  34
  35         }
  36
  37         void set_finished_ok () {
  38                 set_state (FINISHED_OK);
  39         }
  40
  41         void set_finished_error () {
  42                 set_state (FINISHED_ERROR);
  43         }
  44
  45         void run ()
  46         {
  47                 while (1) {
  48                         if (finished ()) {
  49                                 return;
  50                         }
  51                 }
  52         }
  53
  54         string name () const {
  55                 return "";
  56         }
  57
  58         string json_name () const {
  59                 return "";
  60         }
  61 };
  62
  63 BOOST_AUTO_TEST_CASE (job_manager_test)
  64 {
  65         shared_ptr<Film> f;
  66
  67         /* Single job */
  68         shared_ptr<TestJob> a (new TestJob (f));
  69
  70         JobManager::instance()->add (a);
  71         dcpomatic_sleep (1);
  72         BOOST_CHECK_EQUAL (a->running (), true);
  73         a->set_finished_ok ();
  74         dcpomatic_sleep (2);
  75         BOOST_CHECK_EQUAL (a->finished_ok(), true);
  76 }