def test_process_streams_success(self):
processor = FileProcessor('input.mkv', 'output.mkv', 'roku')
# Process empty streams list
res = processor.process_streams([])
self.assertEqual(res, [])
# Process weird streams, ignore
streams = [{'codec_type': 'pr0n', 'codec_name': 'h264', 'index': 0, 'refs': 2}]
res = processor.process_streams(streams)
self.assertEqual(res, [])
# Process 1 video only
streams = [{'codec_type': 'video', 'codec_name': 'h264', 'index': 0, 'refs': 1, 'height': 720}]
res = processor.process_streams(streams)
self.assertEqual(len(res), 1)
self.assertEqual(VideoProcessor.process.call_count, 1)
self.assertEqual(AudioProcessor.process.call_count, 0)
self.assertEqual(SubtitleProcessor.process.call_count, 0)
VideoProcessor.process.reset_mock()
# Process 1 video, 2 audio, 2 subs
streams = [{'codec_type': 'video', 'codec_name': 'h264', 'index': 0, 'refs': 4, 'height': 720},
{'codec_type': 'audio', 'codec_name': 'aac', 'index': 0, 'channels': 2},
{'codec_type': 'audio', 'codec_name': 'aac', 'index': 0, 'channels': 6},
{'codec_type': 'subtitle', 'codec_name': 'srt', 'index': 0},
{'codec_type': 'subtitle', 'codec_name': 'srt', 'index': 0}]
res = processor.process_streams(streams)
self.assertEqual(len(res), 5)
self.assertEqual(VideoProcessor.process.call_count, 1)
self.assertEqual(AudioProcessor.process.call_count, 2)
self.assertEqual(SubtitleProcessor.process.call_count, 2)
def test_process_streams_error(self):
processor = FileProcessor('input.mkv', 'output.mkv', 'roku')
# Process empty streams list
res = processor.process_streams([])
self.assertEqual(res, [])
# Process 1 video, 2 audio, 2 subs
streams = [{'codec_type': 'video', 'codec_name': 'h264', 'index': 0, 'refs': 4, 'height': 480},
{'codec_type': 'audio', 'codec_name': 'aac', 'index': 0, 'channels': 2},
{'codec_type': 'audio', 'codec_name': 'aac', 'index': 0, 'channels': 6},
{'codec_type': 'subtitle', 'codec_name': 'srt', 'index': 0},
{'codec_type': 'subtitle', 'codec_name': 'srt', 'index': 0}]
res = processor.process_streams(streams)
self.assertEqual(type(processor.error), ValueError)
