105 lines
2.7 KiB
Python
105 lines
2.7 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
import core.input
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def obj():
|
||
|
|
return core.input.Object()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def obj2():
|
||
|
|
return core.input.Object()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def cb(mocker):
|
||
|
|
return mocker.MagicMock()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def cb2(mocker):
|
||
|
|
return mocker.MagicMock()
|
||
|
|
|
||
|
|
|
||
|
|
def event(input_object):
|
||
|
|
return {"button": core.input.LEFT_MOUSE, "instance": input_object.id}
|
||
|
|
|
||
|
|
|
||
|
|
def event2(input_object):
|
||
|
|
return {"button": core.input.RIGHT_MOUSE, "instance": input_object.id}
|
||
|
|
|
||
|
|
|
||
|
|
def test_callable_gets_called(obj, cb):
|
||
|
|
core.input.register(obj, event(obj)["button"], cb)
|
||
|
|
core.input.trigger(event(obj))
|
||
|
|
|
||
|
|
cb.assert_called_once_with(event(obj))
|
||
|
|
|
||
|
|
|
||
|
|
def test_nonexistent_callback(obj, obj2, cb):
|
||
|
|
core.input.register(obj, event(obj)["button"], cb)
|
||
|
|
core.input.trigger(event(obj2))
|
||
|
|
|
||
|
|
cb.assert_not_called()
|
||
|
|
|
||
|
|
|
||
|
|
def test_different_events(obj, obj2, cb, cb2):
|
||
|
|
core.input.register(obj, event(obj)["button"], cb)
|
||
|
|
core.input.register(obj, event2(obj)["button"], cb)
|
||
|
|
core.input.register(obj2, event(obj)["button"], cb2)
|
||
|
|
core.input.register(obj2, event2(obj)["button"], cb2)
|
||
|
|
|
||
|
|
core.input.trigger(event(obj))
|
||
|
|
core.input.trigger(event2(obj))
|
||
|
|
|
||
|
|
cb.assert_any_call(event(obj))
|
||
|
|
cb.assert_any_call(event2(obj))
|
||
|
|
cb2.assert_not_called()
|
||
|
|
|
||
|
|
|
||
|
|
def test_multiple_registrations(obj, cb, cb2):
|
||
|
|
core.input.register(obj, event(obj)["button"], cb)
|
||
|
|
core.input.register(obj, event(obj)["button"], cb2)
|
||
|
|
|
||
|
|
core.input.trigger(event(obj))
|
||
|
|
|
||
|
|
cb.assert_called_once_with(event(obj))
|
||
|
|
cb2.assert_called_once_with(event(obj))
|
||
|
|
|
||
|
|
|
||
|
|
def test_event_names():
|
||
|
|
assert core.input.button_name(core.input.LEFT_MOUSE) == "left-mouse"
|
||
|
|
assert core.input.button_name(core.input.RIGHT_MOUSE) == "right-mouse"
|
||
|
|
assert core.input.button_name(core.input.MIDDLE_MOUSE) == "middle-mouse"
|
||
|
|
assert core.input.button_name(core.input.WHEEL_UP) == "wheel-up"
|
||
|
|
assert core.input.button_name(core.input.WHEEL_DOWN) == "wheel-down"
|
||
|
|
assert core.input.button_name(12345) == "n/a"
|
||
|
|
|
||
|
|
|
||
|
|
def test_non_callable_callback(mocker, obj):
|
||
|
|
cli = mocker.patch("core.input.util.cli")
|
||
|
|
cli.execute.return_value = ""
|
||
|
|
|
||
|
|
core.input.register(obj, event(obj)["button"], "sample-command")
|
||
|
|
|
||
|
|
core.input.trigger(event(obj))
|
||
|
|
|
||
|
|
cli.execute.assert_called_once_with("sample-command", wait=False, shell=True)
|
||
|
|
|
||
|
|
|
||
|
|
def test_non_existent_callback(mocker, obj):
|
||
|
|
cli = mocker.patch("core.input.util.cli")
|
||
|
|
cli.execute.return_value = ""
|
||
|
|
cli.execute.side_effect = RuntimeError("some-error")
|
||
|
|
|
||
|
|
core.input.register(obj, event(obj)["button"], "sample-command")
|
||
|
|
|
||
|
|
core.input.trigger(event(obj))
|
||
|
|
|
||
|
|
cli.execute.assert_called_once_with("sample-command", wait=False, shell=True)
|
||
|
|
|
||
|
|
|
||
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|