Solaar/tests/logitech_receiver/test_diversion.py

120 lines
3.3 KiB
Python

import textwrap
from unittest import mock
from unittest.mock import mock_open
import pytest
from logitech_receiver import diversion
from logitech_receiver.base import HIDPPNotification
from logitech_receiver.hidpp20_constants import FEATURE
@pytest.fixture
def rule_config():
rule_content = """
%YAML 1.3
---
- MouseGesture: Mouse Left
- KeyPress:
- [Control_L, Alt_L, Left]
- click
...
---
- MouseGesture: Mouse Up
- KeyPress:
- [Super_L, Up]
- click
...
---
- Test: [thumb_wheel_up, 10]
- KeyPress:
- [Control_L, Page_Down]
- click
...
---
"""
return textwrap.dedent(rule_content)
def test_load_rule_config(rule_config):
expected_rules = [
[
diversion.MouseGesture,
diversion.KeyPress,
],
[diversion.MouseGesture, diversion.KeyPress],
[diversion.Test, diversion.KeyPress],
]
with mock.patch("builtins.open", new=mock_open(read_data=rule_config)):
loaded_rules = diversion._load_rule_config(file_path=mock.Mock())
assert len(loaded_rules.components) == 2 # predefined and user configured rules
user_configured_rules = loaded_rules.components[0]
assert isinstance(user_configured_rules, diversion.Rule)
for components, expected_components in zip(user_configured_rules.components, expected_rules):
for component, expected_component in zip(components.components, expected_components):
assert isinstance(component, expected_component)
def test_diversion_rule():
args = [
{
"Rule": [ # Implement problematic keys for Craft and MX Master
{"Rule": [{"Key": ["Brightness Down", "pressed"]}, {"KeyPress": "XF86_MonBrightnessDown"}]},
{"Rule": [{"Key": ["Brightness Up", "pressed"]}, {"KeyPress": "XF86_MonBrightnessUp"}]},
]
},
]
rule = diversion.Rule(args)
assert len(rule.components) == 1
root_rule = rule.components[0]
assert isinstance(root_rule, diversion.Rule)
assert len(root_rule.components) == 2
for component in root_rule.components:
assert isinstance(component, diversion.Rule)
assert len(component.components) == 2
key = component.components[0]
assert isinstance(key, diversion.Key)
key = component.components[1]
assert isinstance(key, diversion.KeyPress)
def test_key_is_down():
result = diversion.key_is_down(key=diversion.CONTROL.G2)
assert result is False
@pytest.mark.parametrize(
"feature, data",
[
(
FEATURE.REPROG_CONTROLS_V4,
[0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08],
),
(FEATURE.GKEY, [0x01, 0x02, 0x03, 0x04]),
(FEATURE.MKEYS, [0x01, 0x02, 0x03, 0x04]),
(FEATURE.MR, [0x01, 0x02, 0x03, 0x04]),
(FEATURE.THUMB_WHEEL, [0x01, 0x02, 0x03, 0x04, 0x05]),
(FEATURE.DEVICE_UNIT_ID, [0x01, 0x02, 0x03, 0x04, 0x05]),
],
)
def test_process_notification(feature, data):
device_mock = mock.Mock()
notification = HIDPPNotification(
report_id=0x01,
devnumber=1,
sub_id=0x13,
address=0x00,
data=bytes(data),
)
diversion.process_notification(device_mock, notification, feature)