Add native Linux controls and tray menu for BTD 700

tests/test_protocol.py

@@ -0,0 +1,63 @@
+import tempfile
+import unittest
+from pathlib import Path
+from btd700.protocol import Command as C, ProtocolError, acknowledge, decode, encode, validate_text
+from btd700.transport import discover
+
+
+class ProtocolTests(unittest.TestCase):
+    def test_captured_status_response(self):
+        message = decode(bytes.fromhex('34ff0103010301'))
+        self.assertEqual(message.payload, b'\1\3\1')
+        self.assertEqual(message.command, C.GET_MODE)
+
+    def test_captured_get_state_request(self):
+        self.assertEqual(encode(C.GET_STATE), bytes.fromhex('34fe0600') + bytes(60))
+
+    def test_all_non_control_command_numbers_rejected(self):
+        for value in range(256):
+            if value not in set(C):
+                with self.assertRaises(ValueError):
+                    encode(value)
+
+    def test_setter_validation(self):
+        for command, args in ((C.SET_MODE, b'\3\3'), (C.SET_MODE, b'\0\0'),
+                              (C.SET_CODEC, b'\3'), (C.SET_CONNECTION, b'\2'),
+                              (C.SET_BROADCAST, b'\1\3\0'), (C.GET_STATE, b'\1'),
+                              (C.SET_KEY, b'x' * 17), (C.FACTORY_RESET, b'\1')):
+            with self.assertRaises(ValueError):
+                encode(command, args)
+
+    def test_malformed_and_unrelated_reports(self):
+        for report in (b'\x34', b'\x34\xff\1', bytes.fromhex('34ff013d'), bytes.fromhex('34ff0602ff'), bytes.fromhex('34000600')):
+            with self.assertRaises(ProtocolError):
+                decode(report)
+        self.assertIsNone(decode(b'\1\0'))
+        self.assertIsNone(decode(b''))
+
+    def test_event_ack(self):
+        self.assertEqual(acknowledge(15), bytes.fromhex('34fd0f00') + bytes(60))
+        with self.assertRaises(ProtocolError):
+            acknowledge(255)
+
+    def test_name_and_key_bounds(self):
+        self.assertEqual(validate_text('Studio Audio'), b'Studio Audio')
+        self.assertEqual(validate_text(''), b'')
+        self.assertEqual(validate_text('A' * 16, key=True), b'A' * 16)
+        for name in ('Hi', 'x' * 17, ' Musik', 'Musik ', 'Müsik', 'abc\0def'):
+            with self.assertRaises(ValueError):
+                validate_text(name)
+        with self.assertRaises(ValueError):
+            validate_text('abc def', key=True)
+
+    def test_discovery_excludes_updater_and_other_devices(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp)
+            for number, vendor, descriptor in ((0, '3542', b'\x06\xa2\xff\x85\x34'),
+                                               (1, '3542', b'\x06\x00\xff\x85\x05'),
+                                               (2, '046D', b'\x06\xa2\xff\x85\x34')):
+                node = root / f'hidraw{number}' / 'device'
+                node.mkdir(parents=True)
+                (node / 'uevent').write_text(f'HID_ID=0003:0000{vendor}:00003001\nHID_UNIQ=test\n')
+                (node / 'report_descriptor').write_bytes(descriptor)
+            self.assertEqual([d.path for d in discover(root)], ['/dev/hidraw0'])