tests: add tests for INFO FW_INFO function

Signed-off-by: Filipe Laíns <[email protected]>

Author: FFY00

tests/conftest.py

@@ -1,13 +1,39 @@
 # SPDX-License-Identifier: MIT
 # SPDX-FileCopyrightText: 2021 Filipe Laíns <[email protected]>
 
+import contextlib
+import importlib.util
+import os
+import os.path
 import unittest.mock
 
 import pages
 import pytest
 import testsuite
 
 
+root_dir = os.path.abspath(os.path.join(__file__, '..', '..'))
+
+
+@contextlib.contextmanager
+def cd(*path_paths):
+    old_cwd = os.getcwd()
+    os.chdir(os.path.join(*path_paths))
+    try:
+        yield
+    finally:
+        os.chdir(old_cwd)
+
+
+def import_file(name, path):
+    spec = importlib.util.spec_from_file_location(name, path)
+    if not spec.loader:
+        raise ImportError(f'Unable to import `{path}`: no loader')
+    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)  # type: ignore
+    return module
+
+
 @pytest.fixture()
 def device(request):
     return testsuite.Device(
@@ -29,3 +55,12 @@ def basic_device():
     )
     device.hid_send = unittest.mock.MagicMock()
     return device
+
+
+@pytest.fixture()
+def fw_version():
+    with cd(root_dir):
+        configure = import_file('configure', 'configure.py')
+        version = configure.BuildSystemBuilder.calculate_version()
+        is_dirty = configure.BuildSystemBuilder._is_git_dirty()
+    return f'{version}.dirty' if is_dirty else version

tests/test_protocol.py

@@ -8,3 +8,35 @@ def test_dispatch(basic_device):
     basic_device.protocol_dispatch([0x21])  # invalid length long
 
     basic_device.hid_send.assert_not_called()
+
+
+def test_fw_info_vendor(basic_device):
+    basic_device.protocol_dispatch([0x20, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00])
+
+    basic_device.hid_send.assert_called_with(
+        [0x21, 0x00, 0x01] + list(b'openinput-git') + [0x00] * 16
+    )
+
+
+def test_fw_info_version(basic_device, fw_version):
+    basic_device.protocol_dispatch([0x20, 0x00, 0x01, 0x01, 0x00, 0x00, 0x00, 0x00])
+
+    basic_device.hid_send.assert_called_with(
+        [0x21, 0x00, 0x01] + list(fw_version.encode('ascii')) + [0x00] * (29 - len(fw_version))
+    )
+
+
+def test_fw_info_device_name(basic_device):
+    basic_device.protocol_dispatch([0x20, 0x00, 0x01, 0x02, 0x00, 0x00, 0x00, 0x00])
+
+    basic_device.hid_send.assert_called_with(
+        [0x21, 0x00, 0x01] + list(b'basic test device') + [0x00] * 12
+    )
+
+
+def test_fw_info_unsupported(basic_device):
+    basic_device.protocol_dispatch([0x20, 0x00, 0x01, 0xFF, 0x00, 0x00, 0x00, 0x00])
+
+    basic_device.hid_send.assert_called_with(
+        [0x21, 0xFF, 0x01, 0x00, 0x01] + [0x00] * 27
+    )