827 lines
28 KiB
C++
827 lines
28 KiB
C++
// Copyright 2014 Google LLC. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include <packager/mpd/base/xml/xml_node.h>
|
|
|
|
#include <list>
|
|
|
|
#include <absl/flags/declare.h>
|
|
#include <absl/flags/flag.h>
|
|
#include <absl/log/log.h>
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
#include <libxml/tree.h>
|
|
|
|
#include <packager/flag_saver.h>
|
|
#include <packager/mpd/base/segment_info.h>
|
|
#include <packager/mpd/test/mpd_builder_test_helper.h>
|
|
#include <packager/mpd/test/xml_compare.h>
|
|
|
|
ABSL_DECLARE_FLAG(bool, segment_template_constant_duration);
|
|
ABSL_DECLARE_FLAG(bool, dash_add_last_segment_number_when_needed);
|
|
|
|
using ::testing::ElementsAre;
|
|
|
|
namespace shaka {
|
|
namespace xml {
|
|
|
|
namespace {
|
|
|
|
// Template so that it works for ContentProtectionXml and
|
|
// ContentProtectionXml::Element.
|
|
template <typename XmlElement>
|
|
void AddAttribute(const std::string& name,
|
|
const std::string& value,
|
|
XmlElement* content_protection_xml) {
|
|
MediaInfo::ContentProtectionXml::AttributeNameValuePair* attribute =
|
|
content_protection_xml->add_attributes();
|
|
attribute->set_name(name);
|
|
attribute->set_value(value);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
// Make sure XmlEqual() is functioning correctly.
|
|
// TODO(rkuroiwa): Move this to a separate file. This requires it to be TEST
|
|
// due to gtest /test
|
|
TEST(XmlNodeTest, MetaTestXmlElementsEqual) {
|
|
static const char kXml1[] =
|
|
"<A>\n"
|
|
" <B\n"
|
|
" c=\"1\""
|
|
" e=\"foobar\""
|
|
" somelongnameattribute=\"somevalue\">\n"
|
|
" <Bchild childvalue=\"3\"\n"
|
|
" f=\"4\"/>\n"
|
|
" </B>\n"
|
|
" <C />\n"
|
|
"</A>";
|
|
|
|
|
|
// This is same as kXml1 but the attributes are reordered. Note that the
|
|
// children are not reordered.
|
|
static const char kXml1AttributeReorder[] =
|
|
"<A>\n"
|
|
" <B\n"
|
|
" c=\"1\""
|
|
" somelongnameattribute=\"somevalue\"\n"
|
|
" e=\"foobar\">"
|
|
" <Bchild childvalue=\"3\"\n"
|
|
" f=\"4\"/>\n"
|
|
" </B>\n"
|
|
" <C />\n"
|
|
"</A>";
|
|
|
|
// <C> is before <B>.
|
|
static const char kXml1ChildrenReordered[] =
|
|
"<A>\n"
|
|
" <C />\n"
|
|
" <B\n"
|
|
" d=\"2\""
|
|
" c=\"1\""
|
|
" somelongnameattribute=\"somevalue\"\n"
|
|
" e=\"foobar\">"
|
|
" <Bchild childvalue=\"3\"\n"
|
|
" f=\"4\"/>\n"
|
|
" </B>\n"
|
|
"</A>";
|
|
|
|
// <C> is before <B>.
|
|
static const char kXml1RemovedAttributes[] =
|
|
"<A>\n"
|
|
" <B\n"
|
|
" d=\"2\"\n>"
|
|
" <Bchild f=\"4\"/>\n"
|
|
" </B>\n"
|
|
" <C />\n"
|
|
"</A>";
|
|
|
|
static const char kXml2[] =
|
|
"<A>\n"
|
|
" <C />\n"
|
|
"</A>";
|
|
|
|
// In XML <C />, <C></C>, and <C/> mean the same thing.
|
|
static const char kXml2DifferentSyntax[] =
|
|
"<A>\n"
|
|
" <C></C>\n"
|
|
"</A>";
|
|
|
|
static const char kXml2MoreDifferentSyntax[] =
|
|
"<A>\n"
|
|
" <C/>\n"
|
|
"</A>";
|
|
|
|
// Identity.
|
|
ASSERT_TRUE(XmlEqual(kXml1, kXml1));
|
|
|
|
// Equivalent.
|
|
ASSERT_TRUE(XmlEqual(kXml1, kXml1AttributeReorder));
|
|
ASSERT_TRUE(XmlEqual(kXml2, kXml2DifferentSyntax));
|
|
ASSERT_TRUE(XmlEqual(kXml2, kXml2MoreDifferentSyntax));
|
|
|
|
// Different.
|
|
ASSERT_FALSE(XmlEqual(kXml1, kXml2));
|
|
ASSERT_FALSE(XmlEqual(kXml1, kXml1ChildrenReordered));
|
|
ASSERT_FALSE(XmlEqual(kXml1, kXml1RemovedAttributes));
|
|
ASSERT_FALSE(XmlEqual(kXml1AttributeReorder, kXml1ChildrenReordered));
|
|
}
|
|
|
|
// Verify that if contents are different, XmlEqual returns false.
|
|
// This is to catch the case where just using xmlNodeGetContent() on elements
|
|
// that have subelements don't quite work well.
|
|
// xmlNodeGetContent(<A>) (for both <A>s) will return "content1content2".
|
|
// But if it is run on <B> for the first XML, it will return "content1", but
|
|
// for second XML will return "c".
|
|
TEST(XmlNodeTest, MetaTestXmlEqualDifferentContent) {
|
|
ASSERT_FALSE(XmlEqual(
|
|
"<A><B>content1</B><B>content2</B></A>",
|
|
"<A><B>c</B><B>ontent1content2</B></A>"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, ExtractReferencedNamespaces) {
|
|
XmlNode grand_child_with_namespace("grand_ns:grand_child");
|
|
grand_child_with_namespace.SetContent("grand child content");
|
|
|
|
XmlNode child("child1");
|
|
child.SetContent("child1 content");
|
|
ASSERT_TRUE(child.AddChild(std::move(grand_child_with_namespace)));
|
|
|
|
XmlNode child_with_namespace("child_ns:child2");
|
|
child_with_namespace.SetContent("child2 content");
|
|
|
|
XmlNode root("root");
|
|
ASSERT_TRUE(root.AddChild(std::move(child)));
|
|
ASSERT_TRUE(root.AddChild(std::move(child_with_namespace)));
|
|
|
|
EXPECT_THAT(root.ExtractReferencedNamespaces(),
|
|
ElementsAre("child_ns", "grand_ns"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, ExtractReferencedNamespacesFromAttributes) {
|
|
XmlNode child("child");
|
|
ASSERT_TRUE(child.SetStringAttribute("child_attribute_ns:attribute",
|
|
"child attribute value"));
|
|
|
|
XmlNode root("root");
|
|
ASSERT_TRUE(root.AddChild(std::move(child)));
|
|
ASSERT_TRUE(root.SetStringAttribute("root_attribute_ns:attribute",
|
|
"root attribute value"));
|
|
|
|
EXPECT_THAT(root.ExtractReferencedNamespaces(),
|
|
ElementsAre("child_attribute_ns", "root_attribute_ns"));
|
|
}
|
|
|
|
// Verify that AddContentProtectionElements work.
|
|
// xmlReadMemory() (used in XmlEqual()) doesn't like XML fragments that have
|
|
// namespaces without context, e.g. <cenc:pssh> element.
|
|
// The MpdBuilderTests work because the MPD element has xmlns:cenc attribute.
|
|
// Tests that have <cenc:pssh> is in mpd_builder_unittest.
|
|
TEST(XmlNodeTest, AddContentProtectionElements) {
|
|
std::list<ContentProtectionElement> content_protections;
|
|
ContentProtectionElement content_protection_widevine;
|
|
content_protection_widevine.scheme_id_uri =
|
|
"urn:uuid:edef8ba9-79d6-4ace-a3c8-27dcd51d21ed";
|
|
content_protection_widevine.value = "SOME bogus Widevine DRM version";
|
|
Element any_element;
|
|
any_element.name = "AnyElement";
|
|
any_element.content = "any content";
|
|
content_protection_widevine.subelements.push_back(any_element);
|
|
content_protections.push_back(content_protection_widevine);
|
|
|
|
ContentProtectionElement content_protection_clearkey;
|
|
content_protection_clearkey.scheme_id_uri =
|
|
"urn:uuid:1077efec-c0b2-4d02-ace3-3c1e52e2fb4b";
|
|
content_protections.push_back(content_protection_clearkey);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddContentProtectionElements(content_protections));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual(
|
|
"<Representation>\n"
|
|
" <ContentProtection\n"
|
|
" schemeIdUri=\"urn:uuid:edef8ba9-79d6-4ace-a3c8-27dcd51d21ed\"\n"
|
|
" value=\"SOME bogus Widevine DRM version\">\n"
|
|
" <AnyElement>any content</AnyElement>\n"
|
|
" </ContentProtection>\n"
|
|
" <ContentProtection\n"
|
|
" schemeIdUri=\"urn:uuid:1077efec-c0b2-4d02-ace3-3c1e52e2fb4b\">"
|
|
" </ContentProtection>\n"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddEC3AudioInfo) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("ec-3");
|
|
audio_info.set_sampling_frequency(48000);
|
|
audio_info.mutable_codec_specific_data()->set_channel_mask(0xF801);
|
|
audio_info.mutable_codec_specific_data()->set_channel_mpeg_value(
|
|
0xFFFFFFFF);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual(
|
|
"<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"tag:dolby.com,2014:dash:audio_channel_configuration:2011\"\n"
|
|
" value=\"F801\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddEC3AudioInfoMPEGScheme) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("ec-3");
|
|
audio_info.set_sampling_frequency(48000);
|
|
audio_info.mutable_codec_specific_data()->set_channel_mask(0xF801);
|
|
audio_info.mutable_codec_specific_data()->set_channel_mpeg_value(6);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(representation,
|
|
XmlNodeEqual("<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"urn:mpeg:mpegB:cicp:ChannelConfiguration\"\n"
|
|
" value=\"6\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddEC3AudioInfoMPEGSchemeJOC) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("ec-3");
|
|
audio_info.set_sampling_frequency(48000);
|
|
audio_info.mutable_codec_specific_data()->set_channel_mask(0xF801);
|
|
audio_info.mutable_codec_specific_data()->set_channel_mpeg_value(6);
|
|
audio_info.mutable_codec_specific_data()->set_ec3_joc_complexity(16);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual(
|
|
"<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"urn:mpeg:mpegB:cicp:ChannelConfiguration\"\n"
|
|
" value=\"6\"/>\n"
|
|
" <SupplementalProperty\n"
|
|
" schemeIdUri=\n"
|
|
" \"tag:dolby.com,2018:dash:EC3_ExtensionType:2018\"\n"
|
|
" value=\"JOC\"/>\n"
|
|
" <SupplementalProperty\n"
|
|
" schemeIdUri=\n"
|
|
" \"tag:dolby.com,2018:dash:EC3_ExtensionComplexityIndex:2018\"\n"
|
|
" value=\"16\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddAC4AudioInfo) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("ac-4.02.01.02");
|
|
audio_info.set_sampling_frequency(48000);
|
|
auto* codec_data = audio_info.mutable_codec_specific_data();
|
|
codec_data->set_channel_mpeg_value(0xFFFFFFFF);
|
|
codec_data->set_channel_mask(0x0000C7);
|
|
codec_data->set_ac4_ims_flag(false);
|
|
codec_data->set_ac4_cbi_flag(false);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual(
|
|
"<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"tag:dolby.com,2015:dash:audio_channel_configuration:2015\"\n"
|
|
" value=\"0000C7\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddAC4AudioInfoMPEGScheme) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("ac-4.02.01.00");
|
|
audio_info.set_sampling_frequency(48000);
|
|
auto* codec_data = audio_info.mutable_codec_specific_data();
|
|
codec_data->set_channel_mpeg_value(2);
|
|
codec_data->set_channel_mask(0x000001);
|
|
codec_data->set_ac4_ims_flag(false);
|
|
codec_data->set_ac4_cbi_flag(false);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(representation,
|
|
XmlNodeEqual("<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"urn:mpeg:mpegB:cicp:ChannelConfiguration\"\n"
|
|
" value=\"2\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddAC4AudioInfoMPEGSchemeIMS) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("ac-4.02.02.00");
|
|
audio_info.set_sampling_frequency(48000);
|
|
auto* codec_data = audio_info.mutable_codec_specific_data();
|
|
codec_data->set_channel_mpeg_value(2);
|
|
codec_data->set_channel_mask(0x000001);
|
|
codec_data->set_ac4_ims_flag(true);
|
|
codec_data->set_ac4_cbi_flag(false);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"urn:mpeg:mpegB:cicp:ChannelConfiguration\"\n"
|
|
" value=\"2\"/>\n"
|
|
" <SupplementalProperty\n"
|
|
" schemeIdUri=\n"
|
|
" \"tag:dolby.com,2016:dash:virtualized_content:2016\"\n"
|
|
" value=\"1\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
class LiveSegmentTimelineTest : public ::testing::Test {
|
|
public:
|
|
LiveSegmentTimelineTest()
|
|
: saver(&FLAGS_segment_template_constant_duration) {}
|
|
|
|
protected:
|
|
void SetUp() override {
|
|
absl::SetFlag(&FLAGS_segment_template_constant_duration, true);
|
|
media_info_.set_segment_template_url("$Number$.m4s");
|
|
}
|
|
|
|
MediaInfo media_info_;
|
|
|
|
private:
|
|
FlagSaver<bool> saver;
|
|
};
|
|
|
|
TEST_F(LiveSegmentTimelineTest, OneSegmentInfo) {
|
|
const uint32_t kStartNumber = 1;
|
|
const int64_t kStartTime = 0;
|
|
const int64_t kDuration = 100;
|
|
const uint64_t kRepeat = 9;
|
|
const bool kIsLowLatency = false;
|
|
|
|
std::list<SegmentInfo> segment_infos = {
|
|
{kStartTime, kDuration, kRepeat},
|
|
};
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddLiveOnlyInfo(media_info_, segment_infos,
|
|
kStartNumber, kIsLowLatency));
|
|
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation>"
|
|
" <SegmentTemplate media=\"$Number$.m4s\" "
|
|
" startNumber=\"1\" duration=\"100\"/>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(LiveSegmentTimelineTest, OneSegmentInfoNonZeroStartTime) {
|
|
const uint32_t kStartNumber = 1;
|
|
const int64_t kNonZeroStartTime = 500;
|
|
const int64_t kDuration = 100;
|
|
const uint64_t kRepeat = 9;
|
|
const bool kIsLowLatency = false;
|
|
|
|
std::list<SegmentInfo> segment_infos = {
|
|
{kNonZeroStartTime, kDuration, kRepeat},
|
|
};
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddLiveOnlyInfo(media_info_, segment_infos,
|
|
kStartNumber, kIsLowLatency));
|
|
|
|
EXPECT_THAT(representation,
|
|
XmlNodeEqual(
|
|
"<Representation>"
|
|
" <SegmentTemplate media=\"$Number$.m4s\" startNumber=\"1\">"
|
|
" <SegmentTimeline>"
|
|
" <S t=\"500\" d=\"100\" r=\"9\"/>"
|
|
" </SegmentTimeline>"
|
|
" </SegmentTemplate>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(LiveSegmentTimelineTest, OneSegmentInfoMatchingStartTimeAndNumber) {
|
|
const uint32_t kStartNumber = 6;
|
|
const int64_t kNonZeroStartTime = 500;
|
|
const int64_t kDuration = 100;
|
|
const uint64_t kRepeat = 9;
|
|
const bool kIsLowLatency = false;
|
|
|
|
std::list<SegmentInfo> segment_infos = {
|
|
{kNonZeroStartTime, kDuration, kRepeat},
|
|
};
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddLiveOnlyInfo(media_info_, segment_infos,
|
|
kStartNumber, kIsLowLatency));
|
|
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation>"
|
|
" <SegmentTemplate media=\"$Number$.m4s\" "
|
|
" startNumber=\"6\" duration=\"100\"/>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(LiveSegmentTimelineTest, AllSegmentsSameDurationExpectLastOne) {
|
|
const uint32_t kStartNumber = 1;
|
|
const bool kIsLowLatency = false;
|
|
|
|
const int64_t kStartTime1 = 0;
|
|
const int64_t kDuration1 = 100;
|
|
const uint64_t kRepeat1 = 9;
|
|
|
|
const int64_t kStartTime2 = kStartTime1 + (kRepeat1 + 1) * kDuration1;
|
|
const int64_t kDuration2 = 200;
|
|
const uint64_t kRepeat2 = 0;
|
|
|
|
std::list<SegmentInfo> segment_infos = {
|
|
{kStartTime1, kDuration1, kRepeat1},
|
|
{kStartTime2, kDuration2, kRepeat2},
|
|
};
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddLiveOnlyInfo(media_info_, segment_infos,
|
|
kStartNumber, kIsLowLatency));
|
|
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation>"
|
|
" <SegmentTemplate media=\"$Number$.m4s\" "
|
|
" startNumber=\"1\" duration=\"100\"/>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(LiveSegmentTimelineTest, SecondSegmentInfoNonZeroRepeat) {
|
|
const uint32_t kStartNumber = 1;
|
|
const bool kIsLowLatency = false;
|
|
|
|
const int64_t kStartTime1 = 0;
|
|
const int64_t kDuration1 = 100;
|
|
const uint64_t kRepeat1 = 9;
|
|
|
|
const int64_t kStartTime2 = kStartTime1 + (kRepeat1 + 1) * kDuration1;
|
|
const int64_t kDuration2 = 200;
|
|
const uint64_t kRepeat2 = 1;
|
|
|
|
std::list<SegmentInfo> segment_infos = {
|
|
{kStartTime1, kDuration1, kRepeat1},
|
|
{kStartTime2, kDuration2, kRepeat2},
|
|
};
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddLiveOnlyInfo(media_info_, segment_infos,
|
|
kStartNumber, kIsLowLatency));
|
|
|
|
EXPECT_THAT(representation,
|
|
XmlNodeEqual(
|
|
"<Representation>"
|
|
" <SegmentTemplate media=\"$Number$.m4s\" startNumber=\"1\">"
|
|
" <SegmentTimeline>"
|
|
" <S t=\"0\" d=\"100\" r=\"9\"/>"
|
|
" <S t=\"1000\" d=\"200\" r=\"1\"/>"
|
|
" </SegmentTimeline>"
|
|
" </SegmentTemplate>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(LiveSegmentTimelineTest, TwoSegmentInfoWithGap) {
|
|
const uint32_t kStartNumber = 1;
|
|
const bool kIsLowLatency = false;
|
|
|
|
const int64_t kStartTime1 = 0;
|
|
const int64_t kDuration1 = 100;
|
|
const uint64_t kRepeat1 = 9;
|
|
|
|
const uint64_t kGap = 100;
|
|
const int64_t kStartTime2 = kGap + kStartTime1 + (kRepeat1 + 1) * kDuration1;
|
|
const int64_t kDuration2 = 200;
|
|
const uint64_t kRepeat2 = 0;
|
|
|
|
std::list<SegmentInfo> segment_infos = {
|
|
{kStartTime1, kDuration1, kRepeat1},
|
|
{kStartTime2, kDuration2, kRepeat2},
|
|
};
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddLiveOnlyInfo(media_info_, segment_infos,
|
|
kStartNumber, kIsLowLatency));
|
|
|
|
EXPECT_THAT(representation,
|
|
XmlNodeEqual(
|
|
"<Representation>"
|
|
" <SegmentTemplate media=\"$Number$.m4s\" startNumber=\"1\">"
|
|
" <SegmentTimeline>"
|
|
" <S t=\"0\" d=\"100\" r=\"9\"/>"
|
|
" <S t=\"1100\" d=\"200\"/>"
|
|
" </SegmentTimeline>"
|
|
" </SegmentTemplate>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(LiveSegmentTimelineTest, LastSegmentNumberSupplementalProperty) {
|
|
const uint32_t kStartNumber = 1;
|
|
const int64_t kStartTime = 0;
|
|
const int64_t kDuration = 100;
|
|
const uint64_t kRepeat = 9;
|
|
const bool kIsLowLatency = false;
|
|
|
|
std::list<SegmentInfo> segment_infos = {
|
|
{kStartTime, kDuration, kRepeat},
|
|
};
|
|
RepresentationXmlNode representation;
|
|
FlagSaver<bool> segment_number_saver(
|
|
&FLAGS_dash_add_last_segment_number_when_needed);
|
|
absl::SetFlag(&FLAGS_dash_add_last_segment_number_when_needed, true);
|
|
|
|
ASSERT_TRUE(representation.AddLiveOnlyInfo(media_info_, segment_infos,
|
|
kStartNumber, kIsLowLatency));
|
|
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation>"
|
|
"<SupplementalProperty schemeIdUri=\"http://dashif.org/"
|
|
"guidelines/last-segment-number\" value=\"10\"/>"
|
|
" <SegmentTemplate media=\"$Number$.m4s\" "
|
|
" startNumber=\"1\" duration=\"100\"/>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
// Creating a separate Test Suite for RepresentationXmlNode::AddVODOnlyInfo
|
|
class OnDemandVODSegmentTest : public ::testing::Test {
|
|
};
|
|
|
|
TEST_F(OnDemandVODSegmentTest, SegmentBase) {
|
|
const char kTestMediaInfo[] =
|
|
"audio_info {\n"
|
|
" codec: 'mp4a.40.2'\n"
|
|
" sampling_frequency: 44100\n"
|
|
" time_scale: 44100\n"
|
|
" num_channels: 2\n"
|
|
"}\n"
|
|
"init_range {\n"
|
|
" begin: 0\n"
|
|
" end: 863\n"
|
|
"}\n"
|
|
"index_range {\n"
|
|
" begin: 864\n"
|
|
" end: 931\n"
|
|
"}\n"
|
|
"media_file_url: 'encrypted_audio.mp4'\n"
|
|
"media_duration_seconds: 24.009434\n"
|
|
"reference_time_scale: 44100\n"
|
|
"presentation_time_offset: 100\n";
|
|
|
|
const MediaInfo media_info = ConvertToMediaInfo(kTestMediaInfo);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddVODOnlyInfo(media_info, false, 100));
|
|
EXPECT_THAT(representation,
|
|
XmlNodeEqual("<Representation>"
|
|
"<BaseURL>encrypted_audio.mp4</BaseURL>"
|
|
"<SegmentBase indexRange=\"864-931\" "
|
|
"timescale=\"44100\" presentationTimeOffset=\"100\">"
|
|
"<Initialization range=\"0-863\"/>"
|
|
"</SegmentBase>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(OnDemandVODSegmentTest, TextInfoBaseUrl) {
|
|
const char kTextMediaInfo[] =
|
|
"text_info {\n"
|
|
" codec: 'ttml'\n"
|
|
" language: 'en'\n"
|
|
" type: SUBTITLE\n"
|
|
"}\n"
|
|
"media_duration_seconds: 35\n"
|
|
"bandwidth: 1000\n"
|
|
"media_file_url: 'subtitle.xml'\n"
|
|
"container_type: CONTAINER_TEXT\n";
|
|
|
|
const MediaInfo media_info = ConvertToMediaInfo(kTextMediaInfo);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddVODOnlyInfo(media_info, false, 100));
|
|
EXPECT_THAT(representation, XmlNodeEqual("<Representation>"
|
|
"<BaseURL>subtitle.xml</BaseURL>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(OnDemandVODSegmentTest, TextInfoWithPresentationOffset) {
|
|
const char kTextMediaInfo[] =
|
|
"text_info {\n"
|
|
" codec: 'ttml'\n"
|
|
" language: 'en'\n"
|
|
" type: SUBTITLE\n"
|
|
"}\n"
|
|
"media_duration_seconds: 35\n"
|
|
"bandwidth: 1000\n"
|
|
"media_file_url: 'subtitle.xml'\n"
|
|
"container_type: CONTAINER_TEXT\n"
|
|
"presentation_time_offset: 100\n";
|
|
|
|
const MediaInfo media_info = ConvertToMediaInfo(kTextMediaInfo);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddVODOnlyInfo(media_info, false, 100));
|
|
|
|
EXPECT_THAT(representation,
|
|
XmlNodeEqual("<Representation>"
|
|
"<SegmentList presentationTimeOffset=\"100\">"
|
|
"<SegmentURL media=\"subtitle.xml\"/>"
|
|
"</SegmentList>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(OnDemandVODSegmentTest, SegmentListWithoutUrls) {
|
|
const char kTestMediaInfo[] =
|
|
"audio_info {\n"
|
|
" codec: 'mp4a.40.2'\n"
|
|
" sampling_frequency: 44100\n"
|
|
" time_scale: 44100\n"
|
|
" num_channels: 2\n"
|
|
"}\n"
|
|
"init_range {\n"
|
|
" begin: 0\n"
|
|
" end: 863\n"
|
|
"}\n"
|
|
"index_range {\n"
|
|
" begin: 864\n"
|
|
" end: 931\n"
|
|
"}\n"
|
|
"media_file_url: 'encrypted_audio.mp4'\n"
|
|
"media_duration_seconds: 24.009434\n"
|
|
"reference_time_scale: 44100\n"
|
|
"presentation_time_offset: 100\n";
|
|
|
|
const MediaInfo media_info = ConvertToMediaInfo(kTestMediaInfo);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddVODOnlyInfo(media_info, true, 100));
|
|
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation>"
|
|
"<BaseURL>encrypted_audio.mp4</BaseURL>"
|
|
"<SegmentList timescale=\"44100\" duration=\"4410000\" "
|
|
"presentationTimeOffset=\"100\">"
|
|
"<Initialization range=\"0-863\"/>"
|
|
"</SegmentList>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST_F(OnDemandVODSegmentTest, SegmentUrlWithMediaRanges) {
|
|
const char kTextMediaInfo[] =
|
|
"audio_info {\n"
|
|
" codec: 'mp4a.40.2'\n"
|
|
" sampling_frequency: 44100\n"
|
|
" time_scale: 44100\n"
|
|
" num_channels: 2\n"
|
|
"}\n"
|
|
"init_range {\n"
|
|
" begin: 0\n"
|
|
" end: 863\n"
|
|
"}\n"
|
|
"index_range {\n"
|
|
" begin: 864\n"
|
|
" end: 931\n"
|
|
"}\n"
|
|
"media_file_url: 'encrypted_audio.mp4'\n"
|
|
"media_duration_seconds: 24.009434\n"
|
|
"reference_time_scale: 44100\n"
|
|
"presentation_time_offset: 100\n"
|
|
"subsegment_ranges {\n"
|
|
" begin: 932\n"
|
|
" end: 9999\n"
|
|
"}\n"
|
|
"subsegment_ranges {\n"
|
|
" begin: 10000\n"
|
|
" end: 11000\n"
|
|
"}\n";
|
|
|
|
const MediaInfo media_info = ConvertToMediaInfo(kTextMediaInfo);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddVODOnlyInfo(media_info, true, 100));
|
|
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation>"
|
|
"<BaseURL>encrypted_audio.mp4</BaseURL>"
|
|
"<SegmentList timescale=\"44100\" duration=\"4410000\" "
|
|
"presentationTimeOffset=\"100\">"
|
|
"<Initialization range=\"0-863\"/>"
|
|
"<SegmentURL mediaRange=\"932-9999\"/>"
|
|
"<SegmentURL mediaRange=\"10000-11000\"/>"
|
|
"</SegmentList>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
class LowLatencySegmentTest : public ::testing::Test {
|
|
protected:
|
|
void SetUp() override {
|
|
media_info_.set_init_segment_url("init.m4s");
|
|
media_info_.set_segment_template_url("$Number$.m4s");
|
|
media_info_.set_reference_time_scale(90000);
|
|
media_info_.set_availability_time_offset(4.9750987314);
|
|
media_info_.set_segment_duration(450000);
|
|
}
|
|
|
|
MediaInfo media_info_;
|
|
};
|
|
|
|
TEST_F(LowLatencySegmentTest, LowLatencySegmentTemplate) {
|
|
const uint32_t kStartNumber = 1;
|
|
const uint64_t kDuration = 100;
|
|
const uint64_t kRepeat = 0;
|
|
const bool kIsLowLatency = true;
|
|
|
|
std::list<SegmentInfo> segment_infos = {
|
|
{kStartNumber, kDuration, kRepeat},
|
|
};
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddLiveOnlyInfo(media_info_, segment_infos,
|
|
kStartNumber, kIsLowLatency));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation>"
|
|
" <SegmentTemplate timescale=\"90000\" duration=\"450000\" "
|
|
" availabilityTimeOffset=\"4.975099\" "
|
|
" availabilityTimeComplete=\"false\" "
|
|
" initialization=\"init.m4s\" "
|
|
" media=\"$Number$.m4s\" "
|
|
" startNumber=\"1\"/>"
|
|
"</Representation>"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddDTSCAudioInfo) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("dtsc");
|
|
audio_info.set_sampling_frequency(48000);
|
|
audio_info.set_num_channels(6);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual(
|
|
"<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"tag:dts.com,2014:dash:audio_channel_configuration:2012\"\n"
|
|
" value=\"6\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddDTSEAudioInfo) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("dtse");
|
|
audio_info.set_sampling_frequency(48000);
|
|
audio_info.set_num_channels(6);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual(
|
|
"<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"tag:dts.com,2014:dash:audio_channel_configuration:2012\"\n"
|
|
" value=\"6\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
TEST(XmlNodeTest, AddDTSXAudioInfo) {
|
|
MediaInfo::AudioInfo audio_info;
|
|
audio_info.set_codec("dtsx");
|
|
audio_info.set_sampling_frequency(48000);
|
|
audio_info.mutable_codec_specific_data()->set_channel_mask(0x3F);
|
|
|
|
RepresentationXmlNode representation;
|
|
ASSERT_TRUE(representation.AddAudioInfo(audio_info));
|
|
EXPECT_THAT(
|
|
representation,
|
|
XmlNodeEqual("<Representation audioSamplingRate=\"48000\">\n"
|
|
" <AudioChannelConfiguration\n"
|
|
" schemeIdUri=\n"
|
|
" \"tag:dts.com,2018:uhd:audio_channel_configuration\"\n"
|
|
" value=\"0000003F\"/>\n"
|
|
"</Representation>\n"));
|
|
}
|
|
|
|
} // namespace xml
|
|
} // namespace shaka
|