2 Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
4 This file is part of DCP-o-matic.
6 DCP-o-matic is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
11 DCP-o-matic is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with DCP-o-matic. If not, see <http://www.gnu.org/licenses/>.
22 /** @file test/job_manager_test.cc
23 * @brief Test JobManager.
24 * @ingroup selfcontained
28 #include "lib/cross.h"
30 #include "lib/job_manager.h"
31 #include <boost/test/unit_test.hpp>
34 using std::make_shared;
35 using std::shared_ptr;
40 class TestJob : public Job
43 explicit TestJob (shared_ptr<Film> film)
54 void set_finished_ok () {
55 set_state (FINISHED_OK);
58 void set_finished_error () {
59 set_state (FINISHED_ERROR);
68 boost::this_thread::interruption_point();
72 string name () const override {
76 string json_name () const override {
82 BOOST_AUTO_TEST_CASE (job_manager_test1)
84 shared_ptr<Film> film;
87 auto a = make_shared<TestJob>(film);
89 JobManager::instance()->add (a);
90 dcpomatic_sleep_seconds (1);
91 BOOST_CHECK (a->running());
92 a->set_finished_ok ();
93 dcpomatic_sleep_seconds (2);
94 BOOST_CHECK (a->finished_ok());
98 BOOST_AUTO_TEST_CASE (job_manager_test2)
100 shared_ptr<Film> film;
102 vector<shared_ptr<TestJob>> jobs;
103 for (int i = 0; i < 16; ++i) {
104 auto job = make_shared<TestJob>(film);
105 jobs.push_back (job);
106 JobManager::instance()->add (job);
109 dcpomatic_sleep_seconds (1);
110 BOOST_CHECK (jobs[0]->running());
111 jobs[0]->set_finished_ok();
113 dcpomatic_sleep_seconds (1);
114 BOOST_CHECK (!jobs[0]->running());
115 BOOST_CHECK (jobs[1]->running());
117 /* Push our jobs[5] to the top of the list */
118 for (int i = 0; i < 5; ++i) {
119 JobManager::instance()->increase_priority(jobs[5]);
122 dcpomatic_sleep_seconds (1);
123 for (int i = 0; i < 16; ++i) {
124 BOOST_CHECK (i == 5 ? jobs[i]->running() : !jobs[i]->running());
127 /* Set any jobs that are started to be finished, until they're all finished */
129 if (std::find_if(jobs.begin(), jobs.end(), [](shared_ptr<Job> job) { return !job->finished_ok(); }) == jobs.end()) {
133 for (auto job: jobs) {
134 if (job->running()) {
135 job->set_finished_ok();
140 BOOST_REQUIRE (!wait_for_jobs());
144 BOOST_AUTO_TEST_CASE(cancel_job_test)
146 shared_ptr<Film> film;
148 vector<shared_ptr<TestJob>> jobs;
149 for (int i = 0; i < 2; ++i) {
150 auto job = make_shared<TestJob>(film);
152 JobManager::instance()->add(job);
158 dcpomatic_sleep_seconds(1);
160 BOOST_CHECK(jobs[0]->finished_cancelled());
161 BOOST_CHECK(jobs[1]->finished_cancelled());