1
0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-27 21:54:17 +01:00

tests: rewrite dash tests

This commit is contained in:
Johan Andersson 2021-05-23 23:31:21 +02:00
parent 1c3d8f114b
commit 0b35220e2f

View File

@ -1,6 +1,5 @@
import datetime
import os
import unittest
import pytest
from svtplay_dl.fetcher.dash import _dashparse
@ -12,49 +11,56 @@ from svtplay_dl.utils.parser import setup_defaults
def parse(playlist):
with open(os.path.join(os.path.dirname(os.path.realpath(__file__)), "dash-manifests", playlist)) as fd:
manifest = fd.read()
return _dashparse(setup_defaults(), manifest, "http://localhost", {}, None)
streams = {}
for i in list(_dashparse(setup_defaults(), manifest, "http://localhost", {}, None)):
streams[i.bitrate] = i
return streams
class dashtest(unittest.TestCase):
def test_parse_cmore(self):
data = yield from parse("cmore.mpd")
assert len(data[3261.367].files) == 410
assert len(data[3261.367].audio) == 615
assert data[3261.367].segments
def test_parse_cmore():
data = parse("cmore.mpd")
assert len(data[3261].files) == 410
assert len(data[3261].audio) == 615
assert data[3261].segments
def test_parse_fff(self):
data = yield from parse("fff.mpd")
assert len(data[3187.187].files) == 578
assert len(data[3187.187].audio) == 577
assert data[3187.187].segments
def test_parse_nya(self):
data = yield from parse("svtvod.mpd")
assert len(data[2793.0].files) == 350
assert len(data[2793.0].audio) == 350
assert data[2793.0].segments
def test_parse_fff():
data = parse("fff.mpd")
assert len(data[3187].files) == 578
assert len(data[3187].audio) == 577
assert data[3187].segments
def test_parse_live(self):
data = yield from parse("svtplay-live.mpd")
assert len(data[2795.9959999999996].files) == 6
assert len(data[2795.9959999999996].audio) == 6
assert data[2795.9959999999996].segments
def test_parse_live2(self):
data = yield from parse("svtplay-live2.mpd")
def test_parse_nya():
data = parse("svtvod.mpd")
assert len(data[2793].files) == 350
assert len(data[2793].audio) == 350
assert data[2793].segments
def test_parse_live():
data = parse("svtplay-live.mpd")
assert len(data[2795].files) == 6
assert len(data[2795].audio) == 6
assert data[2795].segments
def test_parse_live2():
data = parse("svtplay-live2.mpd")
assert len(data[2892.0].files) == 11
assert len(data[2892.0].audio) == 11
assert data[2892.0].segments
def test_parse_duration(self):
def test_parse_duration():
assert parse_duration("PT3459.520S") == 3459.52
assert parse_duration("PT2.00S") == 2.0
assert parse_duration("PT1H0M30.000S") == 3630.0
assert parse_duration("P1Y1M1DT1H0M30.000S") == 34218030.0
assert parse_duration("mMWroNG") == 0
def test_parse_date(self):
def test_parse_date():
assert isinstance(parse_dates("2021-05-10T06:00:11.451554796Z"), datetime.datetime)
assert isinstance(parse_dates("2021-05-10T06:00:11.45Z"), datetime.datetime)
assert isinstance(parse_dates("2021-05-10T06:00:11Z"), datetime.datetime)