mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-28 06:04:17 +01:00
prio_streams: add unit tests
This commit is contained in:
parent
4740a42050
commit
e781a1e89b
59
lib/svtplay_dl/tests/prio_streams.py
Normal file
59
lib/svtplay_dl/tests/prio_streams.py
Normal file
@ -0,0 +1,59 @@
|
||||
#!/usr/bin/python
|
||||
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil; coding: utf-8 -*-
|
||||
# ex:ts=4:sw=4:sts=4:et:fenc=utf-8
|
||||
|
||||
from __future__ import absolute_import
|
||||
import unittest
|
||||
from svtplay_dl.utils import prio_streams
|
||||
|
||||
class Stream(object):
|
||||
def __init__(self, proto, bitrate):
|
||||
self.proto = proto
|
||||
self.bitrate = bitrate
|
||||
def name(self):
|
||||
return self.proto
|
||||
def __repr__(self):
|
||||
return '%s(%d)' % (self.proto.upper(), self.bitrate)
|
||||
|
||||
class PrioStreamsTest(unittest.TestCase):
|
||||
def _gen_proto_case(self, ordered, unordered, default=True, expected=None):
|
||||
streams = [Stream(x, 100) for x in unordered]
|
||||
|
||||
kwargs = {}
|
||||
if not default:
|
||||
kwargs['protocol_prio'] = ordered
|
||||
if expected is None:
|
||||
expected = [str(Stream(x, 100)) for x in ordered]
|
||||
|
||||
return self.assertEqual(
|
||||
[str(x) for x in prio_streams(streams, **kwargs)],
|
||||
expected
|
||||
)
|
||||
|
||||
def test_default_order(self):
|
||||
return self._gen_proto_case(
|
||||
['hls', 'hds', 'http', 'rtmp'],
|
||||
['rtmp', 'hds', 'hls', 'http']
|
||||
)
|
||||
|
||||
def test_custom_order(self):
|
||||
return self._gen_proto_case(
|
||||
['http', 'rtmp', 'hds', 'hls'],
|
||||
['rtmp', 'hds', 'hls', 'http'],
|
||||
default=False,
|
||||
)
|
||||
|
||||
def test_custom_order_1(self):
|
||||
return self._gen_proto_case(
|
||||
['http'],
|
||||
['rtmp', 'hds', 'hls', 'http'],
|
||||
default=False,
|
||||
)
|
||||
|
||||
def test_proto_unavail(self):
|
||||
return self._gen_proto_case(
|
||||
['http', 'rtmp'],
|
||||
['hds', 'hls', 'https'],
|
||||
default=False,
|
||||
expected=[],
|
||||
)
|
Loading…
Reference in New Issue
Block a user