mirror of
https://github.com/onyx-and-iris/obsws-python.git
synced 2024-11-25 14:20:50 +00:00
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import pytest
|
|
from obsws_python.callback import Callback
|
|
|
|
|
|
class TestCallbacks:
|
|
__test__ = True
|
|
|
|
@classmethod
|
|
def setup_class(cls):
|
|
cls.callback = Callback()
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def wraps_tests(self):
|
|
yield
|
|
self.callback.clear()
|
|
|
|
def test_register_callback(self):
|
|
def on_callback_method():
|
|
pass
|
|
|
|
self.callback.register(on_callback_method)
|
|
assert self.callback.get() == ["CallbackMethod"]
|
|
|
|
def test_register_callbacks(self):
|
|
def on_callback_method_one():
|
|
pass
|
|
|
|
def on_callback_method_two():
|
|
pass
|
|
|
|
self.callback.register((on_callback_method_one, on_callback_method_two))
|
|
assert self.callback.get() == ["CallbackMethodOne", "CallbackMethodTwo"]
|
|
|
|
def test_deregister_callback(self):
|
|
def on_callback_method_one():
|
|
pass
|
|
|
|
def on_callback_method_two():
|
|
pass
|
|
|
|
self.callback.register((on_callback_method_one, on_callback_method_two))
|
|
self.callback.deregister(on_callback_method_one)
|
|
assert self.callback.get() == ["CallbackMethodTwo"]
|
|
|
|
def test_deregister_callbacks(self):
|
|
def on_callback_method_one():
|
|
pass
|
|
|
|
def on_callback_method_two():
|
|
pass
|
|
|
|
def on_callback_method_three():
|
|
pass
|
|
|
|
self.callback.register(
|
|
(on_callback_method_one, on_callback_method_two, on_callback_method_three)
|
|
)
|
|
self.callback.deregister((on_callback_method_two, on_callback_method_three))
|
|
assert self.callback.get() == ["CallbackMethodOne"]
|