X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=test%2Fclosed_caption_test.cc;h=86666a44cc686386ca193af6c42a3a9c272e5010;hp=a837c1a079c87bc4fb1eaaf8065437c2f29c9415;hb=3ea24566e4591c0160379a6bee97c5e62b60a028;hpb=aa458290a8283ec8baddea30b3951fbad005277f

diff --git a/test/closed_caption_test.cc b/test/closed_caption_test.cc
index a837c1a07..86666a44c 100644
--- a/test/closed_caption_test.cc
+++ b/test/closed_caption_test.cc
@@ -19,25 +19,27 @@
 */
 
 #include "lib/film.h"
-#include "lib/caption_content.h"
-#include "lib/text_caption_file_content.h"
+#include "lib/text_content.h"
+#include "lib/string_text_file_content.h"
 #include "test.h"
 #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<TextCaptionFileContent> content (new TextCaptionFileContent (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 ());
 
-	content->caption->set_type (CAPTION_CLOSED);
+	content->only_text()->set_type (TEXT_CLOSED_CAPTION);
 
 	film->make_dcp ();
 	BOOST_REQUIRE (!wait_for_jobs ());
@@ -49,7 +51,51 @@ 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 (check.cpls().front()->reels().front()->closed_caption());
+	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());
+}
+
+/** Test multiple closed captions */
+BOOST_AUTO_TEST_CASE (closed_caption_test2)
+{
+	shared_ptr<Film> film = new_test_film2 ("closed_caption_test2");
+	shared_ptr<StringTextFileContent> content1 (new StringTextFileContent("test/data/subrip.srt"));
+	film->examine_and_add_content (content1);
+	shared_ptr<StringTextFileContent> content2 (new StringTextFileContent("test/data/subrip2.srt"));
+	film->examine_and_add_content (content2);
+	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", "fr-FR"));
+	content2->only_text()->set_type (TEXT_CLOSED_CAPTION);
+	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", "it-IT"));
+
+	film->make_dcp ();
+	BOOST_REQUIRE (!wait_for_jobs ());
+
+	dcp::DCP check (film->dir(film->dcp_name()));
+	check.read ();
+
+	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");
 }