From 96ced40ec2cf6f7541de4f0609a95cd1929e5d95 Mon Sep 17 00:00:00 2001 From: KongQun Yang Date: Fri, 16 Jun 2017 18:36:16 -0700 Subject: [PATCH] Move command parsing tests into separate class Change-Id: I6aeba632d69b1867c7d7745f0c8c9fd8f8e275d9 --- packager/app/test/packager_app.py | 5 +- packager/app/test/packager_test.py | 1566 ++++++++++++++-------------- 2 files changed, 770 insertions(+), 801 deletions(-) diff --git a/packager/app/test/packager_app.py b/packager/app/test/packager_app.py index 1a0b3706ed..8218b7420b 100644 --- a/packager/app/test/packager_app.py +++ b/packager/app/test/packager_app.py @@ -52,9 +52,10 @@ class PackagerApp(object): # Put single-quotes around each entry so that things like '$' signs in # segment templates won't be interpreted as shell variables. self.packaging_command_line = ' '.join(["'%s'" % entry for entry in cmd]) - self.packaging_result = subprocess.call(cmd) - if self.packaging_result != 0: + packaging_result = subprocess.call(cmd) + if packaging_result != 0: print '%s returned non-0 status' % self.packaging_command_line + return packaging_result def GetCommandLine(self): return self.packaging_command_line diff --git a/packager/app/test/packager_test.py b/packager/app/test/packager_test.py index 70b774327e..1844044f3a 100755 --- a/packager/app/test/packager_test.py +++ b/packager/app/test/packager_test.py @@ -58,803 +58,6 @@ class PackagerAppTest(unittest.TestCase): if test_env.options.remove_temp_files_after_test: shutil.rmtree(self.tmp_dir) - def testVersion(self): - self.assertRegexpMatches( - self.packager.Version(), '^packager(.exe)? version ' - r'((?P[\w\.]+)-)?(?P[a-f\d]+)-(debug|release)[\r\n]+.*$') - - def testDumpStreamInfo(self): - test_file = os.path.join(self.test_data_dir, 'bear-640x360.mp4') - stream_info = self.packager.DumpStreamInfo(test_file) - expected_stream_info = ('Found 2 stream(s).\n' - 'Stream [0] type: Video\n' - ' codec_string: avc1.64001e\n' - ' time_scale: 30000\n' - ' duration: 82082 (2.7 seconds)\n' - ' is_encrypted: false\n' - ' codec: H264\n' - ' width: 640\n' - ' height: 360\n' - ' pixel_aspect_ratio: 1:1\n' - ' trick_play_factor: 0\n' - ' nalu_length_size: 4\n\n' - 'Stream [1] type: Audio\n' - ' codec_string: mp4a.40.2\n' - ' time_scale: 44100\n' - ' duration: 121856 (2.8 seconds)\n' - ' is_encrypted: false\n' - ' codec: AAC\n' - ' sample_bits: 16\n' - ' num_channels: 2\n' - ' sampling_frequency: 44100\n' - ' language: und\n') - stream_info = stream_info.replace('\r\n', '\n') - self.assertIn(expected_stream_info, stream_info, - '\nExpecting: \n %s\n\nBut seeing: \n%s' % - (expected_stream_info, stream_info)) - - def testPackageFirstStream(self): - self.packager.Package(self._GetStreams(['0']), self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-v-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-v-golden.mpd') - - def testPackageText(self): - self.packager.Package( - self._GetStreams(['text'], test_files=['subtitle-english.vtt']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'subtitle-english-golden.vtt') - self._DiffGold(self.mpd_output, 'subtitle-english-vtt-golden.mpd') - - # Probably one of the most common scenarios is to package audio and video. - def testPackageAudioVideo(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-golden.mpd') - - def testPackageAudioVideoWithTrickPlay(self): - self.packager.Package( - self._GetStreams(['audio', 'video', 'video,trick_play_factor=1']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') - self._DiffGold(self.output[2], 'bear-640x360-v-trick-1-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-trick-1-golden.mpd') - - def testPackageAudioVideoWithTwoTrickPlay(self): - self.packager.Package( - self._GetStreams(['audio', 'video', 'video,trick_play_factor=1', - 'video,trick_play_factor=2']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') - self._DiffGold(self.output[2], 'bear-640x360-v-trick-1-golden.mp4') - self._DiffGold(self.output[3], 'bear-640x360-v-trick-2-golden.mp4') - self._DiffGold(self.mpd_output, - 'bear-640x360-av-trick-1-trick-2-golden.mpd') - - def testPackageAudioVideoWithTwoTrickPlayDecreasingRate(self): - self.packager.Package( - self._GetStreams(['audio', 'video', 'video,trick_play_factor=2', - 'video,trick_play_factor=1']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') - self._DiffGold(self.output[2], 'bear-640x360-v-trick-2-golden.mp4') - self._DiffGold(self.output[3], 'bear-640x360-v-trick-1-golden.mp4') - # Since the stream descriptors are sorted in packager app, a different - # order of trick play factors gets the same mpd. - self._DiffGold(self.mpd_output, - 'bear-640x360-av-trick-1-trick-2-golden.mpd') - - def testPackageAudioVideoWithLanguageOverride(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], language_override='por-BR'), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-por-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-por-golden.mpd') - - def testPackageAudioVideoWithLanguageOverrideWithSubtag(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], language_override='por-BR'), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-por-BR-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-por-BR-golden.mpd') - - def testPackageAacHe(self): - self.packager.Package( - self._GetStreams( - ['audio'], test_files=['bear-640x360-aac_he-silent_right.mp4']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], - 'bear-640x360-aac_he-silent_right-golden.mp4') - self._DiffGold(self.mpd_output, - 'bear-640x360-aac_he-silent_right-golden.mpd') - - # Package all video, audio, and text. - def testPackageVideoAudioText(self): - audio_video_streams = self._GetStreams(['audio', 'video']) - text_stream = self._GetStreams(['text'], - test_files=['subtitle-english.vtt']) - self.packager.Package(audio_video_streams + text_stream, self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') - self._DiffGold(self.output[2], 'subtitle-english-golden.vtt') - self._DiffGold(self.mpd_output, 'bear-640x360-avt-golden.mpd') - - def testPackageAvcTs(self): - # Currently we only support live packaging for ts. - self.packager.Package( - self._GetStreams(['audio', 'video'], - output_format='ts', - live=True, - test_files=['bear-640x360.ts']), - self._GetFlags(output_hls=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], - 'bear-640x360-a-golden', - output_format='ts') - self._DiffLiveGold(self.output[1], - 'bear-640x360-v-golden', - output_format='ts') - self._DiffGold(self.hls_master_playlist_output, - 'bear-640x360-av-master-golden.m3u8') - self._DiffGold( - os.path.join(self.tmp_dir, 'audio.m3u8'), 'bear-640x360-a-golden.m3u8') - self._DiffGold( - os.path.join(self.tmp_dir, 'video.m3u8'), 'bear-640x360-v-golden.m3u8') - - def testPackageVp8Webm(self): - self.packager.Package( - self._GetStreams(['video'], - output_format='webm', - test_files=['bear-640x360.webm']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-vp8-golden.webm') - self._DiffGold(self.mpd_output, 'bear-640x360-vp8-webm-golden.mpd') - - def testPackageVp9Webm(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], - output_format='webm', - test_files=['bear-320x240-vp9-opus.webm']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-320x240-opus-golden.webm') - self._DiffGold(self.output[1], 'bear-320x240-vp9-golden.webm') - self._DiffGold(self.mpd_output, 'bear-320x240-vp9-opus-webm-golden.mpd') - - def testPackageVp9WebmWithBlockgroup(self): - self.packager.Package( - self._GetStreams(['video'], - output_format='webm', - test_files=['bear-vp9-blockgroup.webm']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-vp9-blockgroup-golden.webm') - - def testPackageVorbisWebm(self): - self.packager.Package( - self._GetStreams(['audio'], - output_format='webm', - test_files=['bear-320x240-audio-only.webm']), - self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-320x240-vorbis-golden.webm') - self._DiffGold(self.mpd_output, 'bear-320x240-vorbis-webm-golden.mpd') - - def testPackageWithEncryption(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithEncryptionWithIncorrectKeyIdLength1(self): - self.encryption_key_id = self.encryption_key_id[0:-2] - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithIncorrectKeyIdLength2(self): - self.encryption_key_id += '12' - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithInvalidKeyIdValue(self): - self.encryption_key_id = self.encryption_key_id[0:-1] + 'g' - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithIncorrectKeyLength1(self): - self.encryption_key = self.encryption_key[0:-2] - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithIncorrectKeyLength2(self): - self.encryption_key += '12' - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithInvalidKeyValue(self): - self.encryption_key = self.encryption_key[0:-1] + 'g' - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithIncorrectIvLength1(self): - self.encryption_iv = self.encryption_iv[0:-2] - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithIncorrectIvLength2(self): - self.encryption_iv += '12' - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithInvalidIvValue(self): - self.encryption_iv = self.encryption_iv[0:-1] + 'g' - self.packager.Package( - self._GetStreams(['video']), self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithInvalidPsshValue1(self): - self.packager.Package( - self._GetStreams(['video']), - self._GetFlags(encryption=True) + ['--pssh=ag']) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionWithInvalidPsshValue2(self): - self.packager.Package( - self._GetStreams(['video']), - self._GetFlags(encryption=True) + ['--pssh=1122']) - self.assertEqual(self.packager.packaging_result, 1) - - def testPackageWithEncryptionOfOnlyVideoStream(self): - self.packager.Package( - self._GetStreams(['audio,skip_encryption=1', 'video']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-a-clear-v-cenc-golden.mpd') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithEncryptionAndTrickPlay(self): - self.packager.Package( - self._GetStreams(['audio', 'video', 'video,trick_play_factor=1']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') - self._DiffGold(self.output[2], 'bear-640x360-v-trick-1-cenc-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-trick-1-cenc-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - self._VerifyDecryption(self.output[2], 'bear-640x360-v-trick-1-golden.mp4') - - # TODO(hmchen): Add a test case that SD and HD AdapatationSet share one trick - # play stream. - def testPackageWithEncryptionAndTwoTrickPlays(self): - self.packager.Package( - self._GetStreams(['audio', 'video', 'video,trick_play_factor=1', - 'video,trick_play_factor=2']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') - self._DiffGold(self.output[2], 'bear-640x360-v-trick-1-cenc-golden.mp4') - self._DiffGold(self.output[3], 'bear-640x360-v-trick-2-cenc-golden.mp4') - self._DiffGold(self.mpd_output, - 'bear-640x360-av-trick-1-trick-2-cenc-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - self._VerifyDecryption(self.output[2], 'bear-640x360-v-trick-1-golden.mp4') - self._VerifyDecryption(self.output[3], 'bear-640x360-v-trick-2-golden.mp4') - - def testPackageWithEncryptionAndNoClearLead(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, clear_lead=0)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], - 'bear-640x360-a-cenc-no-clear-lead-golden.mp4') - self._DiffGold(self.output[1], - 'bear-640x360-v-cenc-no-clear-lead-golden.mp4') - self._DiffGold(self.mpd_output, - 'bear-640x360-av-cenc-no-clear-lead-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithEncryptionAndNoPsshInStream(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, include_pssh_in_stream=False)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cenc-no-pssh-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cenc-no-pssh-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-no-pssh-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithEncryptionCbc1(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, - protection_scheme='cbc1')) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cbc1-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cbc1-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-cbc1-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithEncryptionCens(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, - protection_scheme='cens')) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cens-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cens-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-cens-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithEncryptionCbcs(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, - protection_scheme='cbcs')) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cbcs-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cbcs-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-cbcs-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithWebmSubsampleEncryption(self): - self.packager.Package( - self._GetStreams(['video'], - output_format='webm', - test_files=['bear-320x180-vp9-altref.webm']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-320x180-vp9-altref-enc-golden.webm') - self._VerifyDecryption(self.output[0], - 'bear-320x180-vp9-altref-dec-golden.webm') - - def testPackageWithWebmVp9FullSampleEncryption(self): - self.packager.Package( - self._GetStreams(['video'], - output_format='webm', - test_files=['bear-320x180-vp9-altref.webm']), - self._GetFlags(encryption=True, vp9_subsample_encryption=False)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], - 'bear-320x180-vp9-fullsample-enc-golden.webm') - self._VerifyDecryption(self.output[0], - 'bear-320x180-vp9-altref-dec-golden.webm') - - def testPackageAvcTsWithEncryption(self): - # Currently we only support live packaging for ts. - self.packager.Package( - self._GetStreams(['audio', 'video'], - output_format='ts', - live=True, - test_files=['bear-640x360.ts']), - self._GetFlags(encryption=True, output_hls=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], - 'bear-640x360-a-enc-golden', - output_format='ts') - self._DiffLiveGold(self.output[1], - 'bear-640x360-v-enc-golden', - output_format='ts') - self._DiffGold(self.hls_master_playlist_output, - 'bear-640x360-av-enc-master-golden.m3u8') - self._DiffGold( - os.path.join(self.tmp_dir, 'audio.m3u8'), - 'bear-640x360-a-enc-golden.m3u8') - self._DiffGold( - os.path.join(self.tmp_dir, 'video.m3u8'), - 'bear-640x360-v-enc-golden.m3u8') - - def testPackageAvcTsWithEncryptionExerciseEmulationPrevention(self): - self.encryption_key = 'ad7e9786def9159db6724be06dfcde7a' - # Currently we only support live packaging for ts. - self.packager.Package( - self._GetStreams( - ['video'], - output_format='ts', - live=True, - test_files=['sintel-1024x436.mp4']), - self._GetFlags( - encryption=True, - output_hls=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], - 'sintel-1024x436-v-enc-golden', - output_format='ts') - self._DiffGold(self.hls_master_playlist_output, - 'sintel-1024x436-v-enc-master-golden.m3u8') - self._DiffGold( - os.path.join(self.tmp_dir, 'video.m3u8'), - 'sintel-1024x436-v-enc-golden.m3u8') - - def testPackageWebmWithEncryption(self): - self.packager.Package( - self._GetStreams(['video'], - output_format='webm', - test_files=['bear-640x360.webm']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-vp8-cenc-golden.webm') - self._DiffGold(self.mpd_output, 'bear-640x360-vp8-cenc-webm-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-vp8-golden.webm') - - def testPackageHevcWithEncryption(self): - self.packager.Package( - self._GetStreams(['video'], - test_files=['bear-640x360-hevc.mp4']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-hevc-cenc-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-hevc-cenc-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-hevc-golden.mp4') - - def testPackageVp8Mp4WithEncryption(self): - self.packager.Package( - self._GetStreams(['video'], - output_format='mp4', - test_files=['bear-640x360.webm']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-vp8-cenc-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-vp8-cenc-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-vp8-golden.mp4') - - def testPackageOpusVp9Mp4WithEncryption(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], - output_format='mp4', - test_files=['bear-320x240-vp9-opus.webm']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-320x240-opus-cenc-golden.mp4') - self._DiffGold(self.output[1], 'bear-320x240-vp9-cenc-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-320x240-opus-vp9-cenc-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-320x240-opus-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-320x240-vp9-golden.mp4') - - def testPackageWvmInput(self): - self.encryption_key = '9248d245390e0a49d483ba9b43fc69c3' - self.packager.Package( - self._GetStreams( - ['0', '1', '2', '3'], test_files=['bear-multi-configs.wvm']), - self._GetFlags(decryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - # Output timescale is 90000. - self._DiffGold(self.output[0], 'bear-320x180-v-wvm-golden.mp4') - self._DiffGold(self.output[1], 'bear-320x180-a-wvm-golden.mp4') - self._DiffGold(self.output[2], 'bear-640x360-v-wvm-golden.mp4') - self._DiffGold(self.output[3], 'bear-640x360-a-wvm-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-wvm-golden.mpd') - - # TODO(kqyang): Fix shared_library not supporting strip_parameter_set_nalus - # problem. - @unittest.skipUnless( - test_env.options.static_libpackager, - 'libpackager shared_library does not support ' - '--strip_parameter_set_nalus flag.' - ) - def testPackageWvmInputWithoutStrippingParameterSetNalus(self): - self.encryption_key = '9248d245390e0a49d483ba9b43fc69c3' - self.packager.Package( - self._GetStreams( - ['0', '1', '2', '3'], test_files=['bear-multi-configs.wvm']), - self._GetFlags(strip_parameter_set_nalus=False, decryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - # Output timescale is 90000. - self._DiffGold(self.output[0], 'bear-320x180-avc3-wvm-golden.mp4') - self._DiffGold(self.output[1], 'bear-320x180-a-wvm-golden.mp4') - self._DiffGold(self.output[2], 'bear-640x360-avc3-wvm-golden.mp4') - self._DiffGold(self.output[3], 'bear-640x360-a-wvm-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-avc3-wvm-golden.mpd') - - def testPackageWithEncryptionAndRandomIv(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, random_iv=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._AssertStreamInfo(self.output[0], 'is_encrypted: true') - self._AssertStreamInfo(self.output[1], 'is_encrypted: true') - # The outputs are encrypted with random iv, so they are not the same as - # golden files. - self.assertFalse(self._CompareWithGold(self.output[0], - 'bear-640x360-a-cenc-golden.mp4')) - self.assertFalse(self._CompareWithGold(self.output[1], - 'bear-640x360-v-cenc-golden.mp4')) - self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithEncryptionAndRealClock(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, use_fake_clock=False)) - self.assertEqual(self.packager.packaging_result, 0) - self._AssertStreamInfo(self.output[0], 'is_encrypted: true') - self._AssertStreamInfo(self.output[1], 'is_encrypted: true') - # The outputs are generated with real clock, so they are not the same as - # golden files. - self.assertFalse(self._CompareWithGold(self.output[0], - 'bear-640x360-a-cenc-golden.mp4')) - self.assertFalse(self._CompareWithGold(self.output[1], - 'bear-640x360-v-cenc-golden.mp4')) - self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') - self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') - self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') - - def testPackageWithEncryptionAndNonDashIfIop(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, dash_if_iop=False)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') - self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-non-iop-golden.mpd') - - def testPackageWithEncryptionAndOutputMediaInfo(self): - self.packager.Package( - self._GetStreams(['audio', 'video']), - self._GetFlags(encryption=True, output_media_info=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') - self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') - self._DiffMediaInfoGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') - self._DiffMediaInfoGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') - - def testPackageWithLiveProfile(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], live=True), self._GetFlags()) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-golden') - self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-golden') - self._DiffLiveMpdGold(self.mpd_output, 'bear-640x360-av-live-golden.mpd') - - def testPackageWithLiveStaticProfile(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], live=True), - self._GetFlags(generate_static_mpd=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-golden') - self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-golden') - self._DiffGold(self.mpd_output, 'bear-640x360-av-live-static-golden.mpd') - - def testPackageWithLiveProfileAndEncryption(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], live=True), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-cenc-golden') - self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-cenc-golden') - self._DiffLiveMpdGold(self.mpd_output, - 'bear-640x360-av-live-cenc-golden.mpd') - - def testPackageWithLiveProfileAndEncryptionAndNonDashIfIop(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], live=True), - self._GetFlags(encryption=True, dash_if_iop=False)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-cenc-golden') - self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-cenc-golden') - self._DiffLiveMpdGold(self.mpd_output, - 'bear-640x360-av-live-cenc-non-iop-golden.mpd') - - def testPackageWithLiveProfileAndEncryptionAndMultFiles(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], - live=True, - test_files=['bear-1280x720.mp4', 'bear-640x360.mp4', - 'bear-320x180.mp4']), - self._GetFlags(encryption=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[2], 'bear-640x360-a-live-cenc-golden') - self._DiffLiveGold(self.output[3], 'bear-640x360-v-live-cenc-golden') - # Mpd cannot be validated right now since we don't generate determinstic - # mpd with multiple inputs due to thread racing. - # TODO(kqyang): Generate determinstic mpd or at least validate mpd schema. - - def testPackageWithLiveProfileAndKeyRotation(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], live=True), - self._GetFlags(encryption=True, key_rotation=True)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], - 'bear-640x360-a-live-cenc-rotation-golden') - self._DiffLiveGold(self.output[1], - 'bear-640x360-v-live-cenc-rotation-golden') - self._DiffLiveMpdGold(self.mpd_output, - 'bear-640x360-av-live-cenc-rotation-golden.mpd') - - def testPackageWithLiveProfileAndKeyRotationAndNoPsshInStream(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], live=True), - self._GetFlags( - encryption=True, key_rotation=True, include_pssh_in_stream=False)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], - 'bear-640x360-a-live-cenc-rotation-no-pssh-golden') - self._DiffLiveGold(self.output[1], - 'bear-640x360-v-live-cenc-rotation-no-pssh-golden') - self._DiffLiveMpdGold( - self.mpd_output, - 'bear-640x360-av-live-cenc-rotation-no-pssh-golden.mpd') - - def testPackageWithLiveProfileAndKeyRotationAndNonDashIfIop(self): - self.packager.Package( - self._GetStreams(['audio', 'video'], live=True), - self._GetFlags(encryption=True, - key_rotation=True, - dash_if_iop=False)) - self.assertEqual(self.packager.packaging_result, 0) - self._DiffLiveGold(self.output[0], - 'bear-640x360-a-live-cenc-rotation-golden') - self._DiffLiveGold(self.output[1], - 'bear-640x360-v-live-cenc-rotation-golden') - self._DiffLiveMpdGold( - self.mpd_output, - 'bear-640x360-av-live-cenc-rotation-non-iop-golden.mpd') - - @unittest.skipUnless(test_env.has_aes_flags, 'Requires AES credentials.') - def testWidevineEncryptionWithAes(self): - flags = self._GetFlags(widevine_encryption=True) - flags += [ - '--signer=widevine_test', - '--aes_signing_key=' + test_env.options.aes_signing_key, - '--aes_signing_iv=' + test_env.options.aes_signing_iv - ] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 0) - self._AssertStreamInfo(self.output[0], 'is_encrypted: true') - self._AssertStreamInfo(self.output[1], 'is_encrypted: true') - - def testWidevineEncryptionInvalidContentId(self): - self.widevine_content_id += 'ag' - flags = self._GetFlags(widevine_encryption=True) - flags += [ - '--signer=widevine_test', '--aes_signing_key=1122', - '--aes_signing_iv=3344' - ] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - def testWidevineEncryptionInvalidAesSigningKey(self): - flags = self._GetFlags(widevine_encryption=True) - flags += [ - '--signer=widevine_test', '--aes_signing_key=11ag', - '--aes_signing_iv=3344' - ] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - def testWidevineEncryptionInvalidAesSigningIv(self): - flags = self._GetFlags(widevine_encryption=True) - flags += [ - '--signer=widevine_test', '--aes_signing_key=1122', - '--aes_signing_iv=33ag' - ] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - def testWidevineEncryptionMissingAesSigningKey(self): - flags = self._GetFlags(widevine_encryption=True) - flags += ['--signer=widevine_test', '--aes_signing_iv=3344'] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - def testWidevineEncryptionMissingAesSigningIv(self): - flags = self._GetFlags(widevine_encryption=True) - flags += ['--signer=widevine_test', '--aes_signing_key=1122'] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - def testWidevineEncryptionMissingSigner1(self): - flags = self._GetFlags(widevine_encryption=True) - flags += ['--aes_signing_key=1122', '--aes_signing_iv=3344'] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - def testWidevineEncryptionMissingSigner2(self): - flags = self._GetFlags(widevine_encryption=True) - flags += ['--rsa_signing_key_path=/tmp/test'] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - def testWidevineEncryptionSignerOnly(self): - flags = self._GetFlags(widevine_encryption=True) - flags += ['--signer=widevine_test'] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - def testWidevineEncryptionAesSigningAndRsaSigning(self): - flags = self._GetFlags(widevine_encryption=True) - flags += [ - '--signer=widevine_test', - '--aes_signing_key=1122', - '--aes_signing_iv=3344', - '--rsa_signing_key_path=/tmp/test', - ] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 1) - - @unittest.skipUnless(test_env.has_aes_flags, 'Requires AES credentials.') - def testWidevineEncryptionWithAesAndMultFiles(self): - flags = self._GetFlags(widevine_encryption=True) - flags += [ - '--signer=widevine_test', - '--aes_signing_key=' + test_env.options.aes_signing_key, - '--aes_signing_iv=' + test_env.options.aes_signing_iv - ] - self.packager.Package( - self._GetStreams(['audio', 'video'], - test_files=['bear-1280x720.mp4', 'bear-640x360.mp4', - 'bear-320x180.mp4']), flags) - self.assertEqual(self.packager.packaging_result, 0) - with open(self.mpd_output, 'rb') as f: - print f.read() - # TODO(kqyang): Add some validations. - - @unittest.skipUnless(test_env.has_aes_flags, 'Requires AES credentials.') - def testKeyRotationWithAes(self): - flags = self._GetFlags(widevine_encryption=True, key_rotation=True) - flags += [ - '--signer=widevine_test', - '--aes_signing_key=' + test_env.options.aes_signing_key, - '--aes_signing_iv=' + test_env.options.aes_signing_iv - ] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 0) - self._AssertStreamInfo(self.output[0], 'is_encrypted: true') - self._AssertStreamInfo(self.output[1], 'is_encrypted: true') - - @unittest.skipUnless(test_env.has_rsa_flags, 'Requires RSA credentials.') - def testWidevineEncryptionWithRsa(self): - flags = self._GetFlags(widevine_encryption=True) - flags += [ - '--signer=widevine_test', - '--rsa_signing_key_path=' + test_env.options.rsa_signing_key_path - ] - self.packager.Package(self._GetStreams(['audio', 'video']), flags) - self.assertEqual(self.packager.packaging_result, 0) - self._AssertStreamInfo(self.output[0], 'is_encrypted: true') - self._AssertStreamInfo(self.output[1], 'is_encrypted: true') - def _GetStreams(self, stream_descriptors, language_override=None, @@ -1069,6 +272,625 @@ class PackagerAppTest(unittest.TestCase): self._DiffGold(test_output, golden_file_name) + +class PackagerFunctionalTest(PackagerAppTest): + + def assertPackageSuccess(self, streams, flags=None): + self.assertEqual(self.packager.Package(streams, flags), 0) + + def testVersion(self): + self.assertRegexpMatches( + self.packager.Version(), '^packager(.exe)? version ' + r'((?P[\w\.]+)-)?(?P[a-f\d]+)-(debug|release)[\r\n]+.*$') + + def testDumpStreamInfo(self): + test_file = os.path.join(self.test_data_dir, 'bear-640x360.mp4') + stream_info = self.packager.DumpStreamInfo(test_file) + expected_stream_info = ('Found 2 stream(s).\n' + 'Stream [0] type: Video\n' + ' codec_string: avc1.64001e\n' + ' time_scale: 30000\n' + ' duration: 82082 (2.7 seconds)\n' + ' is_encrypted: false\n' + ' codec: H264\n' + ' width: 640\n' + ' height: 360\n' + ' pixel_aspect_ratio: 1:1\n' + ' trick_play_factor: 0\n' + ' nalu_length_size: 4\n\n' + 'Stream [1] type: Audio\n' + ' codec_string: mp4a.40.2\n' + ' time_scale: 44100\n' + ' duration: 121856 (2.8 seconds)\n' + ' is_encrypted: false\n' + ' codec: AAC\n' + ' sample_bits: 16\n' + ' num_channels: 2\n' + ' sampling_frequency: 44100\n' + ' language: und\n') + stream_info = stream_info.replace('\r\n', '\n') + self.assertIn(expected_stream_info, stream_info, + '\nExpecting: \n %s\n\nBut seeing: \n%s' % + (expected_stream_info, stream_info)) + + def testPackageFirstStream(self): + self.assertPackageSuccess(self._GetStreams(['0']), self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-v-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-v-golden.mpd') + + def testPackageText(self): + self.assertPackageSuccess( + self._GetStreams(['text'], test_files=['subtitle-english.vtt']), + self._GetFlags()) + self._DiffGold(self.output[0], 'subtitle-english-golden.vtt') + self._DiffGold(self.mpd_output, 'subtitle-english-vtt-golden.mpd') + + # Probably one of the most common scenarios is to package audio and video. + def testPackageAudioVideo(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-golden.mpd') + + def testPackageAudioVideoWithTrickPlay(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video', 'video,trick_play_factor=1']), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') + self._DiffGold(self.output[2], 'bear-640x360-v-trick-1-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-trick-1-golden.mpd') + + def testPackageAudioVideoWithTwoTrickPlay(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video', 'video,trick_play_factor=1', + 'video,trick_play_factor=2']), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') + self._DiffGold(self.output[2], 'bear-640x360-v-trick-1-golden.mp4') + self._DiffGold(self.output[3], 'bear-640x360-v-trick-2-golden.mp4') + self._DiffGold(self.mpd_output, + 'bear-640x360-av-trick-1-trick-2-golden.mpd') + + def testPackageAudioVideoWithTwoTrickPlayDecreasingRate(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video', 'video,trick_play_factor=2', + 'video,trick_play_factor=1']), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') + self._DiffGold(self.output[2], 'bear-640x360-v-trick-2-golden.mp4') + self._DiffGold(self.output[3], 'bear-640x360-v-trick-1-golden.mp4') + # Since the stream descriptors are sorted in packager app, a different + # order of trick play factors gets the same mpd. + self._DiffGold(self.mpd_output, + 'bear-640x360-av-trick-1-trick-2-golden.mpd') + + def testPackageAudioVideoWithLanguageOverride(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], language_override='por-BR'), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-a-por-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-por-golden.mpd') + + def testPackageAudioVideoWithLanguageOverrideWithSubtag(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], language_override='por-BR'), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-a-por-BR-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-por-BR-golden.mpd') + + def testPackageAacHe(self): + self.assertPackageSuccess( + self._GetStreams( + ['audio'], test_files=['bear-640x360-aac_he-silent_right.mp4']), + self._GetFlags()) + self._DiffGold(self.output[0], + 'bear-640x360-aac_he-silent_right-golden.mp4') + self._DiffGold(self.mpd_output, + 'bear-640x360-aac_he-silent_right-golden.mpd') + + # Package all video, audio, and text. + def testPackageVideoAudioText(self): + audio_video_streams = self._GetStreams(['audio', 'video']) + text_stream = self._GetStreams(['text'], + test_files=['subtitle-english.vtt']) + self.assertPackageSuccess(audio_video_streams + text_stream, + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-golden.mp4') + self._DiffGold(self.output[2], 'subtitle-english-golden.vtt') + self._DiffGold(self.mpd_output, 'bear-640x360-avt-golden.mpd') + + def testPackageAvcTs(self): + # Currently we only support live packaging for ts. + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], + output_format='ts', + live=True, + test_files=['bear-640x360.ts']), + self._GetFlags(output_hls=True)) + self._DiffLiveGold(self.output[0], + 'bear-640x360-a-golden', + output_format='ts') + self._DiffLiveGold(self.output[1], + 'bear-640x360-v-golden', + output_format='ts') + self._DiffGold(self.hls_master_playlist_output, + 'bear-640x360-av-master-golden.m3u8') + self._DiffGold( + os.path.join(self.tmp_dir, 'audio.m3u8'), 'bear-640x360-a-golden.m3u8') + self._DiffGold( + os.path.join(self.tmp_dir, 'video.m3u8'), 'bear-640x360-v-golden.m3u8') + + def testPackageVp8Webm(self): + self.assertPackageSuccess( + self._GetStreams(['video'], + output_format='webm', + test_files=['bear-640x360.webm']), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-640x360-vp8-golden.webm') + self._DiffGold(self.mpd_output, 'bear-640x360-vp8-webm-golden.mpd') + + def testPackageVp9Webm(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], + output_format='webm', + test_files=['bear-320x240-vp9-opus.webm']), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-320x240-opus-golden.webm') + self._DiffGold(self.output[1], 'bear-320x240-vp9-golden.webm') + self._DiffGold(self.mpd_output, 'bear-320x240-vp9-opus-webm-golden.mpd') + + def testPackageVp9WebmWithBlockgroup(self): + self.assertPackageSuccess( + self._GetStreams(['video'], + output_format='webm', + test_files=['bear-vp9-blockgroup.webm']), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-vp9-blockgroup-golden.webm') + + def testPackageVorbisWebm(self): + self.assertPackageSuccess( + self._GetStreams(['audio'], + output_format='webm', + test_files=['bear-320x240-audio-only.webm']), + self._GetFlags()) + self._DiffGold(self.output[0], 'bear-320x240-vorbis-golden.webm') + self._DiffGold(self.mpd_output, 'bear-320x240-vorbis-webm-golden.mpd') + + def testPackageWithEncryption(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithEncryptionOfOnlyVideoStream(self): + self.assertPackageSuccess( + self._GetStreams(['audio,skip_encryption=1', 'video']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-640x360-a-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-a-clear-v-cenc-golden.mpd') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithEncryptionAndTrickPlay(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video', 'video,trick_play_factor=1']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') + self._DiffGold(self.output[2], 'bear-640x360-v-trick-1-cenc-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-trick-1-cenc-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + self._VerifyDecryption(self.output[2], 'bear-640x360-v-trick-1-golden.mp4') + + # TODO(hmchen): Add a test case that SD and HD AdapatationSet share one trick + # play stream. + def testPackageWithEncryptionAndTwoTrickPlays(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video', 'video,trick_play_factor=1', + 'video,trick_play_factor=2']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') + self._DiffGold(self.output[2], 'bear-640x360-v-trick-1-cenc-golden.mp4') + self._DiffGold(self.output[3], 'bear-640x360-v-trick-2-cenc-golden.mp4') + self._DiffGold(self.mpd_output, + 'bear-640x360-av-trick-1-trick-2-cenc-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + self._VerifyDecryption(self.output[2], 'bear-640x360-v-trick-1-golden.mp4') + self._VerifyDecryption(self.output[3], 'bear-640x360-v-trick-2-golden.mp4') + + def testPackageWithEncryptionAndNoClearLead(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, clear_lead=0)) + self._DiffGold(self.output[0], + 'bear-640x360-a-cenc-no-clear-lead-golden.mp4') + self._DiffGold(self.output[1], + 'bear-640x360-v-cenc-no-clear-lead-golden.mp4') + self._DiffGold(self.mpd_output, + 'bear-640x360-av-cenc-no-clear-lead-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithEncryptionAndNoPsshInStream(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, include_pssh_in_stream=False)) + self._DiffGold(self.output[0], 'bear-640x360-a-cenc-no-pssh-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cenc-no-pssh-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-no-pssh-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithEncryptionCbc1(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, + protection_scheme='cbc1')) + self._DiffGold(self.output[0], 'bear-640x360-a-cbc1-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cbc1-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-cbc1-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithEncryptionCens(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, + protection_scheme='cens')) + self._DiffGold(self.output[0], 'bear-640x360-a-cens-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cens-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-cens-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithEncryptionCbcs(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, + protection_scheme='cbcs')) + self._DiffGold(self.output[0], 'bear-640x360-a-cbcs-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cbcs-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-cbcs-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithWebmSubsampleEncryption(self): + self.assertPackageSuccess( + self._GetStreams(['video'], + output_format='webm', + test_files=['bear-320x180-vp9-altref.webm']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-320x180-vp9-altref-enc-golden.webm') + self._VerifyDecryption(self.output[0], + 'bear-320x180-vp9-altref-dec-golden.webm') + + def testPackageWithWebmVp9FullSampleEncryption(self): + self.assertPackageSuccess( + self._GetStreams(['video'], + output_format='webm', + test_files=['bear-320x180-vp9-altref.webm']), + self._GetFlags(encryption=True, vp9_subsample_encryption=False)) + self._DiffGold(self.output[0], + 'bear-320x180-vp9-fullsample-enc-golden.webm') + self._VerifyDecryption(self.output[0], + 'bear-320x180-vp9-altref-dec-golden.webm') + + def testPackageAvcTsWithEncryption(self): + # Currently we only support live packaging for ts. + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], + output_format='ts', + live=True, + test_files=['bear-640x360.ts']), + self._GetFlags(encryption=True, output_hls=True)) + self._DiffLiveGold(self.output[0], + 'bear-640x360-a-enc-golden', + output_format='ts') + self._DiffLiveGold(self.output[1], + 'bear-640x360-v-enc-golden', + output_format='ts') + self._DiffGold(self.hls_master_playlist_output, + 'bear-640x360-av-enc-master-golden.m3u8') + self._DiffGold( + os.path.join(self.tmp_dir, 'audio.m3u8'), + 'bear-640x360-a-enc-golden.m3u8') + self._DiffGold( + os.path.join(self.tmp_dir, 'video.m3u8'), + 'bear-640x360-v-enc-golden.m3u8') + + def testPackageAvcTsWithEncryptionExerciseEmulationPrevention(self): + self.encryption_key = 'ad7e9786def9159db6724be06dfcde7a' + # Currently we only support live packaging for ts. + self.assertPackageSuccess( + self._GetStreams( + ['video'], + output_format='ts', + live=True, + test_files=['sintel-1024x436.mp4']), + self._GetFlags( + encryption=True, + output_hls=True)) + self._DiffLiveGold(self.output[0], + 'sintel-1024x436-v-enc-golden', + output_format='ts') + self._DiffGold(self.hls_master_playlist_output, + 'sintel-1024x436-v-enc-master-golden.m3u8') + self._DiffGold( + os.path.join(self.tmp_dir, 'video.m3u8'), + 'sintel-1024x436-v-enc-golden.m3u8') + + def testPackageWebmWithEncryption(self): + self.assertPackageSuccess( + self._GetStreams(['video'], + output_format='webm', + test_files=['bear-640x360.webm']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-640x360-vp8-cenc-golden.webm') + self._DiffGold(self.mpd_output, 'bear-640x360-vp8-cenc-webm-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-vp8-golden.webm') + + def testPackageHevcWithEncryption(self): + self.assertPackageSuccess( + self._GetStreams(['video'], + test_files=['bear-640x360-hevc.mp4']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-640x360-hevc-cenc-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-hevc-cenc-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-hevc-golden.mp4') + + def testPackageVp8Mp4WithEncryption(self): + self.assertPackageSuccess( + self._GetStreams(['video'], + output_format='mp4', + test_files=['bear-640x360.webm']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-640x360-vp8-cenc-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-vp8-cenc-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-vp8-golden.mp4') + + def testPackageOpusVp9Mp4WithEncryption(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], + output_format='mp4', + test_files=['bear-320x240-vp9-opus.webm']), + self._GetFlags(encryption=True)) + self._DiffGold(self.output[0], 'bear-320x240-opus-cenc-golden.mp4') + self._DiffGold(self.output[1], 'bear-320x240-vp9-cenc-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-320x240-opus-vp9-cenc-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-320x240-opus-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-320x240-vp9-golden.mp4') + + def testPackageWvmInput(self): + self.encryption_key = '9248d245390e0a49d483ba9b43fc69c3' + self.assertPackageSuccess( + self._GetStreams( + ['0', '1', '2', '3'], test_files=['bear-multi-configs.wvm']), + self._GetFlags(decryption=True)) + # Output timescale is 90000. + self._DiffGold(self.output[0], 'bear-320x180-v-wvm-golden.mp4') + self._DiffGold(self.output[1], 'bear-320x180-a-wvm-golden.mp4') + self._DiffGold(self.output[2], 'bear-640x360-v-wvm-golden.mp4') + self._DiffGold(self.output[3], 'bear-640x360-a-wvm-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-wvm-golden.mpd') + + # TODO(kqyang): Fix shared_library not supporting strip_parameter_set_nalus + # problem. + @unittest.skipUnless( + test_env.options.static_libpackager, + 'libpackager shared_library does not support ' + '--strip_parameter_set_nalus flag.' + ) + def testPackageWvmInputWithoutStrippingParameterSetNalus(self): + self.encryption_key = '9248d245390e0a49d483ba9b43fc69c3' + self.assertPackageSuccess( + self._GetStreams( + ['0', '1', '2', '3'], test_files=['bear-multi-configs.wvm']), + self._GetFlags(strip_parameter_set_nalus=False, decryption=True)) + # Output timescale is 90000. + self._DiffGold(self.output[0], 'bear-320x180-avc3-wvm-golden.mp4') + self._DiffGold(self.output[1], 'bear-320x180-a-wvm-golden.mp4') + self._DiffGold(self.output[2], 'bear-640x360-avc3-wvm-golden.mp4') + self._DiffGold(self.output[3], 'bear-640x360-a-wvm-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-avc3-wvm-golden.mpd') + + def testPackageWithEncryptionAndRandomIv(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, random_iv=True)) + self._AssertStreamInfo(self.output[0], 'is_encrypted: true') + self._AssertStreamInfo(self.output[1], 'is_encrypted: true') + # The outputs are encrypted with random iv, so they are not the same as + # golden files. + self.assertFalse(self._CompareWithGold(self.output[0], + 'bear-640x360-a-cenc-golden.mp4')) + self.assertFalse(self._CompareWithGold(self.output[1], + 'bear-640x360-v-cenc-golden.mp4')) + self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithEncryptionAndRealClock(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, use_fake_clock=False)) + self._AssertStreamInfo(self.output[0], 'is_encrypted: true') + self._AssertStreamInfo(self.output[1], 'is_encrypted: true') + # The outputs are generated with real clock, so they are not the same as + # golden files. + self.assertFalse(self._CompareWithGold(self.output[0], + 'bear-640x360-a-cenc-golden.mp4')) + self.assertFalse(self._CompareWithGold(self.output[1], + 'bear-640x360-v-cenc-golden.mp4')) + self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') + self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4') + self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4') + + def testPackageWithEncryptionAndNonDashIfIop(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, dash_if_iop=False)) + self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') + self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-non-iop-golden.mpd') + + def testPackageWithEncryptionAndOutputMediaInfo(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video']), + self._GetFlags(encryption=True, output_media_info=True)) + self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') + self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') + self._DiffMediaInfoGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') + self._DiffMediaInfoGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') + + def testPackageWithLiveProfile(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], live=True), self._GetFlags()) + self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-golden') + self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-golden') + self._DiffLiveMpdGold(self.mpd_output, 'bear-640x360-av-live-golden.mpd') + + def testPackageWithLiveStaticProfile(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], live=True), + self._GetFlags(generate_static_mpd=True)) + self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-golden') + self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-golden') + self._DiffGold(self.mpd_output, 'bear-640x360-av-live-static-golden.mpd') + + def testPackageWithLiveProfileAndEncryption(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], live=True), + self._GetFlags(encryption=True)) + self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-cenc-golden') + self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-cenc-golden') + self._DiffLiveMpdGold(self.mpd_output, + 'bear-640x360-av-live-cenc-golden.mpd') + + def testPackageWithLiveProfileAndEncryptionAndNonDashIfIop(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], live=True), + self._GetFlags(encryption=True, dash_if_iop=False)) + self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-cenc-golden') + self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-cenc-golden') + self._DiffLiveMpdGold(self.mpd_output, + 'bear-640x360-av-live-cenc-non-iop-golden.mpd') + + def testPackageWithLiveProfileAndEncryptionAndMultFiles(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], + live=True, + test_files=['bear-1280x720.mp4', 'bear-640x360.mp4', + 'bear-320x180.mp4']), + self._GetFlags(encryption=True)) + self._DiffLiveGold(self.output[2], 'bear-640x360-a-live-cenc-golden') + self._DiffLiveGold(self.output[3], 'bear-640x360-v-live-cenc-golden') + # Mpd cannot be validated right now since we don't generate determinstic + # mpd with multiple inputs due to thread racing. + # TODO(kqyang): Generate determinstic mpd or at least validate mpd schema. + + def testPackageWithLiveProfileAndKeyRotation(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], live=True), + self._GetFlags(encryption=True, key_rotation=True)) + self._DiffLiveGold(self.output[0], + 'bear-640x360-a-live-cenc-rotation-golden') + self._DiffLiveGold(self.output[1], + 'bear-640x360-v-live-cenc-rotation-golden') + self._DiffLiveMpdGold(self.mpd_output, + 'bear-640x360-av-live-cenc-rotation-golden.mpd') + + def testPackageWithLiveProfileAndKeyRotationAndNoPsshInStream(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], live=True), + self._GetFlags( + encryption=True, key_rotation=True, include_pssh_in_stream=False)) + self._DiffLiveGold(self.output[0], + 'bear-640x360-a-live-cenc-rotation-no-pssh-golden') + self._DiffLiveGold(self.output[1], + 'bear-640x360-v-live-cenc-rotation-no-pssh-golden') + self._DiffLiveMpdGold( + self.mpd_output, + 'bear-640x360-av-live-cenc-rotation-no-pssh-golden.mpd') + + def testPackageWithLiveProfileAndKeyRotationAndNonDashIfIop(self): + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], live=True), + self._GetFlags(encryption=True, + key_rotation=True, + dash_if_iop=False)) + self._DiffLiveGold(self.output[0], + 'bear-640x360-a-live-cenc-rotation-golden') + self._DiffLiveGold(self.output[1], + 'bear-640x360-v-live-cenc-rotation-golden') + self._DiffLiveMpdGold( + self.mpd_output, + 'bear-640x360-av-live-cenc-rotation-non-iop-golden.mpd') + + @unittest.skipUnless(test_env.has_aes_flags, 'Requires AES credentials.') + def testWidevineEncryptionWithAes(self): + flags = self._GetFlags(widevine_encryption=True) + flags += [ + '--signer=widevine_test', + '--aes_signing_key=' + test_env.options.aes_signing_key, + '--aes_signing_iv=' + test_env.options.aes_signing_iv + ] + self.assertPackageSuccess(self._GetStreams(['audio', 'video']), flags) + self._AssertStreamInfo(self.output[0], 'is_encrypted: true') + self._AssertStreamInfo(self.output[1], 'is_encrypted: true') + + @unittest.skipUnless(test_env.has_aes_flags, 'Requires AES credentials.') + def testWidevineEncryptionWithAesAndMultFiles(self): + flags = self._GetFlags(widevine_encryption=True) + flags += [ + '--signer=widevine_test', + '--aes_signing_key=' + test_env.options.aes_signing_key, + '--aes_signing_iv=' + test_env.options.aes_signing_iv + ] + self.assertPackageSuccess( + self._GetStreams(['audio', 'video'], + test_files=['bear-1280x720.mp4', 'bear-640x360.mp4', + 'bear-320x180.mp4']), flags) + with open(self.mpd_output, 'rb') as f: + print f.read() + # TODO(kqyang): Add some validations. + + @unittest.skipUnless(test_env.has_aes_flags, 'Requires AES credentials.') + def testKeyRotationWithAes(self): + flags = self._GetFlags(widevine_encryption=True, key_rotation=True) + flags += [ + '--signer=widevine_test', + '--aes_signing_key=' + test_env.options.aes_signing_key, + '--aes_signing_iv=' + test_env.options.aes_signing_iv + ] + self.assertPackageSuccess(self._GetStreams(['audio', 'video']), flags) + self._AssertStreamInfo(self.output[0], 'is_encrypted: true') + self._AssertStreamInfo(self.output[1], 'is_encrypted: true') + + @unittest.skipUnless(test_env.has_rsa_flags, 'Requires RSA credentials.') + def testWidevineEncryptionWithRsa(self): + flags = self._GetFlags(widevine_encryption=True) + flags += [ + '--signer=widevine_test', + '--rsa_signing_key_path=' + test_env.options.rsa_signing_key_path + ] + self.assertPackageSuccess(self._GetStreams(['audio', 'video']), flags) + self._AssertStreamInfo(self.output[0], 'is_encrypted: true') + self._AssertStreamInfo(self.output[1], 'is_encrypted: true') + def _AssertStreamInfo(self, stream, info): stream_info = self.packager.DumpStreamInfo(stream) self.assertIn('Found 1 stream(s).', stream_info) @@ -1076,14 +898,160 @@ class PackagerAppTest(unittest.TestCase): def _VerifyDecryption(self, test_encrypted_file, golden_clear_file): output_extension = os.path.splitext(golden_clear_file)[1][1:] - self.packager.Package( + self.assertPackageSuccess( self._GetStreams(['0'], output_format=output_extension, test_files=[test_encrypted_file]), self._GetFlags(decryption=True)) - self.assertEqual(self.packager.packaging_result, 0) self._DiffGold(self.output[-1], golden_clear_file) +class PackagerCommandParsingTest(PackagerAppTest): + + def testPackageWithEncryptionWithIncorrectKeyIdLength1(self): + self.encryption_key_id = self.encryption_key_id[0:-2] + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithIncorrectKeyIdLength2(self): + self.encryption_key_id += '12' + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithInvalidKeyIdValue(self): + self.encryption_key_id = self.encryption_key_id[0:-1] + 'g' + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithIncorrectKeyLength1(self): + self.encryption_key = self.encryption_key[0:-2] + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithIncorrectKeyLength2(self): + self.encryption_key += '12' + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithInvalidKeyValue(self): + self.encryption_key = self.encryption_key[0:-1] + 'g' + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithIncorrectIvLength1(self): + self.encryption_iv = self.encryption_iv[0:-2] + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithIncorrectIvLength2(self): + self.encryption_iv += '12' + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithInvalidIvValue(self): + self.encryption_iv = self.encryption_iv[0:-1] + 'g' + packaging_result = self.packager.Package( + self._GetStreams(['video']), self._GetFlags(encryption=True)) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithInvalidPsshValue1(self): + packaging_result = self.packager.Package( + self._GetStreams(['video']), + self._GetFlags(encryption=True) + ['--pssh=ag']) + self.assertEqual(packaging_result, 1) + + def testPackageWithEncryptionWithInvalidPsshValue2(self): + packaging_result = self.packager.Package( + self._GetStreams(['video']), + self._GetFlags(encryption=True) + ['--pssh=1122']) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionInvalidContentId(self): + self.widevine_content_id += 'ag' + flags = self._GetFlags(widevine_encryption=True) + flags += [ + '--signer=widevine_test', '--aes_signing_key=1122', + '--aes_signing_iv=3344' + ] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionInvalidAesSigningKey(self): + flags = self._GetFlags(widevine_encryption=True) + flags += [ + '--signer=widevine_test', '--aes_signing_key=11ag', + '--aes_signing_iv=3344' + ] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionInvalidAesSigningIv(self): + flags = self._GetFlags(widevine_encryption=True) + flags += [ + '--signer=widevine_test', '--aes_signing_key=1122', + '--aes_signing_iv=33ag' + ] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionMissingAesSigningKey(self): + flags = self._GetFlags(widevine_encryption=True) + flags += ['--signer=widevine_test', '--aes_signing_iv=3344'] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionMissingAesSigningIv(self): + flags = self._GetFlags(widevine_encryption=True) + flags += ['--signer=widevine_test', '--aes_signing_key=1122'] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionMissingSigner1(self): + flags = self._GetFlags(widevine_encryption=True) + flags += ['--aes_signing_key=1122', '--aes_signing_iv=3344'] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionMissingSigner2(self): + flags = self._GetFlags(widevine_encryption=True) + flags += ['--rsa_signing_key_path=/tmp/test'] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionSignerOnly(self): + flags = self._GetFlags(widevine_encryption=True) + flags += ['--signer=widevine_test'] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + def testWidevineEncryptionAesSigningAndRsaSigning(self): + flags = self._GetFlags(widevine_encryption=True) + flags += [ + '--signer=widevine_test', + '--aes_signing_key=1122', + '--aes_signing_iv=3344', + '--rsa_signing_key_path=/tmp/test', + ] + packaging_result = self.packager.Package( + self._GetStreams(['audio', 'video']), flags) + self.assertEqual(packaging_result, 1) + + if __name__ == '__main__': unittest.main()