120 lines
3.9 KiB
C++
120 lines
3.9 KiB
C++
// Copyright 2018 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/mpd_utils.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <memory>
|
|
|
|
#include "packager/mpd/base/adaptation_set.h"
|
|
#include "packager/mpd/base/mpd_options.h"
|
|
#include "packager/mpd/test/mpd_builder_test_helper.h"
|
|
#include "packager/mpd/test/xml_compare.h"
|
|
|
|
namespace shaka {
|
|
namespace {
|
|
const char kNoLanguage[] = "";
|
|
} // namespace
|
|
|
|
class TestableAdaptationSet : public AdaptationSet {
|
|
public:
|
|
TestableAdaptationSet(const MpdOptions& mpd_options,
|
|
uint32_t* representation_counter)
|
|
: AdaptationSet(kNoLanguage, mpd_options, representation_counter) {}
|
|
};
|
|
|
|
class MpdUtilsTest : public ::testing::Test {
|
|
public:
|
|
MpdUtilsTest() : adaptation_set_(mpd_options_, &representation_counter_) {}
|
|
|
|
protected:
|
|
MpdOptions mpd_options_;
|
|
uint32_t representation_counter_ = 0;
|
|
TestableAdaptationSet adaptation_set_;
|
|
};
|
|
|
|
TEST_F(MpdUtilsTest, ContentProtectionGeneral) {
|
|
const char kMediaInfoWithContentProtection[] =
|
|
"video_info {"
|
|
" codec: 'avc1'"
|
|
" width: 1920"
|
|
" height: 1080"
|
|
" time_scale: 3000"
|
|
" frame_duration: 100"
|
|
"}"
|
|
"protected_content {"
|
|
" default_key_id: '0123456789\x3A\x3B\x3C\x3D\x3E\x3F'"
|
|
" content_protection_entry {"
|
|
" uuid: 'my_uuid'"
|
|
" pssh: 'my_pssh'"
|
|
" }"
|
|
"}"
|
|
"container_type: 1";
|
|
const MediaInfo media_info =
|
|
ConvertToMediaInfo(kMediaInfoWithContentProtection);
|
|
|
|
AddContentProtectionElements(media_info, &adaptation_set_);
|
|
ASSERT_TRUE(adaptation_set_.AddRepresentation(media_info));
|
|
|
|
const char kExpectedOutput[] =
|
|
"<AdaptationSet contentType='video' width='1920'"
|
|
" height='1080' frameRate='3000/100'>"
|
|
" <ContentProtection value='cenc'"
|
|
" schemeIdUri='urn:mpeg:dash:mp4protection:2011'"
|
|
" cenc:default_KID='30313233-3435-3637-3839-3a3b3c3d3e3f'/>"
|
|
" <ContentProtection schemeIdUri='urn:uuid:my_uuid'>"
|
|
" <cenc:pssh>bXlfcHNzaA==</cenc:pssh>"
|
|
" </ContentProtection>"
|
|
" <Representation id='0' bandwidth='0' codecs='avc1'"
|
|
" mimeType='video/mp4'/>"
|
|
"</AdaptationSet>";
|
|
EXPECT_THAT(adaptation_set_.GetXml().get(), XmlNodeEqual(kExpectedOutput));
|
|
}
|
|
|
|
TEST_F(MpdUtilsTest, ContentProtectionMarlin) {
|
|
const char kMediaInfoWithContentProtection[] =
|
|
"video_info {"
|
|
" codec: 'avc1'"
|
|
" width: 1920"
|
|
" height: 1080"
|
|
" time_scale: 3000"
|
|
" frame_duration: 100"
|
|
"}"
|
|
"protected_content {"
|
|
" default_key_id: '0123456789\x3A\x3B\x3C\x3D\x3E\x3F'"
|
|
" content_protection_entry {"
|
|
" uuid: '5e629af5-38da-4063-8977-97ffbd9902d4'"
|
|
" }"
|
|
"}"
|
|
"container_type: 1";
|
|
const MediaInfo media_info =
|
|
ConvertToMediaInfo(kMediaInfoWithContentProtection);
|
|
|
|
AddContentProtectionElements(media_info, &adaptation_set_);
|
|
ASSERT_TRUE(adaptation_set_.AddRepresentation(media_info));
|
|
|
|
const char kExpectedOutput[] =
|
|
"<AdaptationSet contentType='video' width='1920'"
|
|
" height='1080' frameRate='3000/100'>"
|
|
" <ContentProtection value='cenc'"
|
|
" schemeIdUri='urn:mpeg:dash:mp4protection:2011'"
|
|
" cenc:default_KID='30313233-3435-3637-3839-3a3b3c3d3e3f'/>"
|
|
" <ContentProtection"
|
|
" schemeIdUri='urn:uuid:5E629AF5-38DA-4063-8977-97FFBD9902D4'>"
|
|
" <mas:MarlinContentIds>"
|
|
" <mas:MarlinContentId>"
|
|
" urn:marlin:kid:303132333435363738393a3b3c3d3e3f"
|
|
" </mas:MarlinContentId>"
|
|
" </mas:MarlinContentIds>"
|
|
" </ContentProtection>"
|
|
" <Representation id='0' bandwidth='0' codecs='avc1'"
|
|
" mimeType='video/mp4'/>"
|
|
"</AdaptationSet>";
|
|
EXPECT_THAT(adaptation_set_.GetXml().get(), XmlNodeEqual(kExpectedOutput));
|
|
}
|
|
|
|
} // namespace shaka
|