Tests: add tests for sign

test/python/test_instructions.py

@@ -12,6 +12,7 @@
     get_setup_app_context_instructions,
     get_reset_app_context_instructions
 )
+from utils.message import RawMessage
 
 TESTS_ROOT_DIR = Path(__file__).parent
 
@@ -251,3 +252,30 @@ def test_get_all_hwm(
 
     hwm = client.get_all_hwm()
     checker.check_all_hwm(False, main_hwm, None, test_hwm, None, chain, hwm)
+
+
+@pytest.mark.parametrize("account", [DEFAULT_ACCOUNT])
+@pytest.mark.parametrize("with_hash", [False, True])
+def test_sign_message(
+        account: Account,
+        with_hash: bool,
+        client: TezosClient,
+        checker: TezosAPDUChecker,
+        firmware: Firmware,
+        navigator: Navigator) -> None:
+    """Test the SIGN(_WITH_HASH) instruction."""
+
+    chain: int = 0
+    main_hwm: int = 0
+    test_hwm: int = 0
+
+    instructions = get_setup_app_context_instructions(firmware)
+
+    with client.setup_app_context(account, chain, main_hwm, test_hwm):
+        navigator.navigate(instructions)
+
+    tx = RawMessage("0270000007000000000000000000000000000000000000000000000000000000000000000014000000000003000000010000000000000000000000000000000000000000000000000000000000000000")
+    with client.sign_message(account, tx, with_hash):
+        pass
+    data = client.get_async_response()
+    checker.check_signature(account, tx, with_hash, data)

test/python/utils/client.py

@@ -208,16 +208,19 @@ def get_all_hwm(self) -> bytes:
     @contextmanager
     def sign_message(self,
                      account: Account,
-                     message: Message) -> Generator[None, None, None]:
-        """Send the SIGN instruction."""
+                     message: Message,
+                     with_hash: bool = False) -> Generator[None, None, None]:
+        """Send the SIGN(_WITH_HASH) instruction."""
+
+        ins = Ins.SIGN_WITH_HASH if with_hash else Ins.SIGN
 
         self._exchange(
-            ins=Ins.SIGN,
+            ins=ins,
             sig_scheme=account.sig_scheme,
             payload=account.path)
 
         with self._exchange_async(
-                ins=Ins.SIGN,
+                ins=ins,
                 index=Index.LAST,
                 payload=message.raw):
             yield