test: FLEX & STAX tests OK on test_get_legacy_public_key_confirm_acce…

…pted
blooo-io · Dec 6, 2024 · f778bf9 · f778bf9
1 parent 7f5a001
commit f778bf9
Show file tree

Hide file tree

Showing 23 changed files with 22 additions and 7 deletions.
diff --git a/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00000.png b/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00000.png
diff --git a/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00001.png b/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00001.png
diff --git a/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00002.png b/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00002.png
diff --git a/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00003.png b/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00003.png
diff --git a/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00004.png b/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00004.png
diff --git a/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00005.png b/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00005.png
diff --git a/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00006.png b/tests/snapshots/flex/test_get_legacy_public_key_confirm_accepted/00006.png
diff --git a/tests/snapshots/stax/test_app_mainmenu/00000.png b/tests/snapshots/stax/test_app_mainmenu/00000.png
diff --git a/tests/snapshots/stax/test_app_mainmenu/00001.png b/tests/snapshots/stax/test_app_mainmenu/00001.png
diff --git a/tests/snapshots/stax/test_app_mainmenu/00002.png b/tests/snapshots/stax/test_app_mainmenu/00002.png
diff --git a/tests/snapshots/stax/test_app_mainmenu/00003.png b/tests/snapshots/stax/test_app_mainmenu/00003.png
diff --git a/tests/snapshots/stax/test_app_mainmenu/00004.png b/tests/snapshots/stax/test_app_mainmenu/00004.png
diff --git a/tests/snapshots/stax/test_app_mainmenu/00005.png b/tests/snapshots/stax/test_app_mainmenu/00005.png
diff --git a/tests/snapshots/stax/test_app_mainmenu/00006.png b/tests/snapshots/stax/test_app_mainmenu/00006.png
diff --git a/tests/snapshots/stax/test_app_mainmenu/00007.png b/tests/snapshots/stax/test_app_mainmenu/00007.png
diff --git a/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00000.png b/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00000.png
diff --git a/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00001.png b/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00001.png
diff --git a/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00002.png b/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00002.png
diff --git a/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00003.png b/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00003.png
diff --git a/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00004.png b/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00004.png
diff --git a/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00005.png b/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00005.png
diff --git a/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00006.png b/tests/snapshots/stax/test_get_legacy_public_key_confirm_accepted/00006.png
diff --git a/tests/test_pubkey_cmd.py b/tests/test_pubkey_cmd.py
@@ -10,6 +10,7 @@
 from ragger.bip import calculate_public_key_and_chaincode, CurveChoice
 from ragger.error import ExceptionRAPDU
 from ragger.navigator import NavInsID, NavIns
+from ragger.firmware import Firmware
 from utils import navigate_until_text_and_compare
 
 
@@ -33,6 +34,15 @@
     NavInsID.BOTH_CLICK,
 ]
 
+flex_stax_accept_instructions = [
+    NavInsID.WAIT_FOR_SCREEN_CHANGE,
+    NavInsID.SWIPE_CENTER_TO_LEFT,
+    NavInsID.USE_CASE_CHOICE_CONFIRM,
+    NavInsID.SWIPE_CENTER_TO_LEFT,
+    NavInsID.USE_CASE_CHOICE_CONFIRM,
+    NavInsID.WAIT_FOR_HOME_SCREEN,
+]
+
 
 # In this test we check that the GET_PUBLIC_KEY works in confirmation mode
 @pytest.mark.active_test_scope
@@ -41,14 +51,19 @@ def test_get_legacy_public_key_confirm_accepted(
 ):
     client = BoilerplateCommandSender(backend)
     path = "m/1105/0/0/0/0/2/0/0"
+    if firmware.is_nano:
+        instructions = nano_accept_instructions
+    else:
+        instructions = flex_stax_accept_instructions
+
     with client.get_public_key_with_confirmation(path=path):
-        if firmware.is_nano:
-            navigator.navigate_and_compare(
-                default_screenshot_path,
-                test_name,
-                nano_accept_instructions,
-                screen_change_before_first_instruction=False,
-            )
+
+        navigator.navigate_and_compare(
+            default_screenshot_path,
+            test_name,
+            instructions,
+            screen_change_before_first_instruction=False,
+        )
 
     response = client.get_async_response().data
     print("km------------------|response:", response.hex())