Add some unit tests.

1 files changed, 213 insertions, 0 deletions

diff --git a/test/kdm_naming_test.cc b/test/kdm_naming_test.cc
new file mode 100644
index 000000000..d45e4a44e
--- /dev/null
+++ b/test/kdm_naming_test.cc
@@ -0,0 +1,213 @@
+/*
+    Copyright (C) 2020 Carl Hetherington <cth@carlh.net>
+
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic 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 2 of the License, or
+    (at your option) any later version.
+
+    DCP-o-matic 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 DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
+
+*/
+
+#include "lib/cinema.h"
+#include "lib/screen.h"
+#include "lib/config.h"
+#include "lib/content_factory.h"
+#include "lib/film.h"
+#include "lib/kdm_with_metadata.h"
+#include "test.h"
+#include <boost/test/unit_test.hpp>
+#include <boost/shared_ptr.hpp>
+
+using std::list;
+using std::string;
+using std::vector;
+using boost::shared_ptr;
+using boost::optional;
+using boost::dynamic_pointer_cast;
+
+static
+bool
+confirm_overwrite (boost::filesystem::path)
+{
+	return true;
+}
+
+static shared_ptr<dcpomatic::Screen> cinema_a_screen_1;
+static shared_ptr<dcpomatic::Screen> cinema_a_screen_2;
+static shared_ptr<dcpomatic::Screen> cinema_b_screen_x;
+static shared_ptr<dcpomatic::Screen> cinema_b_screen_y;
+static shared_ptr<dcpomatic::Screen> cinema_b_screen_z;
+
+BOOST_AUTO_TEST_CASE (single_kdm_naming_test)
+{
+	Config* c = Config::instance();
+
+	dcp::Certificate cert = c->decryption_chain()->leaf();
+
+	/* Cinema A: UTC +4:30 */
+	shared_ptr<Cinema> cinema_a (new Cinema("Cinema A", list<string>(), "", 4, 30));
+	cinema_a_screen_1.reset(new dcpomatic::Screen("Screen 1", "", cert, vector<TrustedDevice>()));
+	cinema_a->add_screen (cinema_a_screen_1);
+	cinema_a_screen_2.reset(new dcpomatic::Screen("Screen 2", "", cert, vector<TrustedDevice>()));
+	cinema_a->add_screen (cinema_a_screen_2);
+	c->add_cinema (cinema_a);
+
+	/* Cinema B: UTC -1:00 */
+	shared_ptr<Cinema> cinema_b (new Cinema("Cinema B", list<string>(), "", -1, 0));
+	cinema_b_screen_x.reset(new dcpomatic::Screen("Screen X", "", cert, vector<TrustedDevice>()));
+	cinema_b->add_screen (cinema_b_screen_x);
+	cinema_b_screen_y.reset(new dcpomatic::Screen("Screen Y", "", cert, vector<TrustedDevice>()));
+	cinema_b->add_screen (cinema_b_screen_y);
+	cinema_b_screen_z.reset(new dcpomatic::Screen("Screen Z", "", cert, vector<TrustedDevice>()));
+	cinema_b->add_screen (cinema_b_screen_z);
+	c->add_cinema (cinema_a);
+
+	/* Film */
+	boost::filesystem::remove_all ("build/test/single_kdm_naming_test");
+	shared_ptr<Film> film = new_test_film2 ("single_kdm_naming_test");
+	film->set_name ("my_great_film");
+	film->examine_and_add_content (content_factory("test/data/flat_black.png").front());
+	BOOST_REQUIRE (!wait_for_jobs());
+	film->set_encrypted (true);
+	film->make_dcp ();
+	BOOST_REQUIRE (!wait_for_jobs());
+	film->write_metadata ();
+	vector<CPLSummary> cpls = film->cpls ();
+	BOOST_REQUIRE(cpls.size() == 1);
+
+	dcp::LocalTime from (cert.not_before());
+	from.add_months (2);
+	dcp::LocalTime until (cert.not_after());
+	until.add_months (-2);
+
+	string const from_string = from.date() + " " + from.time_of_day(true, false);
+	string const until_string = until.date() + " " + until.time_of_day(true, false);
+
+	boost::filesystem::path cpl = cpls.front().cpl_file;
+	KDMWithMetadataPtr kdm = kdm_for_screen (
+			film,
+			cpls.front().cpl_file,
+			cinema_a_screen_1,
+			boost::posix_time::time_from_string(from_string),
+			boost::posix_time::time_from_string(until_string),
+			dcp::MODIFIED_TRANSITIONAL_1,
+			false,
+			optional<int>()
+			);
+
+	list<KDMWithMetadataPtr> kdms;
+	kdms.push_back (kdm);
+
+	write_files (
+		kdms,
+		boost::filesystem::path("build/test/single_kdm_naming_test"),
+		dcp::NameFormat("KDM %c - %s - %f - %b - %e"),
+		&confirm_overwrite
+		);
+
+	string from_time = from.time_of_day (true, false);
+	boost::algorithm::replace_all (from_time, ":", "-");
+	string until_time = until.time_of_day (true, false);
+	boost::algorithm::replace_all (until_time, ":", "-");
+
+	string const ref = String::compose("KDM_Cinema_A_-_Screen_1_-_my_great_film_-_%1_%2_-_%3_%4.xml", from.date(), from_time, until.date(), until_time);
+	BOOST_CHECK_MESSAGE (boost::filesystem::exists("build/test/single_kdm_naming_test/" + ref), "File " << ref << " not found");
+}
+
+
+BOOST_AUTO_TEST_CASE (directory_kdm_naming_test, * boost::unit_test::depends_on("single_kdm_naming_test"))
+{
+	using boost::filesystem::path;
+
+	dcp::Certificate cert = Config::instance()->decryption_chain()->leaf();
+
+	/* Film */
+	boost::filesystem::remove_all ("build/test/directory_kdm_naming_test");
+	shared_ptr<Film> film = new_test_film2 ("directory_kdm_naming_test");
+	film->set_name ("my_great_film");
+	film->examine_and_add_content (content_factory("test/data/flat_black.png").front());
+	BOOST_REQUIRE (!wait_for_jobs());
+	film->set_encrypted (true);
+	film->make_dcp ();
+	BOOST_REQUIRE (!wait_for_jobs());
+	film->write_metadata ();
+	vector<CPLSummary> cpls = film->cpls ();
+	BOOST_REQUIRE(cpls.size() == 1);
+
+	dcp::LocalTime from (cert.not_before());
+	from.add_months (2);
+	dcp::LocalTime until (cert.not_after());
+	until.add_months (-2);
+
+	string const from_string = from.date() + " " + from.time_of_day(true, false);
+	string const until_string = until.date() + " " + until.time_of_day(true, false);
+
+	list<shared_ptr<dcpomatic::Screen> > screens;
+	screens.push_back (cinema_a_screen_2);
+	screens.push_back (cinema_b_screen_x);
+	screens.push_back (cinema_a_screen_1);
+	screens.push_back (cinema_b_screen_z);
+
+	path const cpl = cpls.front().cpl_file;
+	string const cpl_id = cpls.front().cpl_id;
+
+	list<KDMWithMetadataPtr> kdms;
+	BOOST_FOREACH (shared_ptr<dcpomatic::Screen> i, screens) {
+		KDMWithMetadataPtr kdm = kdm_for_screen (
+				film,
+				cpls.front().cpl_file,
+				i,
+				boost::posix_time::time_from_string(from_string),
+				boost::posix_time::time_from_string(until_string),
+				dcp::MODIFIED_TRANSITIONAL_1,
+				false,
+				optional<int>()
+				);
+
+		kdms.push_back (kdm);
+	}
+
+	write_directories (
+		collect(kdms),
+		path("build/test/directory_kdm_naming_test"),
+		dcp::NameFormat("%c - %s - %f - %b - %e"),
+		dcp::NameFormat("KDM %c - %s - %f - %b - %e - %i"),
+		&confirm_overwrite
+		);
+
+	string from_time = from.time_of_day (true, false);
+	boost::algorithm::replace_all (from_time, ":", "-");
+	string until_time = until.time_of_day (true, false);
+	boost::algorithm::replace_all (until_time, ":", "-");
+
+	path const base = "build/test/directory_kdm_naming_test";
+	list<KDMWithMetadataPtr>::const_iterator i = kdms.begin ();
+
+	path dir_a = String::compose("Cinema_A_-_%s_-_my_great_film_-_%1_%2_-_%3_%4", from.date(), from_time, until.date(), until_time);
+	BOOST_CHECK_MESSAGE (boost::filesystem::exists(base / dir_a), "Directory " << dir_a << " not found");
+	path dir_b = String::compose("Cinema_B_-_%s_-_my_great_film_-_%1_%2_-_%3_%4", from.date(), from_time, until.date(), until_time);
+	BOOST_CHECK_MESSAGE (boost::filesystem::exists(base / dir_b), "Directory " << dir_b << " not found");
+
+	path ref = String::compose("KDM_Cinema_A_-_Screen_2_-_my_great_film_-_%1_%2_-_%3_%4_-_%5.xml", from.date(), from_time, until.date(), until_time, cpl_id);
+	BOOST_CHECK_MESSAGE (boost::filesystem::exists(base / dir_a / ref), "File " << ref << " not found");
+
+	ref = String::compose("KDM_Cinema_B_-_Screen_X_-_my_great_film_-_%1_%2_-_%3_%4_-_%5.xml", from.date(), from_time, until.date(), until_time, cpl_id);
+	BOOST_CHECK_MESSAGE (boost::filesystem::exists(base / dir_b / ref), "File " << ref << " not found");
+
+	ref = String::compose("KDM_Cinema_A_-_Screen_1_-_my_great_film_-_%1_%2_-_%3_%4_-_%5.xml", from.date(), from_time, until.date(), until_time, cpl_id);
+	BOOST_CHECK_MESSAGE (boost::filesystem::exists(base / dir_a / ref), "File " << ref << " not found");
+
+	ref = String::compose("KDM_Cinema_B_-_Screen_Z_-_my_great_film_-_%1_%2_-_%3_%4_-_%5.xml", from.date(), from_time, until.date(), until_time, cpl_id);
+	BOOST_CHECK_MESSAGE (boost::filesystem::exists(base / dir_b / ref), "File " << ref << " not found");
+}
+