From cd5383135a73ad7461ab848b6af32f236ad59ffa Mon Sep 17 00:00:00 2001
From: Ilya Zhuravlev <whatever@xyz.is>
Date: Wed, 24 Mar 2021 08:28:06 -0400
Subject: [PATCH] fix unittests

---
 src/main/python/test/test_keyboard.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/main/python/test/test_keyboard.py b/src/main/python/test/test_keyboard.py
index fd4bdf2..876920a 100644
--- a/src/main/python/test/test_keyboard.py
+++ b/src/main/python/test/test_keyboard.py
@@ -31,6 +31,9 @@ class SimulatedDevice:
         out += b"\x00" * (MSG_LEN - len(out))
         self.expect_data.append((inp, out))
 
+    def expect_via_protocol(self, via_protocol):
+        self.expect("01", struct.pack("BB", 1, via_protocol))
+
     def expect_keyboard_id(self, kbid):
         self.expect("FE00", struct.pack("<IQ", 0, kbid))
 
@@ -93,6 +96,7 @@ class TestKeyboard(unittest.TestCase):
     @staticmethod
     def prepare_keyboard(layout, keymap, encoders=None):
         dev = SimulatedDevice()
+        dev.expect_via_protocol(9)
         dev.expect_keyboard_id(0)
         dev.expect_layout(layout)
         dev.expect_layers(len(keymap))