Cleanup: test tidying.
[dcpomatic.git] / test / cpl_hash_test.cc
1 /*
2     Copyright (C) 2020-2021 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
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.
10
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.
15
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/>.
18
19 */
20
21
22 /** @file  test/cpl_hash_test.cc
23  *  @brief Make sure that <Hash> tags are always written to CPLs where required.
24  *  @ingroup feature
25  */
26
27
28 #include "lib/content_factory.h"
29 #include "lib/cross.h"
30 #include "lib/dcp_content.h"
31 #include "lib/film.h"
32 #include "test.h"
33 #include <boost/algorithm/string.hpp>
34 #include <boost/test/unit_test.hpp>
35
36
37 using std::make_shared;
38 using std::string;
39
40
41 BOOST_AUTO_TEST_CASE (hash_added_to_imported_dcp_test)
42 {
43         using namespace boost::filesystem;
44
45         string const ov_name = "hash_added_to_imported_dcp_test_ov";
46         auto ov = new_test_film2(
47                 ov_name,
48                 content_factory("test/data/flat_red.png")
49                 );
50         make_and_verify_dcp (ov);
51
52         /* Remove <Hash> tags from the CPL */
53         for (auto i: directory_iterator(String::compose("build/test/%1/%2", ov_name, ov->dcp_name()))) {
54                 if (boost::algorithm::starts_with(i.path().filename().string(), "cpl_")) {
55                         dcp::File in(i.path(), "r");
56                         BOOST_REQUIRE (in);
57                         dcp::File out(i.path().string() + ".tmp", "w");
58                         BOOST_REQUIRE (out);
59                         char buffer[256];
60                         while (in.gets(buffer, sizeof(buffer))) {
61                                 if (string(buffer).find("Hash") == string::npos) {
62                                         out.puts(buffer);
63                                 }
64                         }
65                         in.close();
66                         out.close();
67                         rename (i.path().string() + ".tmp", i.path());
68                 }
69         }
70
71         string const vf_name = "hash_added_to_imported_dcp_test_vf";
72         auto ov_content = make_shared<DCPContent>(String::compose("build/test/%1/%2", ov_name, ov->dcp_name()));
73         auto vf = new_test_film2 (
74                 vf_name, { ov_content }
75                 );
76
77         ov_content->set_reference_video (true);
78         make_and_verify_dcp (vf, {dcp::VerificationNote::Code::EXTERNAL_ASSET});
79
80         /* Check for Hash tags in the VF DCP */
81         int hashes = 0;
82         for (auto i: directory_iterator(String::compose("build/test/%1/%2", vf_name, vf->dcp_name()))) {
83                 if (boost::algorithm::starts_with(i.path().filename().string(), "cpl_")) {
84                         dcp::File in(i.path(), "r");
85                         BOOST_REQUIRE (in);
86                         char buffer[256];
87                         while (in.gets(buffer, sizeof(buffer))) {
88                                 if (string(buffer).find("Hash") != string::npos) {
89                                         ++hashes;
90                                 }
91                         }
92                 }
93         }
94         BOOST_CHECK_EQUAL (hashes, 2);
95 }
96