steven-y-e
/
vial
1
0
Fork 0
Code Pull Requests Releases Activity

fix unittests

main
Ilya Zhuravlev 2020-12-02 10:17:52 -05:00
parent 2e0de37b77
commit 74ed115c16
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/python/test/test_keyboard.py
View File

@ -31,6 +31,9 @@ class SimulatedDevice:
out = bytes.fromhex(out) out = bytes.fromhex(out)
self.expect_data.append((inp, out)) self.expect_data.append((inp, out))
def expect_keyboard_id(self, kbid):
self.expect("FE00", struct.pack("<IQ", 0, kbid))
def expect_layout(self, layout): def expect_layout(self, layout):
compressed = lzma.compress(layout.encode("utf-8")) compressed = lzma.compress(layout.encode("utf-8"))
self.expect("FE01", struct.pack("<I", len(compressed))) self.expect("FE01", struct.pack("<I", len(compressed)))
@ -61,8 +64,8 @@ class SimulatedDevice:
if data != inp: if data != inp:
raise Exception("Got unexpected data at index {}: expected={} got={}".format( raise Exception("Got unexpected data at index {}: expected={} got={}".format(
dev.expect_idx, dev.expect_idx,
data.hex(), inp.hex(),
inp.hex() data.hex()
)) ))
dev.expect_idx += 1 dev.expect_idx += 1
return out return out
@ -79,6 +82,7 @@ class TestKeyboard(unittest.TestCase):
@staticmethod @staticmethod
def prepare_keyboard(layout, keymap): def prepare_keyboard(layout, keymap):
dev = SimulatedDevice() dev = SimulatedDevice()
dev.expect_keyboard_id(0)
dev.expect_layout(layout) dev.expect_layout(layout)
dev.expect_layers(len(keymap)) dev.expect_layers(len(keymap))
dev.expect_keymap(keymap) dev.expect_keymap(keymap)