Update navigation for Stax

LedgerHQ · Sep 20, 2024 · b06b920 · b06b920
1 parent f0db13f
commit b06b920
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/functional/test_sign_cmd.py b/tests/functional/test_sign_cmd.py
@@ -90,7 +90,7 @@ def test_sign_transaction_newaccount_accepted(test_name, firmware, backend, navi
     messages = split_message(payload, MAX_CHUNK_SIZE)
     assert len(messages) == 2
 
-    if firmware.device.startswith("nano"):
+    if firmware.is_nano:
         instructions = get_nano_review_instructions(2) + get_nano_review_instructions(7)
     else:
         instructions = [NavInsID.USE_CASE_REVIEW_TAP] * 5
@@ -99,10 +99,13 @@ def test_sign_transaction_newaccount_accepted(test_name, firmware, backend, navi
                                        test_name + "/part1",
                                        instructions)
 
-    if firmware.device.startswith("nano"):
+    if firmware.is_nano:
         instructions = get_nano_review_instructions(6) + get_nano_review_instructions(8)
     else:
-        instructions = [NavInsID.USE_CASE_REVIEW_TAP] * 6
+        if firmware == Firmware.FLEX:
+            instructions = [NavInsID.USE_CASE_REVIEW_TAP] * 6
+        else:
+            instructions = [NavInsID.USE_CASE_REVIEW_TAP] * 5
         instructions.append(NavInsID.USE_CASE_REVIEW_CONFIRM)
         instructions.append(NavInsID.USE_CASE_STATUS_DISMISS)
     with client.send_async_sign_message_full(messages[1], False):