mpv-test: Fix some tests, add property observer decorator test

Also fix ass-style-override
This commit is contained in:
jaseg 2017-05-26 14:01:45 +02:00
parent 8608c43854
commit 11f534897c
2 changed files with 52 additions and 14 deletions

View file

@ -30,6 +30,9 @@ class TestProperties(unittest.TestCase):
def setUp(self): def setUp(self):
self.m = mpv.MPV() self.m = mpv.MPV()
def tearDown(self):
self.m.terminate()
def test_sanity(self): def test_sanity(self):
for name, (ptype, access, *_args) in mpv.ALL_PROPERTIES.items(): for name, (ptype, access, *_args) in mpv.ALL_PROPERTIES.items():
self.assertTrue('r' in access or 'w' in access) self.assertTrue('r' in access or 'w' in access)
@ -75,7 +78,9 @@ class TestProperties(unittest.TestCase):
if 'r' in access: if 'r' in access:
name = name.replace('-', '_') name = name.replace('-', '_')
with self.subTest(property_name=name), self.swallow_mpv_errors([ with self.subTest(property_name=name), self.swallow_mpv_errors([
mpv.ErrorCode.PROPERTY_UNAVAILABLE, mpv.ErrorCode.PROPERTY_ERROR]): mpv.ErrorCode.PROPERTY_UNAVAILABLE,
mpv.ErrorCode.PROPERTY_ERROR,
mpv.ErrorCode.PROPERTY_NOT_FOUND]):
rv = getattr(self.m, name) rv = getattr(self.m, name)
if rv is not None and callable(ptype): if rv is not None and callable(ptype):
# Technically, any property can return None (even if of type e.g. int) # Technically, any property can return None (even if of type e.g. int)
@ -92,7 +97,8 @@ class TestProperties(unittest.TestCase):
with self.subTest(property_name=name), self.swallow_mpv_errors([ with self.subTest(property_name=name), self.swallow_mpv_errors([
mpv.ErrorCode.PROPERTY_UNAVAILABLE, mpv.ErrorCode.PROPERTY_UNAVAILABLE,
mpv.ErrorCode.PROPERTY_ERROR, mpv.ErrorCode.PROPERTY_ERROR,
mpv.ErrorCode.PROPERTY_FORMAT]): # This is due to a bug with option-mapped properties in mpv 0.18.1 mpv.ErrorCode.PROPERTY_FORMAT,
mpv.ErrorCode.PROPERTY_NOT_FOUND]): # This is due to a bug with option-mapped properties in mpv 0.18.1
if ptype == int: if ptype == int:
setattr(self.m, name, 0) setattr(self.m, name, 0)
setattr(self.m, name, 1) setattr(self.m, name, 1)
@ -126,13 +132,9 @@ class TestProperties(unittest.TestCase):
self.m[name] self.m[name]
def tearDown(self):
del self.m
class ObservePropertyTest(unittest.TestCase): class ObservePropertyTest(unittest.TestCase):
def test_observe_property(self): def test_observe_property(self):
handler = mock.Mock() handler = mock.Mock()
handler.observed_mpv_properties = []
m = mpv.MPV() m = mpv.MPV()
m.loop = 'inf' m.loop = 'inf'
@ -151,8 +153,43 @@ class ObservePropertyTest(unittest.TestCase):
m.loop = 'no' m.loop = 'no'
m.loop = 'inf' m.loop = 'inf'
m.terminate() # needed for synchronization of event thread m.terminate() # needed for synchronization of event thread
handler.assert_has_calls([mock.call('loop', False), mock.call('loop', 'inf')]) handler.assert_has_calls([mock.call('loop', 'no'), mock.call('loop', 'inf')])
def test_property_observer_decorator(self):
handler = mock.Mock()
m = mpv.MPV()
m.loop = 'inf'
m.mute = True
@m.property_observer('mute')
@m.property_observer('loop')
def foo(*args, **kwargs):
handler(*args, **kwargs)
m.mute = False
m.loop = 'no'
self.assertEqual(m.mute, False)
self.assertEqual(m.loop, 'no')
m.mute = True
m.loop = 'inf'
self.assertEqual(m.mute, True)
self.assertEqual(m.loop, 'inf')
time.sleep(0.02)
foo.unobserve_mpv_properties()
m.mute = False
m.loop = 'no'
m.mute = True
m.loop = 'inf'
m.terminate() # needed for synchronization of event thread
handler.assert_has_calls([
mock.call('mute', False),
mock.call('loop', 'no'),
mock.call('mute', True),
mock.call('loop', 'inf')])
class TestLifecycle(unittest.TestCase): class TestLifecycle(unittest.TestCase):
def test_create_destroy(self): def test_create_destroy(self):
@ -170,6 +207,7 @@ class TestLifecycle(unittest.TestCase):
m = mpv.MPV('no-video', 'cursor-autohide-fs-only', 'fs') m = mpv.MPV('no-video', 'cursor-autohide-fs-only', 'fs')
self.assertTrue(m.fullscreen) self.assertTrue(m.fullscreen)
self.assertEqual(m.cursor_autohide, '1000') self.assertEqual(m.cursor_autohide, '1000')
m.terminate()
def test_options(self): def test_options(self):
with self.assertRaises(AttributeError): with self.assertRaises(AttributeError):
@ -178,6 +216,7 @@ class TestLifecycle(unittest.TestCase):
self.assertEqual(m.osd_level, 0) self.assertEqual(m.osd_level, 0)
self.assertEqual(m.loop, 'inf') self.assertEqual(m.loop, 'inf')
self.assertEqual(m.deinterlace, 'no') self.assertEqual(m.deinterlace, 'no')
m.terminate()
def test_event_callback(self): def test_event_callback(self):
handler = mock.Mock() handler = mock.Mock()
@ -194,7 +233,7 @@ class TestLifecycle(unittest.TestCase):
], any_order=True) ], any_order=True)
handler.reset_mock() handler.reset_mock()
del m m.terminate()
handler.assert_not_called() handler.assert_not_called()
def test_log_handler(self): def test_log_handler(self):
@ -202,8 +241,8 @@ class TestLifecycle(unittest.TestCase):
m = mpv.MPV('no-video', log_handler=handler) m = mpv.MPV('no-video', log_handler=handler)
m.play(TESTVID) m.play(TESTVID)
m.wait_for_playback() m.wait_for_playback()
del m m.terminate()
handler.assert_any_call('info', 'cplayer', 'Playing: ./test.webm') handler.assert_any_call('info', 'cplayer', 'Playing: test.webm')
class RegressionTests(unittest.TestCase): class RegressionTests(unittest.TestCase):

7
mpv.py
View file

@ -662,8 +662,6 @@ class MPV(object):
my_handler.unregister_mpv_properties() my_handler.unregister_mpv_properties()
``` """ ``` """
fmt = force_fmt or MpvFormat.NODE fmt = force_fmt or MpvFormat.NODE
handler.observed_mpv_properties = getattr(handler, 'observed_mpv_properties', []) + [name]
handler.unregister_mpv_properties = lambda: self.unobserve_property(None, handler)
self._property_handlers[name][fmt].append(handler) self._property_handlers[name][fmt].append(handler)
_mpv_observe_property(self._event_handle, hash(name)&0xffffffffffffffff, name.encode('utf-8'), fmt) _mpv_observe_property(self._event_handle, hash(name)&0xffffffffffffffff, name.encode('utf-8'), fmt)
@ -671,6 +669,7 @@ class MPV(object):
""" Function decorator to register a property observer. See ```MPV.observe_property``` for details. """ """ Function decorator to register a property observer. See ```MPV.observe_property``` for details. """
def wrapper(fun): def wrapper(fun):
self.observe_property(name, fun, force_fmt=force_fmt) self.observe_property(name, fun, force_fmt=force_fmt)
fun.unobserve_mpv_properties = lambda: self.unobserve_property(None, fun)
return fun return fun
return wrapper return wrapper
@ -718,7 +717,6 @@ class MPV(object):
def _register_script_message_handler_internal(self, target, handler): def _register_script_message_handler_internal(self, target, handler):
handler.mpv_message_targets = getattr(handler, 'mpv_script_message_targets', []) + [target] handler.mpv_message_targets = getattr(handler, 'mpv_script_message_targets', []) + [target]
handler.unregister_mpv_messages = lambda: self.unregister_message_handler(handler)
self._message_handlers[target] = handler self._message_handlers[target] = handler
def unregister_message_handler(self, target_or_handler): def unregister_message_handler(self, target_or_handler):
@ -750,6 +748,7 @@ class MPV(object):
""" """
def register(handler): def register(handler):
self._register_message_handler_internal(target, handler) self._register_message_handler_internal(target, handler)
handler.unregister_mpv_messages = lambda: self.unregister_message_handler(handler)
return handler return handler
return register return register
@ -1072,7 +1071,7 @@ ALL_PROPERTIES = {
'packet-sub-bitrate': (float, 'r'), 'packet-sub-bitrate': (float, 'r'),
# 'ass-use-margins': (bool, 'rw'), # 'ass-use-margins': (bool, 'rw'),
'ass-vsfilter-aspect-compat': (bool, 'rw'), 'ass-vsfilter-aspect-compat': (bool, 'rw'),
'ass-style-override': (bool, 'rw'), 'ass-style-override': (str, 'rw'),
# 'stream-capture': (str, 'rw'), # 'stream-capture': (str, 'rw'),
'tv-brightness': (int, 'rw'), 'tv-brightness': (int, 'rw'),
'tv-contrast': (int, 'rw'), 'tv-contrast': (int, 'rw'),