2020-12-03 19:03:53 +00:00
|
|
|
#!/usr/bin/python3
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# key-mapper - GUI for device specific keyboard mappings
|
|
|
|
# Copyright (C) 2020 sezanzeb <proxima@hip70890b.de>
|
|
|
|
#
|
|
|
|
# This file is part of key-mapper.
|
|
|
|
#
|
|
|
|
# key-mapper is free software: you can redistribute it and/or modify
|
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
|
# (at your option) any later version.
|
|
|
|
#
|
|
|
|
# key-mapper is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
# along with key-mapper. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
|
|
|
|
import unittest
|
2020-12-06 13:23:00 +00:00
|
|
|
import asyncio
|
2020-12-19 15:04:07 +00:00
|
|
|
import time
|
2020-12-03 19:03:53 +00:00
|
|
|
|
2020-12-24 00:26:34 +00:00
|
|
|
from evdev import ecodes
|
2020-12-19 15:04:07 +00:00
|
|
|
from evdev.ecodes import EV_KEY, EV_ABS, ABS_HAT0X, ABS_HAT0Y, KEY_A, ABS_X, \
|
|
|
|
EV_REL, REL_X, BTN_TL
|
2020-12-03 19:03:53 +00:00
|
|
|
|
2020-12-06 13:23:00 +00:00
|
|
|
from keymapper.dev.keycode_mapper import should_map_event_as_btn, \
|
|
|
|
active_macros, handle_keycode
|
|
|
|
from keymapper.state import system_mapping
|
|
|
|
from keymapper.dev.macros import parse
|
|
|
|
from keymapper.config import config
|
2020-12-19 20:50:27 +00:00
|
|
|
from keymapper.mapping import Mapping
|
2020-12-06 13:23:00 +00:00
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
from tests.test import InputEvent, UInput, uinput_write_history, \
|
2020-12-24 00:26:34 +00:00
|
|
|
cleanup
|
2020-12-19 15:04:07 +00:00
|
|
|
|
|
|
|
|
|
|
|
def wait(func, timeout=1.0):
|
|
|
|
"""Wait for func to return True."""
|
|
|
|
iterations = 0
|
|
|
|
sleepytime = 0.1
|
|
|
|
while not func():
|
|
|
|
time.sleep(sleepytime)
|
|
|
|
iterations += 1
|
|
|
|
if iterations * sleepytime > timeout:
|
|
|
|
raise Exception('Timeout')
|
|
|
|
|
|
|
|
|
|
|
|
def calculate_event_number(holdtime, before, after):
|
|
|
|
"""
|
|
|
|
Parameters
|
|
|
|
----------
|
|
|
|
holdtime : int
|
|
|
|
in ms, how long was the key held down
|
|
|
|
before : int
|
|
|
|
how many extra k() calls are executed before h()
|
|
|
|
after : int
|
|
|
|
how many extra k() calls are executed after h()
|
|
|
|
"""
|
|
|
|
keystroke_sleep = config.get('macros.keystroke_sleep_ms', 10)
|
|
|
|
# down and up: two sleeps per k
|
|
|
|
# one initial k(a):
|
|
|
|
events = before * 2
|
|
|
|
holdtime -= keystroke_sleep * 2
|
|
|
|
# hold events
|
|
|
|
events += (holdtime / (keystroke_sleep * 2)) * 2
|
|
|
|
# one trailing k(c)
|
|
|
|
events += after * 2
|
|
|
|
return events
|
2020-12-03 19:03:53 +00:00
|
|
|
|
|
|
|
|
|
|
|
class TestKeycodeMapper(unittest.TestCase):
|
2020-12-19 20:50:27 +00:00
|
|
|
def setUp(self):
|
|
|
|
self.mapping = Mapping()
|
|
|
|
|
2020-12-06 13:23:00 +00:00
|
|
|
def tearDown(self):
|
|
|
|
# make sure all macros are stopped by tests
|
2020-12-19 15:04:07 +00:00
|
|
|
for macro in active_macros.values():
|
2020-12-06 13:23:00 +00:00
|
|
|
if macro.holding:
|
|
|
|
macro.release_key()
|
|
|
|
self.assertFalse(macro.holding)
|
|
|
|
self.assertFalse(macro.running)
|
|
|
|
|
2020-12-24 00:26:34 +00:00
|
|
|
cleanup()
|
2020-12-19 15:04:07 +00:00
|
|
|
|
|
|
|
def test_d_pad(self):
|
|
|
|
ev_1 = (EV_ABS, ABS_HAT0X, 1)
|
|
|
|
ev_2 = (EV_ABS, ABS_HAT0X, -1)
|
|
|
|
ev_3 = (EV_ABS, ABS_HAT0X, 0)
|
|
|
|
|
|
|
|
ev_4 = (EV_ABS, ABS_HAT0Y, 1)
|
|
|
|
ev_5 = (EV_ABS, ABS_HAT0Y, -1)
|
|
|
|
ev_6 = (EV_ABS, ABS_HAT0Y, 0)
|
|
|
|
|
|
|
|
_key_to_code = {
|
|
|
|
ev_1: 51,
|
|
|
|
ev_2: 52,
|
|
|
|
ev_4: 54,
|
|
|
|
ev_5: 55,
|
|
|
|
}
|
|
|
|
|
|
|
|
uinput = UInput()
|
|
|
|
# a bunch of d-pad key down events at once
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_1), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_4), uinput)
|
|
|
|
|
|
|
|
# release all of them
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_3), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_6), uinput)
|
|
|
|
|
|
|
|
# repeat with other values
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_2), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_5), uinput)
|
|
|
|
|
|
|
|
# release all of them again
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_3), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_6), uinput)
|
|
|
|
|
|
|
|
self.assertEqual(len(uinput_write_history), 8)
|
|
|
|
|
|
|
|
self.assertEqual(uinput_write_history[0].t, (EV_KEY, 51, 1))
|
|
|
|
self.assertEqual(uinput_write_history[1].t, (EV_KEY, 54, 1))
|
|
|
|
|
|
|
|
self.assertEqual(uinput_write_history[2].t, (EV_KEY, 51, 0))
|
|
|
|
self.assertEqual(uinput_write_history[3].t, (EV_KEY, 54, 0))
|
|
|
|
|
|
|
|
self.assertEqual(uinput_write_history[4].t, (EV_KEY, 52, 1))
|
|
|
|
self.assertEqual(uinput_write_history[5].t, (EV_KEY, 55, 1))
|
|
|
|
|
|
|
|
self.assertEqual(uinput_write_history[6].t, (EV_KEY, 52, 0))
|
|
|
|
self.assertEqual(uinput_write_history[7].t, (EV_KEY, 55, 0))
|
2020-12-06 14:11:13 +00:00
|
|
|
|
2020-12-03 19:03:53 +00:00
|
|
|
def test_should_map_event_as_btn(self):
|
|
|
|
self.assertTrue(should_map_event_as_btn(EV_ABS, ABS_HAT0X))
|
|
|
|
self.assertTrue(should_map_event_as_btn(EV_KEY, KEY_A))
|
|
|
|
self.assertFalse(should_map_event_as_btn(EV_ABS, ABS_X))
|
|
|
|
self.assertFalse(should_map_event_as_btn(EV_REL, REL_X))
|
|
|
|
|
2020-12-24 00:26:34 +00:00
|
|
|
self.assertFalse(should_map_event_as_btn(EV_ABS, ecodes.ABS_MT_SLOT))
|
|
|
|
self.assertFalse(should_map_event_as_btn(EV_ABS, ecodes.ABS_MT_TOOL_Y))
|
|
|
|
self.assertFalse(should_map_event_as_btn(EV_ABS, ecodes.ABS_MT_POSITION_X))
|
|
|
|
|
2020-12-06 13:23:00 +00:00
|
|
|
def test_handle_keycode(self):
|
2020-12-19 15:04:07 +00:00
|
|
|
_key_to_code = {
|
|
|
|
(EV_KEY, 1, 1): 101,
|
|
|
|
(EV_KEY, 2, 1): 102
|
2020-12-06 13:23:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
uinput = UInput()
|
2020-12-19 15:04:07 +00:00
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(EV_KEY, 1, 1), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(EV_KEY, 3, 1), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(EV_KEY, 2, 1), uinput)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertEqual(len(uinput_write_history), 3)
|
|
|
|
self.assertEqual(uinput_write_history[0].t, (EV_KEY, 101, 1))
|
|
|
|
self.assertEqual(uinput_write_history[1].t, (EV_KEY, 3, 1))
|
|
|
|
self.assertEqual(uinput_write_history[2].t, (EV_KEY, 102, 1))
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
def test_handle_keycode_macro(self):
|
|
|
|
history = []
|
|
|
|
|
|
|
|
code_a = 100
|
|
|
|
code_b = 101
|
|
|
|
system_mapping.clear()
|
|
|
|
system_mapping._set('a', code_a)
|
|
|
|
system_mapping._set('b', code_b)
|
|
|
|
|
|
|
|
macro_mapping = {
|
2020-12-19 20:50:27 +00:00
|
|
|
(EV_KEY, 1, 1): parse('k(a)', self.mapping),
|
|
|
|
(EV_KEY, 2, 1): parse('r(5, k(b))', self.mapping)
|
2020-12-06 13:23:00 +00:00
|
|
|
}
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
macro_mapping[(EV_KEY, 1, 1)].set_handler(lambda *args: history.append(args))
|
|
|
|
macro_mapping[(EV_KEY, 2, 1)].set_handler(lambda *args: history.append(args))
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 1), None)
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 2, 1), None)
|
|
|
|
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
|
|
|
|
sleeptime = config.get('macros.keystroke_sleep_ms', 10) * 12
|
|
|
|
|
|
|
|
# let the mainloop run for some time so that the macro does its stuff
|
|
|
|
loop.run_until_complete(asyncio.sleep(sleeptime / 1000 + 0.1))
|
|
|
|
|
|
|
|
# 6 keycodes written, with down and up events
|
|
|
|
self.assertEqual(len(history), 12)
|
|
|
|
self.assertIn((code_a, 1), history)
|
|
|
|
self.assertIn((code_a, 0), history)
|
|
|
|
self.assertIn((code_b, 1), history)
|
|
|
|
self.assertIn((code_b, 0), history)
|
|
|
|
|
|
|
|
def test_hold(self):
|
|
|
|
history = []
|
|
|
|
|
|
|
|
code_a = 100
|
|
|
|
code_b = 101
|
|
|
|
code_c = 102
|
|
|
|
system_mapping.clear()
|
|
|
|
system_mapping._set('a', code_a)
|
|
|
|
system_mapping._set('b', code_b)
|
|
|
|
system_mapping._set('c', code_c)
|
|
|
|
|
|
|
|
macro_mapping = {
|
2020-12-19 20:50:27 +00:00
|
|
|
(EV_KEY, 1, 1): parse('k(a).h(k(b)).k(c)', self.mapping)
|
2020-12-06 13:23:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
def handler(*args):
|
|
|
|
history.append(args)
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
macro_mapping[(EV_KEY, 1, 1)].set_handler(handler)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
"""start macro"""
|
|
|
|
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 1), None)
|
|
|
|
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
|
|
|
|
# let the mainloop run for some time so that the macro does its stuff
|
|
|
|
sleeptime = 500
|
|
|
|
keystroke_sleep = config.get('macros.keystroke_sleep_ms', 10)
|
|
|
|
loop.run_until_complete(asyncio.sleep(sleeptime / 1000))
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertTrue(active_macros[(EV_KEY, 1)].holding)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 1)].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
"""stop macro"""
|
|
|
|
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 0), None)
|
|
|
|
|
|
|
|
loop.run_until_complete(asyncio.sleep(keystroke_sleep * 10 / 1000))
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
events = calculate_event_number(sleeptime, 1, 1)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
self.assertGreater(len(history), events * 0.9)
|
|
|
|
self.assertLess(len(history), events * 1.1)
|
|
|
|
|
|
|
|
self.assertIn((code_a, 1), history)
|
|
|
|
self.assertIn((code_a, 0), history)
|
|
|
|
self.assertIn((code_b, 1), history)
|
|
|
|
self.assertIn((code_b, 0), history)
|
|
|
|
self.assertIn((code_c, 1), history)
|
|
|
|
self.assertIn((code_c, 0), history)
|
|
|
|
self.assertGreater(history.count((code_b, 1)), 1)
|
|
|
|
self.assertGreater(history.count((code_b, 0)), 1)
|
|
|
|
|
|
|
|
# it's stopped and won't write stuff anymore
|
|
|
|
count_before = len(history)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.2))
|
|
|
|
count_after = len(history)
|
|
|
|
self.assertEqual(count_before, count_after)
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertFalse(active_macros[(EV_KEY, 1)].holding)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 1)].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
def test_hold_2(self):
|
|
|
|
# test irregular input patterns
|
|
|
|
code_a = 100
|
|
|
|
code_b = 101
|
|
|
|
code_c = 102
|
|
|
|
code_d = 103
|
|
|
|
system_mapping.clear()
|
|
|
|
system_mapping._set('a', code_a)
|
|
|
|
system_mapping._set('b', code_b)
|
|
|
|
system_mapping._set('c', code_c)
|
|
|
|
system_mapping._set('d', code_d)
|
|
|
|
|
|
|
|
macro_mapping = {
|
2020-12-19 20:50:27 +00:00
|
|
|
(EV_KEY, 1, 1): parse('h(k(b))', self.mapping),
|
|
|
|
(EV_KEY, 2, 1): parse('k(c).r(1, r(1, r(1, h(k(a))))).k(d)', self.mapping),
|
|
|
|
(EV_KEY, 3, 1): parse('h(k(b))', self.mapping)
|
2020-12-06 13:23:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
history = []
|
|
|
|
|
|
|
|
def handler(*args):
|
|
|
|
history.append(args)
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
macro_mapping[(EV_KEY, 1, 1)].set_handler(handler)
|
|
|
|
macro_mapping[(EV_KEY, 2, 1)].set_handler(handler)
|
|
|
|
macro_mapping[(EV_KEY, 3, 1)].set_handler(handler)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
"""start macro 2"""
|
|
|
|
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 2, 1), None)
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
# starting code_c written
|
|
|
|
self.assertEqual(history.count((code_c, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_c, 0)), 1)
|
|
|
|
|
|
|
|
# spam garbage events
|
|
|
|
for _ in range(5):
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 1), None)
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 3, 1), None)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.05))
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertTrue(active_macros[(EV_KEY, 1)].holding)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 1)].running)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 2)].holding)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 2)].running)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 3)].holding)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 3)].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
# there should only be one code_c in the events, because no key
|
|
|
|
# up event was ever done so the hold just continued
|
|
|
|
self.assertEqual(history.count((code_c, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_c, 0)), 1)
|
|
|
|
# without an key up event on 2, it won't write code_d
|
|
|
|
self.assertNotIn((code_d, 1), history)
|
|
|
|
self.assertNotIn((code_d, 0), history)
|
|
|
|
|
|
|
|
# stop macro 2
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 2, 0), None)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
|
|
|
|
# it stopped and didn't restart, so the count stays at 1
|
|
|
|
self.assertEqual(history.count((code_c, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_c, 0)), 1)
|
|
|
|
# and the trailing d was written
|
|
|
|
self.assertEqual(history.count((code_d, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_d, 0)), 1)
|
|
|
|
|
|
|
|
# it's stopped and won't write stuff anymore
|
|
|
|
count_before = history.count((code_a, 1))
|
|
|
|
self.assertGreater(count_before, 1)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
count_after = history.count((code_a, 1))
|
|
|
|
self.assertEqual(count_before, count_after)
|
|
|
|
|
|
|
|
"""restart macro 2"""
|
|
|
|
|
|
|
|
history = []
|
|
|
|
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 2, 1), None)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
self.assertEqual(history.count((code_c, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_c, 0)), 1)
|
|
|
|
|
|
|
|
# spam garbage events again, this time key-up events on all other
|
|
|
|
# macros
|
|
|
|
for _ in range(5):
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 0), None)
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 3, 0), None)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.05))
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertFalse(active_macros[(EV_KEY, 1)].holding)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 1)].running)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 2)].holding)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 2)].running)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 3)].holding)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 3)].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
# stop macro 2
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 2, 0), None)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
# was started only once
|
|
|
|
self.assertEqual(history.count((code_c, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_c, 0)), 1)
|
|
|
|
# and the trailing d was also written only once
|
|
|
|
self.assertEqual(history.count((code_d, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_d, 0)), 1)
|
|
|
|
|
|
|
|
# stop all macros
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 0), None)
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 3, 0), None)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
|
|
|
|
# it's stopped and won't write stuff anymore
|
|
|
|
count_before = len(history)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
count_after = len(history)
|
|
|
|
self.assertEqual(count_before, count_after)
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertFalse(active_macros[(EV_KEY, 1)].holding)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 1)].running)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 2)].holding)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 2)].running)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 3)].holding)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 3)].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
def test_hold_3(self):
|
|
|
|
# test irregular input patterns
|
|
|
|
code_a = 100
|
|
|
|
code_b = 101
|
|
|
|
code_c = 102
|
|
|
|
system_mapping.clear()
|
|
|
|
system_mapping._set('a', code_a)
|
|
|
|
system_mapping._set('b', code_b)
|
|
|
|
system_mapping._set('c', code_c)
|
|
|
|
|
|
|
|
macro_mapping = {
|
2020-12-19 20:50:27 +00:00
|
|
|
(EV_KEY, 1, 1): parse('k(a).h(k(b)).k(c)', self.mapping),
|
2020-12-06 13:23:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
history = []
|
|
|
|
|
|
|
|
def handler(*args):
|
|
|
|
history.append(args)
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
macro_mapping[(EV_KEY, 1, 1)].set_handler(handler)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 1), None)
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
for _ in range(5):
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertTrue(active_macros[(EV_KEY, 1)].holding)
|
|
|
|
self.assertTrue(active_macros[(EV_KEY, 1)].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 1), None)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.05))
|
|
|
|
|
|
|
|
# duplicate key down events don't do anything
|
|
|
|
self.assertEqual(history.count((code_a, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_a, 0)), 1)
|
|
|
|
self.assertEqual(history.count((code_c, 1)), 0)
|
|
|
|
self.assertEqual(history.count((code_c, 0)), 0)
|
|
|
|
|
|
|
|
# stop
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(EV_KEY, 1, 0), None)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
self.assertEqual(history.count((code_a, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_a, 0)), 1)
|
|
|
|
self.assertEqual(history.count((code_c, 1)), 1)
|
|
|
|
self.assertEqual(history.count((code_c, 0)), 1)
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertFalse(active_macros[(EV_KEY, 1)].holding)
|
|
|
|
self.assertFalse(active_macros[(EV_KEY, 1)].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
# it's stopped and won't write stuff anymore
|
|
|
|
count_before = len(history)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
count_after = len(history)
|
|
|
|
self.assertEqual(count_before, count_after)
|
|
|
|
|
|
|
|
def test_hold_two(self):
|
2020-12-19 15:04:07 +00:00
|
|
|
# holding two macros at the same time
|
2020-12-06 13:23:00 +00:00
|
|
|
history = []
|
|
|
|
|
|
|
|
code_1 = 100
|
|
|
|
code_2 = 101
|
|
|
|
code_3 = 102
|
|
|
|
code_a = 103
|
|
|
|
code_b = 104
|
|
|
|
code_c = 105
|
|
|
|
system_mapping.clear()
|
|
|
|
system_mapping._set('1', code_1)
|
|
|
|
system_mapping._set('2', code_2)
|
|
|
|
system_mapping._set('3', code_3)
|
|
|
|
system_mapping._set('a', code_a)
|
|
|
|
system_mapping._set('b', code_b)
|
|
|
|
system_mapping._set('c', code_c)
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
key_1 = (EV_KEY, 1)
|
|
|
|
key_2 = (EV_ABS, ABS_HAT0X)
|
|
|
|
down_1 = (*key_1, 1)
|
|
|
|
down_2 = (*key_2, -1)
|
|
|
|
up_1 = (*key_1, 0)
|
|
|
|
up_2 = (*key_2, 0)
|
|
|
|
|
2020-12-06 13:23:00 +00:00
|
|
|
macro_mapping = {
|
2020-12-19 20:50:27 +00:00
|
|
|
down_1: parse('k(1).h(k(2)).k(3)', self.mapping),
|
|
|
|
down_2: parse('k(a).h(k(b)).k(c)', self.mapping)
|
2020-12-06 13:23:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
def handler(*args):
|
|
|
|
history.append(args)
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
macro_mapping[down_1].set_handler(handler)
|
|
|
|
macro_mapping[down_2].set_handler(handler)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
|
|
|
|
# key up won't do anything
|
2020-12-19 15:04:07 +00:00
|
|
|
uinput = UInput()
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(*up_1), uinput)
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(*up_2), uinput)
|
2020-12-06 13:23:00 +00:00
|
|
|
loop.run_until_complete(asyncio.sleep(0.1))
|
|
|
|
self.assertEqual(len(active_macros), 0)
|
|
|
|
|
|
|
|
"""start macros"""
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
handle_keycode({}, macro_mapping, InputEvent(*down_1), None)
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(*down_2), None)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
# let the mainloop run for some time so that the macro does its stuff
|
|
|
|
sleeptime = 500
|
|
|
|
keystroke_sleep = config.get('macros.keystroke_sleep_ms', 10)
|
|
|
|
loop.run_until_complete(asyncio.sleep(sleeptime / 1000))
|
|
|
|
|
|
|
|
self.assertEqual(len(active_macros), 2)
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertTrue(active_macros[key_1].holding)
|
|
|
|
self.assertTrue(active_macros[key_1].running)
|
|
|
|
self.assertTrue(active_macros[key_2].holding)
|
|
|
|
self.assertTrue(active_macros[key_2].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
"""stop macros"""
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
handle_keycode({}, macro_mapping, InputEvent(*up_1), None)
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(*up_2), None)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
loop.run_until_complete(asyncio.sleep(keystroke_sleep * 10 / 1000))
|
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
self.assertFalse(active_macros[key_1].holding)
|
|
|
|
self.assertFalse(active_macros[key_1].running)
|
|
|
|
self.assertFalse(active_macros[key_2].holding)
|
|
|
|
self.assertFalse(active_macros[key_2].running)
|
2020-12-06 13:23:00 +00:00
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
events = calculate_event_number(sleeptime, 1, 1) * 2
|
2020-12-06 13:23:00 +00:00
|
|
|
|
|
|
|
self.assertGreater(len(history), events * 0.9)
|
|
|
|
self.assertLess(len(history), events * 1.1)
|
|
|
|
|
|
|
|
self.assertIn((code_a, 1), history)
|
|
|
|
self.assertIn((code_a, 0), history)
|
|
|
|
self.assertIn((code_b, 1), history)
|
|
|
|
self.assertIn((code_b, 0), history)
|
|
|
|
self.assertIn((code_c, 1), history)
|
|
|
|
self.assertIn((code_c, 0), history)
|
|
|
|
self.assertIn((code_1, 1), history)
|
|
|
|
self.assertIn((code_1, 0), history)
|
|
|
|
self.assertIn((code_2, 1), history)
|
|
|
|
self.assertIn((code_2, 0), history)
|
|
|
|
self.assertIn((code_3, 1), history)
|
|
|
|
self.assertIn((code_3, 0), history)
|
|
|
|
self.assertGreater(history.count((code_b, 1)), 1)
|
|
|
|
self.assertGreater(history.count((code_b, 0)), 1)
|
|
|
|
self.assertGreater(history.count((code_2, 1)), 1)
|
|
|
|
self.assertGreater(history.count((code_2, 0)), 1)
|
2020-12-03 19:03:53 +00:00
|
|
|
|
2020-12-06 13:23:00 +00:00
|
|
|
# it's stopped and won't write stuff anymore
|
|
|
|
count_before = len(history)
|
|
|
|
loop.run_until_complete(asyncio.sleep(0.2))
|
|
|
|
count_after = len(history)
|
|
|
|
self.assertEqual(count_before, count_after)
|
2020-12-06 11:54:14 +00:00
|
|
|
|
2020-12-19 15:04:07 +00:00
|
|
|
def test_two_d_pad_macros(self):
|
|
|
|
# executing two macros that stop automatically at the same time
|
|
|
|
|
|
|
|
code_1 = 61
|
|
|
|
code_2 = 62
|
|
|
|
system_mapping.clear()
|
|
|
|
system_mapping._set('1', code_1)
|
|
|
|
system_mapping._set('2', code_2)
|
|
|
|
|
|
|
|
# try two concurrent macros with D-Pad events because they are
|
|
|
|
# more difficult to manage, since their only difference is their
|
|
|
|
# value, and one of them is negative.
|
|
|
|
down_1 = (EV_ABS, ABS_HAT0X, 1)
|
|
|
|
down_2 = (EV_ABS, ABS_HAT0X, -1)
|
|
|
|
|
|
|
|
repeats = 10
|
|
|
|
|
|
|
|
macro_mapping = {
|
2020-12-19 20:50:27 +00:00
|
|
|
down_1: parse(f'r({repeats}, k(1))', self.mapping),
|
|
|
|
down_2: parse(f'r({repeats}, k(2))', self.mapping)
|
2020-12-19 15:04:07 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
history = []
|
|
|
|
|
|
|
|
def handler(*args):
|
|
|
|
history.append(args)
|
|
|
|
|
|
|
|
macro_mapping[down_1].set_handler(handler)
|
|
|
|
macro_mapping[down_2].set_handler(handler)
|
|
|
|
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(*down_1), None)
|
|
|
|
handle_keycode({}, macro_mapping, InputEvent(*down_2), None)
|
|
|
|
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
sleeptime = config.get('macros.keystroke_sleep_ms') / 1000
|
|
|
|
loop.run_until_complete(asyncio.sleep(1.1 * repeats * 2 * sleeptime))
|
|
|
|
|
|
|
|
self.assertEqual(len(history), repeats * 4)
|
|
|
|
|
|
|
|
self.assertEqual(history.count((code_1, 1)), 10)
|
|
|
|
self.assertEqual(history.count((code_1, 0)), 10)
|
|
|
|
self.assertEqual(history.count((code_2, 1)), 10)
|
|
|
|
self.assertEqual(history.count((code_2, 0)), 10)
|
|
|
|
|
|
|
|
def test_normalize(self):
|
|
|
|
# -1234 to -1, 5678 to 1, 0 to 0
|
|
|
|
|
|
|
|
key_1 = (EV_KEY, BTN_TL)
|
|
|
|
ev_1 = (*key_1, 5678)
|
|
|
|
ev_2 = (*key_1, 0)
|
|
|
|
|
|
|
|
# doesn't really matter if it makes sense, the A key reports
|
|
|
|
# negative values now.
|
|
|
|
key_2 = (EV_KEY, KEY_A)
|
|
|
|
ev_3 = (*key_2, -1234)
|
|
|
|
ev_4 = (*key_2, 0)
|
|
|
|
|
|
|
|
_key_to_code = {
|
|
|
|
(*key_1, 1): 41,
|
|
|
|
(*key_2, -1): 42
|
|
|
|
}
|
|
|
|
|
|
|
|
uinput = UInput()
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_1), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_2), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_3), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_4), uinput)
|
|
|
|
|
|
|
|
self.assertEqual(len(uinput_write_history), 4)
|
|
|
|
self.assertEqual(uinput_write_history[0].t, (EV_KEY, 41, 1))
|
|
|
|
self.assertEqual(uinput_write_history[1].t, (EV_KEY, 41, 0))
|
|
|
|
self.assertEqual(uinput_write_history[2].t, (EV_KEY, 42, 1))
|
|
|
|
self.assertEqual(uinput_write_history[3].t, (EV_KEY, 42, 0))
|
|
|
|
|
|
|
|
def test_filter_trigger_spam(self):
|
|
|
|
trigger = (EV_KEY, BTN_TL)
|
|
|
|
|
|
|
|
_key_to_code = {
|
|
|
|
(*trigger, 1): 51,
|
|
|
|
(*trigger, -1): 52
|
|
|
|
}
|
|
|
|
|
|
|
|
uinput = UInput()
|
|
|
|
|
|
|
|
"""positive"""
|
|
|
|
|
|
|
|
for i in range(1, 20):
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*trigger, i), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*trigger, 0), uinput)
|
|
|
|
|
|
|
|
self.assertEqual(len(uinput_write_history), 2)
|
|
|
|
|
|
|
|
"""negative"""
|
|
|
|
|
|
|
|
for i in range(1, 20):
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*trigger, -i), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*trigger, 0), uinput)
|
|
|
|
|
|
|
|
self.assertEqual(len(uinput_write_history), 4)
|
|
|
|
self.assertEqual(uinput_write_history[0].t, (EV_KEY, 51, 1))
|
|
|
|
self.assertEqual(uinput_write_history[1].t, (EV_KEY, 51, 0))
|
|
|
|
self.assertEqual(uinput_write_history[2].t, (EV_KEY, 52, 1))
|
|
|
|
self.assertEqual(uinput_write_history[3].t, (EV_KEY, 52, 0))
|
|
|
|
|
|
|
|
def test_ignore_hold(self):
|
|
|
|
key = (EV_KEY, KEY_A)
|
|
|
|
ev_1 = (*key, 1)
|
|
|
|
ev_2 = (*key, 2)
|
|
|
|
ev_3 = (*key, 0)
|
|
|
|
|
|
|
|
_key_to_code = {
|
|
|
|
(*key, 1): 21,
|
|
|
|
}
|
|
|
|
|
|
|
|
uinput = UInput()
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_1), uinput)
|
|
|
|
for _ in range(10):
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_2), uinput)
|
|
|
|
handle_keycode(_key_to_code, {}, InputEvent(*ev_3), uinput)
|
|
|
|
|
|
|
|
self.assertEqual(len(uinput_write_history), 2)
|
|
|
|
self.assertEqual(uinput_write_history[0].t, (EV_KEY, 21, 1))
|
|
|
|
self.assertEqual(uinput_write_history[1].t, (EV_KEY, 21, 0))
|
|
|
|
# linux will generate events with value 2 after key-mapper injected
|
|
|
|
# the key-press
|
|
|
|
|
2020-12-06 11:54:14 +00:00
|
|
|
|
2020-12-03 19:03:53 +00:00
|
|
|
if __name__ == "__main__":
|
|
|
|
unittest.main()
|