2016-03-25 08:39:07 +00:00
|
|
|
// Copyright 2016 Google Inc. 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 <gmock/gmock.h>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
|
2017-06-15 20:00:28 +00:00
|
|
|
#include "packager/base/files/file_path.h"
|
2017-07-10 18:26:22 +00:00
|
|
|
#include "packager/file/file.h"
|
2016-03-25 08:39:07 +00:00
|
|
|
#include "packager/hls/base/master_playlist.h"
|
|
|
|
#include "packager/hls/base/media_playlist.h"
|
|
|
|
#include "packager/hls/base/mock_media_playlist.h"
|
2016-07-07 19:34:07 +00:00
|
|
|
#include "packager/version/version.h"
|
2016-03-25 08:39:07 +00:00
|
|
|
|
2016-05-20 21:19:33 +00:00
|
|
|
namespace shaka {
|
2016-03-25 08:39:07 +00:00
|
|
|
namespace hls {
|
|
|
|
|
2018-01-26 23:53:58 +00:00
|
|
|
using base::FilePath;
|
|
|
|
using ::testing::_;
|
2016-03-25 08:39:07 +00:00
|
|
|
using ::testing::AtLeast;
|
2017-06-14 19:25:17 +00:00
|
|
|
using ::testing::NotNull;
|
2016-03-25 08:39:07 +00:00
|
|
|
using ::testing::Return;
|
|
|
|
using ::testing::ReturnRef;
|
2017-06-14 19:25:17 +00:00
|
|
|
using ::testing::SetArgPointee;
|
2017-06-03 00:05:47 +00:00
|
|
|
using ::testing::StrEq;
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
namespace {
|
|
|
|
const char kDefaultMasterPlaylistName[] = "playlist.m3u8";
|
2018-01-17 23:43:41 +00:00
|
|
|
const char kDefaultLanguage[] = "en";
|
2017-06-14 19:25:17 +00:00
|
|
|
const uint32_t kWidth = 800;
|
|
|
|
const uint32_t kHeight = 600;
|
2017-06-30 20:42:46 +00:00
|
|
|
const HlsPlaylistType kVodPlaylist = HlsPlaylistType::kVod;
|
2018-01-26 23:53:58 +00:00
|
|
|
|
|
|
|
std::unique_ptr<MockMediaPlaylist> CreateVideoPlaylist(
|
|
|
|
const std::string& filename,
|
|
|
|
const std::string& codec,
|
|
|
|
uint64_t bitrate) {
|
|
|
|
const char kNoName[] = "";
|
|
|
|
const char kNoGroup[] = "";
|
|
|
|
|
|
|
|
std::unique_ptr<MockMediaPlaylist> playlist(
|
|
|
|
new MockMediaPlaylist(kVodPlaylist, filename, kNoName, kNoGroup));
|
|
|
|
|
|
|
|
playlist->SetStreamTypeForTesting(
|
|
|
|
MediaPlaylist::MediaPlaylistStreamType::kVideo);
|
|
|
|
playlist->SetCodecForTesting(codec);
|
|
|
|
|
|
|
|
EXPECT_CALL(*playlist, Bitrate())
|
|
|
|
.Times(AtLeast(1))
|
|
|
|
.WillRepeatedly(Return(bitrate));
|
|
|
|
EXPECT_CALL(*playlist, GetDisplayResolution(NotNull(), NotNull()))
|
|
|
|
.WillRepeatedly(DoAll(SetArgPointee<0>(kWidth), SetArgPointee<1>(kHeight),
|
|
|
|
Return(true)));
|
|
|
|
|
|
|
|
return playlist;
|
|
|
|
}
|
|
|
|
|
2018-02-02 22:45:52 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> CreateIframePlaylist(
|
|
|
|
const std::string& filename,
|
|
|
|
const std::string& codec,
|
|
|
|
uint64_t bitrate) {
|
|
|
|
auto playlist = CreateVideoPlaylist(filename, codec, bitrate);
|
|
|
|
playlist->SetStreamTypeForTesting(
|
|
|
|
MediaPlaylist::MediaPlaylistStreamType::kVideoIFramesOnly);
|
|
|
|
return playlist;
|
|
|
|
}
|
|
|
|
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> CreateAudioPlaylist(
|
|
|
|
const std::string& filename,
|
|
|
|
const std::string& name,
|
|
|
|
const std::string& group,
|
|
|
|
const std::string& codec,
|
|
|
|
const std::string& language,
|
|
|
|
uint64_t channels,
|
|
|
|
uint64_t bitrate) {
|
|
|
|
std::unique_ptr<MockMediaPlaylist> playlist(
|
|
|
|
new MockMediaPlaylist(kVodPlaylist, filename, name, group));
|
|
|
|
|
|
|
|
EXPECT_CALL(*playlist, GetNumChannels()).WillRepeatedly(Return(channels));
|
|
|
|
|
|
|
|
playlist->SetStreamTypeForTesting(
|
|
|
|
MediaPlaylist::MediaPlaylistStreamType::kAudio);
|
|
|
|
playlist->SetCodecForTesting(codec);
|
2018-04-05 01:27:57 +00:00
|
|
|
playlist->SetLanguageForTesting(language);
|
2018-01-26 23:53:58 +00:00
|
|
|
|
|
|
|
EXPECT_CALL(*playlist, Bitrate())
|
|
|
|
.Times(AtLeast(1))
|
|
|
|
.WillRepeatedly(Return(bitrate));
|
|
|
|
EXPECT_CALL(*playlist, GetDisplayResolution(NotNull(), NotNull())).Times(0);
|
|
|
|
|
|
|
|
return playlist;
|
|
|
|
}
|
2018-01-31 19:14:15 +00:00
|
|
|
|
2018-02-02 22:45:52 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> CreateTextPlaylist(
|
|
|
|
const std::string& filename,
|
|
|
|
const std::string& name,
|
|
|
|
const std::string& group,
|
|
|
|
const std::string& language) {
|
2018-01-31 19:14:15 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> playlist(
|
|
|
|
new MockMediaPlaylist(kVodPlaylist, filename, name, group));
|
|
|
|
|
|
|
|
playlist->SetStreamTypeForTesting(
|
|
|
|
MediaPlaylist::MediaPlaylistStreamType::kSubtitle);
|
2018-04-05 01:27:57 +00:00
|
|
|
playlist->SetLanguageForTesting(language);
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
return playlist;
|
|
|
|
}
|
2016-03-25 08:39:07 +00:00
|
|
|
} // namespace
|
|
|
|
|
|
|
|
class MasterPlaylistTest : public ::testing::Test {
|
|
|
|
protected:
|
2017-06-15 20:00:28 +00:00
|
|
|
MasterPlaylistTest()
|
2018-01-17 23:43:41 +00:00
|
|
|
: master_playlist_(kDefaultMasterPlaylistName, kDefaultLanguage),
|
2017-06-15 20:00:28 +00:00
|
|
|
test_output_dir_("memory://test_dir"),
|
|
|
|
master_playlist_path_(
|
|
|
|
FilePath::FromUTF8Unsafe(test_output_dir_)
|
|
|
|
.Append(FilePath::FromUTF8Unsafe(kDefaultMasterPlaylistName))
|
|
|
|
.AsUTF8Unsafe()) {}
|
2016-03-25 08:39:07 +00:00
|
|
|
|
2018-01-26 23:53:58 +00:00
|
|
|
void SetUp() override { SetPackagerVersionForTesting("test"); }
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
MasterPlaylist master_playlist_;
|
|
|
|
std::string test_output_dir_;
|
2017-06-15 20:00:28 +00:00
|
|
|
std::string master_playlist_path_;
|
2016-03-25 08:39:07 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistOneVideo) {
|
2018-01-26 23:53:58 +00:00
|
|
|
const uint64_t kBitRate = 435889;
|
|
|
|
|
|
|
|
std::unique_ptr<MockMediaPlaylist> mock_playlist =
|
|
|
|
CreateVideoPlaylist("media1.m3u8", "avc1", kBitRate);
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://myplaylistdomain.com/";
|
2018-02-02 17:14:46 +00:00
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(kBaseUrl, test_output_dir_,
|
|
|
|
{mock_playlist.get()}));
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
std::string actual;
|
2017-07-10 18:26:22 +00:00
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
2016-07-07 19:34:07 +00:00
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2017-06-14 19:25:17 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=435889,CODECS=\"avc1\",RESOLUTION=800x600\n"
|
2016-03-25 08:39:07 +00:00
|
|
|
"http://myplaylistdomain.com/media1.m3u8\n";
|
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
|
|
|
|
2018-02-02 22:45:52 +00:00
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistOneIframePlaylist) {
|
|
|
|
const uint64_t kBitRate = 435889;
|
|
|
|
|
|
|
|
std::unique_ptr<MockMediaPlaylist> mock_playlist =
|
|
|
|
CreateIframePlaylist("media1.m3u8", "avc1", kBitRate);
|
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://myplaylistdomain.com/";
|
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(kBaseUrl, test_output_dir_,
|
|
|
|
{mock_playlist.get()}));
|
|
|
|
|
|
|
|
std::string actual;
|
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-02-02 22:45:52 +00:00
|
|
|
"#EXT-X-I-FRAME-STREAM-INF:BANDWIDTH=435889,CODECS=\"avc1\",RESOLUTION="
|
|
|
|
"800x600,URI=\"http://myplaylistdomain.com/media1.m3u8\"\n";
|
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
|
|
|
|
2016-03-25 08:39:07 +00:00
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistVideoAndAudio) {
|
2018-01-26 23:53:58 +00:00
|
|
|
const uint64_t kVideo1BitRate = 300000;
|
|
|
|
const uint64_t kVideo2BitRate = 700000;
|
|
|
|
|
|
|
|
const uint64_t kAudio1BitRate = 50000;
|
|
|
|
const uint64_t kAudio2BitRate = 60000;
|
|
|
|
|
|
|
|
const uint64_t kAudio1Channels = 2;
|
|
|
|
const uint64_t kAudio2Channels = 5;
|
|
|
|
|
2016-03-25 08:39:07 +00:00
|
|
|
// First video, sd.m3u8.
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> sd_video_playlist =
|
|
|
|
CreateVideoPlaylist("sd.m3u8", "sdvideocodec", kVideo1BitRate);
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
// Second video, hd.m3u8.
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> hd_video_playlist =
|
|
|
|
CreateVideoPlaylist("hd.m3u8", "hdvideocodec", kVideo2BitRate);
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
// First audio, english.m3u8.
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> english_playlist =
|
|
|
|
CreateAudioPlaylist("eng.m3u8", "english", "audiogroup", "audiocodec",
|
|
|
|
"en", kAudio1Channels, kAudio1BitRate);
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
// Second audio, spanish.m3u8.
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> spanish_playlist =
|
|
|
|
CreateAudioPlaylist("spa.m3u8", "espanol", "audiogroup", "audiocodec",
|
|
|
|
"es", kAudio2Channels, kAudio2BitRate);
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://playlists.org/";
|
2018-02-02 17:14:46 +00:00
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(
|
|
|
|
kBaseUrl, test_output_dir_,
|
|
|
|
{sd_video_playlist.get(), hd_video_playlist.get(), english_playlist.get(),
|
|
|
|
spanish_playlist.get()}));
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
std::string actual;
|
2017-07-10 18:26:22 +00:00
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
2016-07-07 19:34:07 +00:00
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2017-12-09 01:53:17 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://playlists.org/eng.m3u8\","
|
|
|
|
"GROUP-ID=\"audiogroup\",LANGUAGE=\"en\",NAME=\"english\","
|
2018-01-17 23:43:41 +00:00
|
|
|
"DEFAULT=YES,AUTOSELECT=YES,CHANNELS=\"2\"\n"
|
2017-12-09 01:53:17 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://playlists.org/spa.m3u8\","
|
|
|
|
"GROUP-ID=\"audiogroup\",LANGUAGE=\"es\",NAME=\"espanol\","
|
2018-01-17 23:43:41 +00:00
|
|
|
"AUTOSELECT=YES,CHANNELS=\"5\"\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2017-12-09 01:53:17 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=360000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audiogroup\"\n"
|
2016-03-25 08:39:07 +00:00
|
|
|
"http://playlists.org/sd.m3u8\n"
|
2017-12-09 01:53:17 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=760000,CODECS=\"hdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audiogroup\"\n"
|
2016-03-25 08:39:07 +00:00
|
|
|
"http://playlists.org/hd.m3u8\n";
|
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistMultipleAudioGroups) {
|
2018-01-26 23:53:58 +00:00
|
|
|
const uint64_t kVideoBitRate = 300000;
|
|
|
|
|
|
|
|
const uint64_t kAudio1BitRate = 50000;
|
|
|
|
const uint64_t kAudio2BitRate = 100000;
|
|
|
|
|
|
|
|
const uint64_t kAudio1Channels = 1;
|
|
|
|
const uint64_t kAudio2Channels = 8;
|
|
|
|
|
|
|
|
// First video, sd.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> video_playlist =
|
|
|
|
CreateVideoPlaylist("video.m3u8", "videocodec", kVideoBitRate);
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
// First audio, eng_lo.m3u8.
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> eng_lo_playlist = CreateAudioPlaylist(
|
|
|
|
"eng_lo.m3u8", "english_lo", "audio_lo", "audiocodec_lo", "en",
|
|
|
|
kAudio1Channels, kAudio1BitRate);
|
2016-03-25 08:39:07 +00:00
|
|
|
|
2018-01-17 23:43:41 +00:00
|
|
|
// Second audio, eng_hi.m3u8.
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> eng_hi_playlist = CreateAudioPlaylist(
|
|
|
|
"eng_hi.m3u8", "english_hi", "audio_hi", "audiocodec_hi", "en",
|
|
|
|
kAudio2Channels, kAudio2BitRate);
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://anydomain.com/";
|
2018-02-02 17:14:46 +00:00
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(
|
|
|
|
kBaseUrl, test_output_dir_,
|
|
|
|
{video_playlist.get(), eng_lo_playlist.get(), eng_hi_playlist.get()}));
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
std::string actual;
|
2017-07-10 18:26:22 +00:00
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
2016-03-25 08:39:07 +00:00
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
2016-07-07 19:34:07 +00:00
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2017-12-09 01:53:17 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://anydomain.com/eng_hi.m3u8\","
|
|
|
|
"GROUP-ID=\"audio_hi\",LANGUAGE=\"en\",NAME=\"english_hi\","
|
2018-01-17 23:43:41 +00:00
|
|
|
"DEFAULT=YES,AUTOSELECT=YES,CHANNELS=\"8\"\n"
|
2017-12-09 01:53:17 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://anydomain.com/eng_lo.m3u8\","
|
|
|
|
"GROUP-ID=\"audio_lo\",LANGUAGE=\"en\",NAME=\"english_lo\","
|
2018-01-17 23:43:41 +00:00
|
|
|
"DEFAULT=YES,AUTOSELECT=YES,CHANNELS=\"1\"\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2017-12-09 01:53:17 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=400000,CODECS=\"videocodec,audiocodec_hi\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio_hi\"\n"
|
2016-03-25 08:39:07 +00:00
|
|
|
"http://anydomain.com/video.m3u8\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2017-12-09 01:53:17 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"videocodec,audiocodec_lo\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio_lo\"\n"
|
2016-03-25 08:39:07 +00:00
|
|
|
"http://anydomain.com/video.m3u8\n";
|
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
|
|
|
|
2018-01-17 23:43:41 +00:00
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistSameAudioGroupSameLanguage) {
|
|
|
|
// First video, video.m3u8.
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> video_playlist =
|
|
|
|
CreateVideoPlaylist("video.m3u8", "videocodec", 300000);
|
2018-01-17 23:43:41 +00:00
|
|
|
|
|
|
|
// First audio, eng_lo.m3u8.
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> eng_lo_playlist = CreateAudioPlaylist(
|
|
|
|
"eng_lo.m3u8", "english", "audio", "audiocodec", "en", 1, 50000);
|
2018-01-17 23:43:41 +00:00
|
|
|
|
2018-01-26 23:53:58 +00:00
|
|
|
std::unique_ptr<MockMediaPlaylist> eng_hi_playlist = CreateAudioPlaylist(
|
|
|
|
"eng_hi.m3u8", "english", "audio", "audiocodec", "en", 8, 100000);
|
2018-01-17 23:43:41 +00:00
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://anydomain.com/";
|
2018-02-02 17:14:46 +00:00
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(
|
|
|
|
kBaseUrl, test_output_dir_,
|
|
|
|
{video_playlist.get(), eng_lo_playlist.get(), eng_hi_playlist.get()}));
|
2018-01-17 23:43:41 +00:00
|
|
|
|
|
|
|
std::string actual;
|
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-17 23:43:41 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://anydomain.com/eng_lo.m3u8\","
|
|
|
|
"GROUP-ID=\"audio\",LANGUAGE=\"en\",NAME=\"english\","
|
|
|
|
"DEFAULT=YES,AUTOSELECT=YES,CHANNELS=\"1\"\n"
|
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://anydomain.com/eng_hi.m3u8\","
|
2018-02-02 17:52:31 +00:00
|
|
|
"GROUP-ID=\"audio\",LANGUAGE=\"en\",NAME=\"english\",CHANNELS=\"8\"\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-17 23:43:41 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=400000,CODECS=\"videocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio\"\n"
|
|
|
|
"http://anydomain.com/video.m3u8\n";
|
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistVideosAndTexts) {
|
|
|
|
// Video, sd.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> video1 =
|
|
|
|
CreateVideoPlaylist("sd.m3u8", "sdvideocodec", 300000);
|
|
|
|
|
|
|
|
// Video, hd.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> video2 =
|
|
|
|
CreateVideoPlaylist("hd.m3u8", "sdvideocodec", 600000);
|
|
|
|
|
|
|
|
// Text, eng.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> text_eng =
|
2018-02-02 22:45:52 +00:00
|
|
|
CreateTextPlaylist("eng.m3u8", "english", "textgroup", "en");
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
// Text, fr.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> text_fr =
|
2018-02-02 22:45:52 +00:00
|
|
|
CreateTextPlaylist("fr.m3u8", "french", "textgroup", "fr");
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://playlists.org/";
|
2018-02-02 17:14:46 +00:00
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(
|
|
|
|
kBaseUrl, test_output_dir_,
|
|
|
|
{video1.get(), video2.get(), text_eng.get(), text_fr.get()}));
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
std::string actual;
|
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=SUBTITLES,URI=\"http://playlists.org/eng.m3u8\","
|
2018-02-02 17:52:31 +00:00
|
|
|
"GROUP-ID=\"textgroup\",LANGUAGE=\"en\",NAME=\"english\",DEFAULT=YES,"
|
|
|
|
"AUTOSELECT=YES\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=SUBTITLES,URI=\"http://playlists.org/fr.m3u8\","
|
2018-02-02 17:52:31 +00:00
|
|
|
"GROUP-ID=\"textgroup\",LANGUAGE=\"fr\",NAME=\"french\",AUTOSELECT=YES\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=300000,CODECS=\"sdvideocodec\","
|
|
|
|
"RESOLUTION=800x600,SUBTITLES=\"textgroup\"\n"
|
|
|
|
"http://playlists.org/sd.m3u8\n"
|
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=600000,CODECS=\"sdvideocodec\","
|
|
|
|
"RESOLUTION=800x600,SUBTITLES=\"textgroup\"\n"
|
|
|
|
"http://playlists.org/hd.m3u8\n";
|
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistVideoAndTextGroups) {
|
|
|
|
// Video, sd.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> video =
|
|
|
|
CreateVideoPlaylist("sd.m3u8", "sdvideocodec", 300000);
|
|
|
|
|
|
|
|
// Text, eng.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> text_eng =
|
2018-02-02 22:45:52 +00:00
|
|
|
CreateTextPlaylist("eng.m3u8", "english", "en-text-group", "en");
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
// Text, fr.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> text_fr =
|
2018-02-02 22:45:52 +00:00
|
|
|
CreateTextPlaylist("fr.m3u8", "french", "fr-text-group", "fr");
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://playlists.org/";
|
2018-02-02 17:14:46 +00:00
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(
|
|
|
|
kBaseUrl, test_output_dir_,
|
|
|
|
{video.get(), text_eng.get(), text_fr.get()}));
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
std::string actual;
|
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=SUBTITLES,URI=\"http://playlists.org/eng.m3u8\","
|
2018-02-05 19:21:28 +00:00
|
|
|
"GROUP-ID=\"en-text-group\",LANGUAGE=\"en\",NAME=\"english\","
|
|
|
|
"DEFAULT=YES,AUTOSELECT=YES\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=SUBTITLES,URI=\"http://playlists.org/fr.m3u8\","
|
2018-02-05 19:21:28 +00:00
|
|
|
"GROUP-ID=\"fr-text-group\",LANGUAGE=\"fr\",NAME=\"french\","
|
|
|
|
"AUTOSELECT=YES\n"
|
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=300000,CODECS=\"sdvideocodec\","
|
|
|
|
"RESOLUTION=800x600,SUBTITLES=\"en-text-group\"\n"
|
|
|
|
"http://playlists.org/sd.m3u8\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=300000,CODECS=\"sdvideocodec\","
|
|
|
|
"RESOLUTION=800x600,SUBTITLES=\"fr-text-group\"\n"
|
|
|
|
"http://playlists.org/sd.m3u8\n";
|
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistVideoAndAudioAndText) {
|
|
|
|
// Video, sd.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> video =
|
|
|
|
CreateVideoPlaylist("sd.m3u8", "sdvideocodec", 300000);
|
|
|
|
|
|
|
|
// Audio, english.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> audio = CreateAudioPlaylist(
|
|
|
|
"eng.m3u8", "english", "audiogroup", "audiocodec", "en", 2, 50000);
|
|
|
|
|
|
|
|
// Text, english.m3u8.
|
|
|
|
std::unique_ptr<MockMediaPlaylist> text =
|
2018-02-02 22:45:52 +00:00
|
|
|
CreateTextPlaylist("eng.m3u8", "english", "textgroup", "en");
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://playlists.org/";
|
2018-02-02 17:14:46 +00:00
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(
|
|
|
|
kBaseUrl, test_output_dir_, {video.get(), audio.get(), text.get()}));
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
std::string actual;
|
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://playlists.org/eng.m3u8\","
|
|
|
|
"GROUP-ID=\"audiogroup\",LANGUAGE=\"en\",NAME=\"english\","
|
|
|
|
"DEFAULT=YES,AUTOSELECT=YES,CHANNELS=\"2\"\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=SUBTITLES,URI=\"http://playlists.org/eng.m3u8\","
|
2018-02-02 17:52:31 +00:00
|
|
|
"GROUP-ID=\"textgroup\",LANGUAGE=\"en\",NAME=\"english\",DEFAULT=YES,"
|
|
|
|
"AUTOSELECT=YES\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audiogroup\",SUBTITLES=\"textgroup\"\n"
|
|
|
|
"http://playlists.org/sd.m3u8\n";
|
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
|
|
|
|
2018-02-02 22:45:52 +00:00
|
|
|
TEST_F(MasterPlaylistTest, WriteMasterPlaylistMixedPlaylistsDifferentGroups) {
|
2018-01-31 19:14:15 +00:00
|
|
|
const uint64_t kAudioChannels = 2;
|
|
|
|
const uint64_t kAudioBitRate = 50000;
|
|
|
|
const uint64_t kVideoBitRate = 300000;
|
2018-02-02 22:45:52 +00:00
|
|
|
const uint64_t kIframeBitRate = 100000;
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
std::unique_ptr<MockMediaPlaylist> media_playlists[] = {
|
|
|
|
// AUDIO
|
|
|
|
CreateAudioPlaylist("audio-1.m3u8", "audio 1", "audio-group-1",
|
|
|
|
"audiocodec", "en", kAudioChannels, kAudioBitRate),
|
|
|
|
CreateAudioPlaylist("audio-2.m3u8", "audio 2", "audio-group-2",
|
|
|
|
"audiocodec", "en", kAudioChannels, kAudioBitRate),
|
|
|
|
|
|
|
|
// SUBTITLES
|
2018-02-02 22:45:52 +00:00
|
|
|
CreateTextPlaylist("text-1.m3u8", "text 1", "text-group-1", "en"),
|
|
|
|
CreateTextPlaylist("text-2.m3u8", "text 2", "text-group-2", "en"),
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
// VIDEO
|
|
|
|
CreateVideoPlaylist("video-1.m3u8", "sdvideocodec", kVideoBitRate),
|
|
|
|
CreateVideoPlaylist("video-2.m3u8", "sdvideocodec", kVideoBitRate),
|
2018-02-02 22:45:52 +00:00
|
|
|
|
|
|
|
// I-Frame
|
|
|
|
CreateIframePlaylist("iframe-1.m3u8", "sdvideocodec", kIframeBitRate),
|
|
|
|
CreateIframePlaylist("iframe-2.m3u8", "sdvideocodec", kIframeBitRate),
|
2018-01-31 19:14:15 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
// Add all the media playlists to the master playlist.
|
2018-02-02 17:14:46 +00:00
|
|
|
std::list<MediaPlaylist*> media_playlist_list;
|
2018-01-31 19:14:15 +00:00
|
|
|
for (const auto& media_playlist : media_playlists) {
|
2018-02-02 17:14:46 +00:00
|
|
|
media_playlist_list.push_back(media_playlist.get());
|
2018-01-31 19:14:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
const char kBaseUrl[] = "http://playlists.org/";
|
2018-02-02 17:14:46 +00:00
|
|
|
EXPECT_TRUE(master_playlist_.WriteMasterPlaylist(kBaseUrl, test_output_dir_,
|
|
|
|
media_playlist_list));
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
std::string actual;
|
|
|
|
ASSERT_TRUE(File::ReadFileToString(master_playlist_path_.c_str(), &actual));
|
|
|
|
|
|
|
|
const std::string expected =
|
|
|
|
"#EXTM3U\n"
|
|
|
|
"## Generated with https://github.com/google/shaka-packager version "
|
|
|
|
"test\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://playlists.org/audio-1.m3u8\","
|
|
|
|
"GROUP-ID=\"audio-group-1\",LANGUAGE=\"en\",NAME=\"audio 1\","
|
|
|
|
"DEFAULT=YES,AUTOSELECT=YES,CHANNELS=\"2\"\n"
|
|
|
|
"#EXT-X-MEDIA:TYPE=AUDIO,URI=\"http://playlists.org/audio-2.m3u8\","
|
|
|
|
"GROUP-ID=\"audio-group-2\",LANGUAGE=\"en\",NAME=\"audio 2\","
|
|
|
|
"DEFAULT=YES,AUTOSELECT=YES,CHANNELS=\"2\"\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=SUBTITLES,URI=\"http://playlists.org/text-1.m3u8\","
|
2018-02-05 19:21:28 +00:00
|
|
|
"GROUP-ID=\"text-group-1\",LANGUAGE=\"en\",NAME=\"text 1\","
|
|
|
|
"DEFAULT=YES,AUTOSELECT=YES\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-MEDIA:TYPE=SUBTITLES,URI=\"http://playlists.org/text-2.m3u8\","
|
2018-02-05 19:21:28 +00:00
|
|
|
"GROUP-ID=\"text-group-2\",LANGUAGE=\"en\",NAME=\"text 2\","
|
|
|
|
"DEFAULT=YES,AUTOSELECT=YES\n"
|
|
|
|
"\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio-group-1\",SUBTITLES=\"text-group-1\"\n"
|
|
|
|
"http://playlists.org/video-1.m3u8\n"
|
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio-group-1\",SUBTITLES=\"text-group-1\"\n"
|
|
|
|
"http://playlists.org/video-2.m3u8\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio-group-1\",SUBTITLES=\"text-group-2\"\n"
|
|
|
|
"http://playlists.org/video-1.m3u8\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio-group-1\",SUBTITLES=\"text-group-2\"\n"
|
|
|
|
"http://playlists.org/video-2.m3u8\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio-group-2\",SUBTITLES=\"text-group-1\"\n"
|
|
|
|
"http://playlists.org/video-1.m3u8\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio-group-2\",SUBTITLES=\"text-group-1\"\n"
|
|
|
|
"http://playlists.org/video-2.m3u8\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio-group-2\",SUBTITLES=\"text-group-2\"\n"
|
|
|
|
"http://playlists.org/video-1.m3u8\n"
|
2018-01-31 19:14:15 +00:00
|
|
|
"#EXT-X-STREAM-INF:BANDWIDTH=350000,CODECS=\"sdvideocodec,audiocodec\","
|
|
|
|
"RESOLUTION=800x600,AUDIO=\"audio-group-2\",SUBTITLES=\"text-group-2\"\n"
|
2018-02-02 22:45:52 +00:00
|
|
|
"http://playlists.org/video-2.m3u8\n"
|
2018-02-05 19:21:28 +00:00
|
|
|
"\n"
|
2018-02-02 22:45:52 +00:00
|
|
|
"#EXT-X-I-FRAME-STREAM-INF:BANDWIDTH=100000,CODECS=\"sdvideocodec\","
|
|
|
|
"RESOLUTION=800x600,URI=\"http://playlists.org/iframe-1.m3u8\"\n"
|
|
|
|
"#EXT-X-I-FRAME-STREAM-INF:BANDWIDTH=100000,CODECS=\"sdvideocodec\","
|
|
|
|
"RESOLUTION=800x600,URI=\"http://playlists.org/iframe-2.m3u8\"\n";
|
2018-01-31 19:14:15 +00:00
|
|
|
|
|
|
|
ASSERT_EQ(expected, actual);
|
|
|
|
}
|
2016-03-25 08:39:07 +00:00
|
|
|
} // namespace hls
|
2016-05-20 21:19:33 +00:00
|
|
|
} // namespace shaka
|