navigator.py: Add use case navigation function

LedgerHQ · Apr 4, 2024 · fd05dfc · fd05dfc
1 parent 987c599
commit fd05dfc
Showing 1 changed file with 105 additions and 0 deletions.
diff --git a/src/ragger/navigator/navigator.py b/src/ragger/navigator/navigator.py
@@ -573,3 +573,108 @@ def navigate_until_text(self,
                                              None, None, timeout,
                                              screen_change_before_first_instruction,
                                              screen_change_after_last_instruction)
+
+    def review_approve(self,
+                       path: Optional[Path] = None,
+                       test_case_name: Optional[Path] = None,
+                       approve_screen_text: Optional[str] = None):
+        if self._firmware.device.startswith("nano"):
+            navigate_instruction = NavInsID.RIGHT_CLICK
+            validation_instructions = [NavInsID.BOTH_CLICK]
+            if not approve_screen_text:
+                approve_screen_text = "Approve"
+
+        elif self._firmware.device.startswith("stax"):
+            navigate_instruction = NavInsID.USE_CASE_REVIEW_TAP
+            validation_instructions = [NavInsID.USE_CASE_REVIEW_CONFIRM,
+                                       NavInsID.USE_CASE_STATUS_DISMISS]
+            if not approve_screen_text:
+                approve_screen_text = "Hold to sign"
+
+        else:
+            raise NotImplementedError("Device not supported")
+
+        self.navigate_until_text_and_compare(navigate_instruction,
+                                             validation_instructions,
+                                             approve_screen_text,
+                                             path,
+                                             test_case_name)
+
+    def review_reject(self,
+                      path: Optional[Path] = None,
+                      test_case_name: Optional[Path] = None,
+                      reject_screen_text: Optional[str] = None):
+        if self._firmware.device.startswith("nano"):
+            navigate_instruction = NavInsID.RIGHT_CLICK
+            validation_instructions = [NavInsID.BOTH_CLICK]
+
+            if not reject_screen_text:
+                reject_screen_text = "Reject"
+
+        elif self._firmware.device.startswith("stax"):
+            navigate_instruction = NavInsID.USE_CASE_REVIEW_TAP
+            validation_instructions = [NavInsID.USE_CASE_REVIEW_REJECT,
+                                       NavInsID.USE_CASE_CHOICE_CONFIRM,
+                                       NavInsID.USE_CASE_STATUS_DISMISS]
+            if not reject_screen_text:
+                reject_screen_text = "Hold to sign"
+        else:
+            raise NotImplementedError("Device not supported")
+
+        self.navigate_until_text_and_compare(navigate_instruction,
+                                             validation_instructions,
+                                             reject_screen_text,
+                                             path,
+                                             test_case_name)
+
+    def address_review_approve(self,
+                               path: Optional[Path] = None,
+                               test_case_name: Optional[Path] = None,
+                               approve_screen_text: Optional[str] = None):
+        if self._firmware.device.startswith("nano"):
+            navigate_instruction = NavInsID.RIGHT_CLICK
+            validation_instructions = [NavInsID.BOTH_CLICK]
+            if not approve_screen_text:
+                approve_screen_text = "Approve"
+
+        elif self._firmware.device.startswith("stax"):
+            navigate_instruction = NavInsID.USE_CASE_REVIEW_TAP
+            validation_instructions = [NavInsID.USE_CASE_ADDRESS_CONFIRMATION_CONFIRM,
+                                       NavInsID.USE_CASE_STATUS_DISMISS]
+            if not approve_screen_text:
+                approve_screen_text = "Confirm"
+
+        else:
+            raise NotImplementedError("Device not supported")
+
+        self.navigate_until_text_and_compare(navigate_instruction,
+                                             validation_instructions,
+                                             approve_screen_text,
+                                             path,
+                                             test_case_name)
+
+    def address_review_reject(self,
+                              path: Optional[Path] = None,
+                              test_case_name: Optional[Path] = None,
+                              reject_screen_text: Optional[str] = None):
+        if self._firmware.device.startswith("nano"):
+            navigate_instruction = NavInsID.RIGHT_CLICK
+            validation_instructions = [NavInsID.BOTH_CLICK]
+
+            if not reject_screen_text:
+                reject_screen_text = "Reject"
+
+        elif self._firmware.device.startswith("stax"):
+            navigate_instruction = NavInsID.USE_CASE_REVIEW_TAP
+            validation_instructions = [NavInsID.USE_CASE_ADDRESS_CONFIRMATION_CANCEL,
+                                       NavInsID.USE_CASE_STATUS_DISMISS]
+            if not reject_screen_text:
+                reject_screen_text = "Confirm"
+        else:
+            raise NotImplementedError("Device not supported")
+
+        self.navigate_until_text_and_compare(navigate_instruction,
+                                             validation_instructions,
+                                             reject_screen_text,
+                                             path,
+                                             test_case_name)