Add test for 'confirm_parameter_selector'

tests/ragger/test_blind_sign.py

@@ -3,15 +3,21 @@ import json
 import pytest
 from web3 import Web3
 
-from client.client import EthAppClient, StatusWord
-
 from ragger.backend import BackendInterface
 from ragger.firmware import Firmware
 from ragger.navigator import Navigator, NavInsID
+from ragger.navigator.navigation_scenario import NavigateWithScenario
 from ragger.error import ExceptionRAPDU
 
 from constants import ABIS_FOLDER
 
+from client.client import EthAppClient, StatusWord
+from client.settings import SettingID, settings_toggle
+import client.response_parser as ResponseParser
+from client.utils import recover_transaction
+
+
+BIP32_PATH = "m/44'/60'/0'/0/0"
 
 # Token approval, would require loading the "internal plugin" &
 # providing the token metadata from the CAL
@@ -42,7 +48,7 @@ def test_blind_sign(firmware: Firmware,
         "chainId": 1
     }
     with pytest.raises(ExceptionRAPDU) as e:
-        with app_client.sign("m/44'/60'/0'/0/0", tx_params):
+        with app_client.sign(BIP32_PATH, tx_params):
             pass
     assert e.value.status == StatusWord.INVALID_DATA
 
@@ -56,3 +62,77 @@ def test_blind_sign(firmware: Firmware,
     navigator.navigate_and_compare(default_screenshot_path,
                                    "blind-signed_approval",
                                    moves)
+
+
+# Token approval, would require loading the "internal plugin" &
+# providing the token metadata from the CAL
+def test_sign_parameter_selector(firmware: Firmware,
+                                 backend: BackendInterface,
+                                 navigator: Navigator,
+                                 scenario_navigator: NavigateWithScenario,
+                                 test_name: str,
+                                 default_screenshot_path: Path):
+    app_client = EthAppClient(backend)
+
+    with app_client.get_public_addr(bip32_path=BIP32_PATH, display=False):
+        pass
+    _, DEVICE_ADDR, _ = ResponseParser.pk_addr(app_client.response().data)
+
+    with open(f"{ABIS_FOLDER}/erc20.json", encoding="utf-8") as file:
+        abi = json.load(file)
+
+    contract_name = "approve"
+    count = 0
+    for elt in abi:
+        if elt["name"] == contract_name:
+            count = len(elt["inputs"])
+            break
+    assert count == 2, "Invalid inputs number"
+    tx_params = {
+        "nonce": 235,
+        "maxFeePerGas": Web3.to_wei(100, "gwei"),
+        "maxPriorityFeePerGas": Web3.to_wei(10, "gwei"),
+        "gas": 44001,
+        # Maker: Dai Stablecoin
+        "to": bytes.fromhex("6b175474e89094c44da98b954eedeac495271d0f"),
+        "data": Web3().eth.contract(abi=abi).encodeABI(contract_name, [
+            # Uniswap Protocol: Permit2
+            bytes.fromhex("000000000022d473030f116ddee9f6b43ac78ba3"),
+            Web3.to_wei("2", "ether")
+        ]),
+        "chainId": 1
+    }
+
+    settings_toggle(firmware, navigator, [SettingID.DEBUG_DATA, SettingID.BLIND_SIGNING])
+
+    with app_client.sign(BIP32_PATH, tx_params):
+        if firmware.device.startswith("nano"):
+            end_text = "Approve"
+            nav_inst = NavInsID.RIGHT_CLICK
+            valid_instr = [NavInsID.BOTH_CLICK]
+        else:
+            end_text = "Confirm"
+            nav_inst = NavInsID.USE_CASE_REVIEW_TAP
+            valid_instr = [NavInsID.USE_CASE_REVIEW_CONFIRM]
+
+        # Loop for "Selector" + the contract inputs
+        for step in range(count + 1):
+            navigator.navigate_until_text_and_compare(nav_inst,
+                                                      valid_instr,
+                                                      end_text,
+                                                      default_screenshot_path,
+                                                      f"{test_name}/step_{step}",
+                                                      screen_change_after_last_instruction=False)
+            step +=1
+
+        # Transaction review
+        if firmware.device.startswith("nano"):
+            end_text = "Accept"
+        else:
+            end_text = "Sign"
+        scenario_navigator.review_approve(default_screenshot_path, f"{test_name}/step_{step}", end_text)
+
+    # verify signature
+    vrs = ResponseParser.signature(app_client.response().data)
+    addr = recover_transaction(tx_params, vrs)
+    assert addr == DEVICE_ADDR