def test_static(self, sleep):
reader = MagicMock()
worker = DASHStreamWorker(reader)
reader.representation_id = 1
representation = Mock(id=1, mimeType="video/mp4", height=720)
segments = [Mock(url="init_segment"), Mock(url="first_segment"), Mock(url="second_segment")]
representation.segments.return_value = [segments[0]]
worker.mpd = Mock(dynamic=False,
publishTime=1,
periods=[
Mock(adaptationSets=[
Mock(contentProtection=None,
representations=[
representation
])
])
])
worker.mpd.type = "static"
worker.mpd.minimumUpdatePeriod.total_seconds.return_value = 0
worker.mpd.periods[0].duration.total_seconds.return_value = 0
representation.segments.return_value = segments
self.assertSequenceEqual(list(worker.iter_segments()), segments)
representation.segments.assert_called_with(init=True)
def test_dynamic_reload(self, mpdClass, sleep):
reader = MagicMock()
worker = DASHStreamWorker(reader)
reader.representation_id = 1
representation = Mock(id=1, mimeType="video/mp4", height=720)
segments = [Mock(url="init_segment"), Mock(url="first_segment"), Mock(url="second_segment")]
representation.segments.return_value = [segments[0]]
mpdClass.return_value = worker.mpd = Mock(dynamic=True,
publishTime=1,
periods=[
Mock(adaptationSets=[
Mock(contentProtection=None,
representations=[
representation
])
])
])
worker.mpd.type = "dynamic"
worker.mpd.minimumUpdatePeriod.total_seconds.return_value = 0
worker.mpd.periods[0].duration.total_seconds.return_value = 0
segment_iter = worker.iter_segments()
representation.segments.return_value = segments[:1]
self.assertEqual(next(segment_iter), segments[0])
representation.segments.assert_called_with(init=True)
representation.segments.return_value = segments[1:]
self.assertSequenceEqual([next(segment_iter), next(segment_iter)], segments[1:])
representation.segments.assert_called_with(init=False)
def test_static_refresh_wait(self, sleep, time):
"""
Verify the fix for https://github.com/streamlink/streamlink/issues/2873
"""
time.return_value = 1
reader = MagicMock()
worker = DASHStreamWorker(reader)
reader.representation_id = 1
reader.mime_type = "video/mp4"
representation = Mock(id=1, mimeType="video/mp4", height=720)
segments = [
Mock(url="init_segment"),
Mock(url="first_segment"),
Mock(url="second_segment")
]
representation.segments.return_value = [segments[0]]
worker.mpd = Mock(dynamic=False,
publishTime=1,
periods=[
Mock(adaptationSets=[
Mock(contentProtection=None,
representations=[representation])
])
])
worker.mpd.type = "static"
for duration in (0, 204.32):
worker.mpd.minimumUpdatePeriod.total_seconds.return_value = 0
worker.mpd.periods[
0].duration.total_seconds.return_value = duration
representation.segments.return_value = segments
self.assertSequenceEqual(list(worker.iter_segments()), segments)
representation.segments.assert_called_with(init=True)
sleep.assert_called_with(5)
def test_static(self, sleep):
reader = MagicMock()
worker = DASHStreamWorker(reader)
reader.representation_id = 1
reader.mime_type = "video/mp4"
representation = Mock(id=1, mimeType="video/mp4", height=720)
segments = [Mock(url="init_segment"), Mock(url="first_segment"), Mock(url="second_segment")]
representation.segments.return_value = [segments[0]]
worker.mpd = Mock(dynamic=False,
publishTime=1,
periods=[
Mock(adaptationSets=[
Mock(contentProtection=None,
representations=[
representation
])
])
])
worker.mpd.type = "static"
worker.mpd.minimumUpdatePeriod.total_seconds.return_value = 0
worker.mpd.periods[0].duration.total_seconds.return_value = 0
representation.segments.return_value = segments
self.assertSequenceEqual(list(worker.iter_segments()), segments)
representation.segments.assert_called_with(init=True)
def test_duplicate_rep_id(self, sleep):
representation_vid = Mock(id=1, mimeType="video/mp4", height=720)
representation_aud = Mock(id=1, mimeType="audio/aac", lang='en')
mpd = Mock(dynamic=False,
publishTime=1,
periods=[
Mock(adaptationSets=[
Mock(contentProtection=None,
representations=[representation_vid]),
Mock(contentProtection=None,
representations=[representation_aud])
])
])
self.assertEqual(
representation_vid,
DASHStreamWorker.get_representation(mpd, 1, "video/mp4"))
self.assertEqual(
representation_aud,
DASHStreamWorker.get_representation(mpd, 1, "audio/aac"))
def test_duplicate_rep_id(self, sleep):
representation_vid = Mock(id=1, mimeType="video/mp4", height=720)
representation_aud = Mock(id=1, mimeType="audio/aac", lang='en')
mpd = Mock(dynamic=False,
publishTime=1,
periods=[
Mock(adaptationSets=[
Mock(contentProtection=None,
representations=[
representation_vid
]),
Mock(contentProtection=None,
representations=[
representation_aud
])
])
])
self.assertEqual(representation_vid, DASHStreamWorker.get_representation(mpd, 1, "video/mp4"))
self.assertEqual(representation_aud, DASHStreamWorker.get_representation(mpd, 1, "audio/aac"))