Subtitle language test updates.

[dcpomatic.git] / test / closed_caption_test.cc

diff --git a/test/closed_caption_test.cc b/test/closed_caption_test.cc
index 35abf1b02c5a14f5d80f8815329967f7588d1a0a..86666a44cc686386ca193af6c42a3a9c272e5010 100644 (file)
--- a/test/closed_caption_test.cc
+++ b/test/closed_caption_test.cc
@@ -25,15 +25,17 @@
 #include <dcp/dcp.h>
 #include <dcp/cpl.h>
 #include <dcp/reel.h>
+#include <dcp/reel_closed_caption_asset.h>
 #include <boost/test/unit_test.hpp>
 
+using std::list;
 using boost::shared_ptr;
 
 /** Basic test that Interop closed captions are written */
 BOOST_AUTO_TEST_CASE (closed_caption_test1)
 {
        shared_ptr<Film> film = new_test_film2 ("closed_caption_test1");
-       shared_ptr<StringTextFileContent> content (new StringTextFileContent (film, "test/data/subrip.srt"));
+       shared_ptr<StringTextFileContent> content (new StringTextFileContent("test/data/subrip.srt"));
        film->examine_and_add_content (content);
        BOOST_REQUIRE (!wait_for_jobs ());
 
@@ -49,8 +51,8 @@ BOOST_AUTO_TEST_CASE (closed_caption_test1)
        dcp::DCP check (film->dir(film->dcp_name()));
        check.read ();
 
-       BOOST_REQUIRE_EQUAL (check.cpls().size(), 1);
-       BOOST_REQUIRE_EQUAL (check.cpls().front()->reels().size(), 1);
+       BOOST_REQUIRE_EQUAL (check.cpls().size(), 1U);
+       BOOST_REQUIRE_EQUAL (check.cpls().front()->reels().size(), 1U);
        BOOST_REQUIRE (!check.cpls().front()->reels().front()->closed_captions().empty());
 }
 
@@ -58,20 +60,20 @@ BOOST_AUTO_TEST_CASE (closed_caption_test1)
 BOOST_AUTO_TEST_CASE (closed_caption_test2)
 {
        shared_ptr<Film> film = new_test_film2 ("closed_caption_test2");
-       shared_ptr<StringTextFileContent> content1 (new StringTextFileContent (film, "test/data/subrip.srt"));
+       shared_ptr<StringTextFileContent> content1 (new StringTextFileContent("test/data/subrip.srt"));
        film->examine_and_add_content (content1);
-       shared_ptr<StringTextFileContent> content2 (new StringTextFileContent (film, "test/data/subrip2.srt"));
+       shared_ptr<StringTextFileContent> content2 (new StringTextFileContent("test/data/subrip2.srt"));
        film->examine_and_add_content (content2);
-       shared_ptr<StringTextFileContent> content3 (new StringTextFileContent (film, "test/data/subrip3.srt"));
+       shared_ptr<StringTextFileContent> content3 (new StringTextFileContent("test/data/subrip3.srt"));
        film->examine_and_add_content (content3);
        BOOST_REQUIRE (!wait_for_jobs ());
 
        content1->only_text()->set_type (TEXT_CLOSED_CAPTION);
-       content1->only_text()->set_dcp_track (DCPTextTrack("First track", "French"));
+       content1->only_text()->set_dcp_track (DCPTextTrack("First track", "fr-FR"));
        content2->only_text()->set_type (TEXT_CLOSED_CAPTION);
-       content2->only_text()->set_dcp_track (DCPTextTrack("Second track", "German"));
+       content2->only_text()->set_dcp_track (DCPTextTrack("Second track", "de-DE"));
        content3->only_text()->set_type (TEXT_CLOSED_CAPTION);
-       content3->only_text()->set_dcp_track (DCPTextTrack("Third track", "Italian"));
+       content3->only_text()->set_dcp_track (DCPTextTrack("Third track", "it-IT"));
 
        film->make_dcp ();
        BOOST_REQUIRE (!wait_for_jobs ());
@@ -79,8 +81,21 @@ BOOST_AUTO_TEST_CASE (closed_caption_test2)
        dcp::DCP check (film->dir(film->dcp_name()));
        check.read ();
 
-       BOOST_REQUIRE_EQUAL (check.cpls().size(), 1);
-       BOOST_REQUIRE_EQUAL (check.cpls().front()->reels().size(), 1);
-       std::cout << !check.cpls().front()->reels().front()->closed_captions().size() << "\n";
-       BOOST_REQUIRE_EQUAL (!check.cpls().front()->reels().front()->closed_captions().size(), 3);
+       BOOST_REQUIRE_EQUAL (check.cpls().size(), 1U);
+       BOOST_REQUIRE_EQUAL (check.cpls().front()->reels().size(), 1U);
+       list<shared_ptr<dcp::ReelClosedCaptionAsset> > ccaps = check.cpls().front()->reels().front()->closed_captions();
+       BOOST_REQUIRE_EQUAL (ccaps.size(), 3U);
+
+       list<shared_ptr<dcp::ReelClosedCaptionAsset> >::const_iterator i = ccaps.begin ();
+       BOOST_CHECK_EQUAL ((*i)->annotation_text(), "First track");
+       BOOST_REQUIRE (static_cast<bool>((*i)->language()));
+       BOOST_CHECK_EQUAL ((*i)->language().get(), "fr-FR");
+       ++i;
+       BOOST_CHECK_EQUAL ((*i)->annotation_text(), "Second track");
+       BOOST_REQUIRE (static_cast<bool>((*i)->language()));
+       BOOST_CHECK_EQUAL ((*i)->language().get(), "de-DE");
+       ++i;
+       BOOST_CHECK_EQUAL ((*i)->annotation_text(), "Third track");
+       BOOST_REQUIRE (static_cast<bool>((*i)->language()));
+       BOOST_CHECK_EQUAL ((*i)->language().get(), "it-IT");
 }