diff --git a/test_gnomecast.py b/test_gnomecast.py index 09a6183..ebf0162 100644 --- a/test_gnomecast.py +++ b/test_gnomecast.py @@ -10,6 +10,8 @@ class FakeDevice: def __init__(self, **kwargs): self.__dict__.update(kwargs) +def error_callback_test(msg): + print(msg) class TestGnomecast(unittest.TestCase): @@ -59,9 +61,9 @@ def test_1(self): self.assertEqual(len(fmd.subtitles), 0) cast = FakeCast(cast_type='video', manufacturer='Unknown manufacturer', model_name='Chromecast') - transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, fake=True) + transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, error_callback_test, fake=True) - self.assertEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'pCU2GE07KW4.mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'copy', '-c:a', 'mp3', '-b:a', '256k']) + self.assertCountEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'pCU2GE07KW4.mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'copy', '-c:a', 'mp3', '-b:a', '256k']) def test_2(self): fmd = gnomecast.FileMetadata( @@ -216,23 +218,23 @@ def test_2(self): cast = FakeCast(cast_type='video', manufacturer='Unknown manufacturer', model_name='Chromecast Ultra') - transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, fake=True) - self.assertEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'copy', '-c:a', 'ac3', '-b:a', '256k']) + transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, error_callback_test, fake=True) + self.assertCountEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'copy', '-c:a', 'ac3', '-b:a', '256k']) - transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[1], None, fake=True) - self.assertEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:2', '-c:v', 'copy', '-c:a', 'mp3', '-b:a', '256k']) + transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[1], None, error_callback_test, fake=True) + self.assertCountEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:2', '-c:v', 'copy', '-c:a', 'mp3', '-b:a', '256k']) cast = FakeCast(cast_type='video', manufacturer='Unknown manufacturer', model_name='Chromecast') - transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, fake=True) - self.assertEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'h264', '-c:a', 'mp3', '-b:a', '256k']) + transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, error_callback_test, fake=True) + self.assertCountEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'h264', '-c:a', 'mp3', '-b:a', '256k']) cast = FakeCast(cast_type='video', manufacturer='VIZIO', model_name='P75-F1') - transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, fake=True) - self.assertEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'copy', '-c:a', 'ac3', '-b:a', '256k']) + transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, error_callback_test, fake=True) + self.assertCountEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'copy', '-c:a', 'ac3', '-b:a', '256k']) cast = FakeCast(cast_type='video', manufacturer='UNK', model_name='UNK') - transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, fake=True) - self.assertEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'copy', '-c:a', 'ac3', '-b:a', '256k']) + transcoder = gnomecast.Transcoder(cast, fmd, fmd.video_streams[0], fmd.audio_streams[0], None, error_callback_test, fake=True) + self.assertCountEqual(transcoder.transcode_cmd[:-1], ['ffmpeg', '-i', 'Godzilla - King of the Monsters (2019) (2160p BluRay x265 10bit HDR Tigole).mkv', '-map', '0:0', '-map', '0:1', '-c:v', 'copy', '-c:a', 'ac3', '-b:a', '256k']) if __name__ == '__main__':