1
0
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:
Olof Johansson 2016-03-13 16:52:54 +01:00
parent 4740a42050
commit e781a1e89b

View 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=[],
)