Fix test for changes to content_factory.
[dcpomatic.git] / test / required_disk_space_test.cc
1 /*
2     Copyright (C) 2016 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 #include "lib/content_factory.h"
22 #include "lib/film.h"
23 #include "lib/dcp_content.h"
24 #include "test.h"
25 #include <boost/test/unit_test.hpp>
26
27 using boost::shared_ptr;
28 using boost::dynamic_pointer_cast;
29
30 void check_within_n (int64_t a, int64_t b, int64_t n)
31 {
32         BOOST_CHECK (abs (a - b) <= n);
33 }
34
35
36 BOOST_AUTO_TEST_CASE (required_disk_space_test)
37 {
38         shared_ptr<Film> film = new_test_film ("required_disk_space_test");
39         film->set_j2k_bandwidth (100000000);
40         film->set_audio_channels (6);
41         shared_ptr<Content> content_a = content_factory(film, "test/data/flat_blue.png").front();
42         BOOST_REQUIRE (content_a);
43         film->examine_and_add_content (content_a);
44         shared_ptr<DCPContent> content_b (new DCPContent (film, "test/data/burnt_subtitle_test_dcp"));
45         film->examine_and_add_content (content_b);
46         wait_for_jobs ();
47         film->write_metadata ();
48
49         check_within_n (
50                 film->required_disk_space(),
51                 289LL * (100000000 / 8) / 24 +  // video
52                 289LL * 48000 * 6 * 3 / 24 +    // audio
53                 65536,                          // extra
54                 16
55                 );
56
57         content_b->set_reference_video (true);
58
59         check_within_n (
60                 film->required_disk_space(),
61                 240LL * (100000000 / 8) / 24 +  // video
62                 289LL * 48000 * 6 * 3 / 24 +    // audio
63                 65536,                          // extra
64                 16
65                 );
66
67         content_b->set_reference_audio (true);
68
69         check_within_n (
70                 film->required_disk_space(),
71                 240LL * (100000000 / 8) / 24 +  // video
72                 240LL * 48000 * 6 * 3 / 24 +    // audio
73                 65536,                          // extra
74                 16
75                 );
76 }