Test flow order

test/python/test_flow_order.py

@@ -0,0 +1,125 @@
+import pytest
+
+from typing import List
+from ragger.utils import RAPDU, prefix_with_len, create_currency_config
+from ragger.error import ExceptionRAPDU
+
+from .apps.exchange import ExchangeClient, Rate, SubCommand, Errors, Command, P2_EXTEND, P2_MORE, EXCHANGE_CLASS
+from .apps.exchange_transaction_builder import get_partner_curve, craft_tx, encode_tx, extract_payout_ticker, extract_refund_ticker, LEGACY_SUBCOMMANDS, ALL_SUBCOMMANDS, NEW_SUBCOMMANDS, craft_transaction_proposal
+from .apps.signing_authority import SigningAuthority, LEDGER_SIGNER
+from .apps import cal as cal
+
+# Some valid infos for TX. Content is irrelevant for the test
+
+SWAP_TX_INFOS = {
+     "payin_address": b"0xd692Cb1346262F584D17B4B470954501f6715a82",
+     "payin_extra_id": b"",
+     "refund_address": b"0xDad77910DbDFdE764fC21FCD4E74D71bBACA6D8D",
+     "refund_extra_id": b"",
+     "payout_address": b"bc1qer57ma0fzhqys2cmydhuj9cprf9eg0nw922a8j",
+     "payout_extra_id": b"",
+     "currency_from": "ETH",
+     "currency_to": "BTC",
+     "amount_to_provider": bytes.fromhex("013fc3a717fb5000"),
+     "amount_to_wallet": b"\x0b\xeb\xc2\x00",
+}
+FUND_TX_INFOS = {
+    "user_id": "John Wick",
+    "account_name": "Remember Daisy",
+    "in_currency": "ETH",
+    "in_amount": b"\032\200\250]$T\000",
+    "in_address": "0x252fb4acbe0de4f0bd2409a5ed59a71e4ef1d2bc"
+}
+SELL_TX_INFOS = {
+    "trader_email": "[email protected]",
+    "out_currency": "USD",
+    "out_amount": {"coefficient": b"\x01", "exponent": 3},
+    "in_currency": "ETH",
+    "in_amount": b"\032\200\250]$T\000",
+    "in_address": "0x252fb4acbe0de4f0bd2409a5ed59a71e4ef1d2bc"
+}
+TX_INFOS = {
+    SubCommand.SWAP: SWAP_TX_INFOS,
+    SubCommand.SWAP_NG: SWAP_TX_INFOS,
+    SubCommand.FUND: FUND_TX_INFOS,
+    SubCommand.FUND_NG: FUND_TX_INFOS,
+    SubCommand.SELL: SELL_TX_INFOS,
+    SubCommand.SELL_NG: SELL_TX_INFOS,
+}
+FEES = 100
+
+def all_commands_for_subcommand_except(s: SubCommand, c: Command) -> List[Command]:
+    ret = [Command.SET_PARTNER_KEY,
+           Command.CHECK_PARTNER,
+           Command.PROCESS_TRANSACTION_RESPONSE,
+           Command.CHECK_TRANSACTION_SIGNATURE,
+           Command.START_SIGNING_TRANSACTION]
+    if s == SubCommand.SWAP or s == SubCommand.SWAP_NG:
+        ret += [Command.CHECK_PAYOUT_ADDRESS, Command.CHECK_REFUND_ADDRESS]
+    elif s == SubCommand.FUND or s == SubCommand.SELL:
+        ret += [Command.CHECK_ASSET_IN_LEGACY]
+    else:
+        ret += [Command.CHECK_ASSET_IN]
+
+    if c in ret:
+        ret.remove(c)
+
+    return ret
+
+def try_all_commands_for_subcommand_except(ex: ExchangeClient, s: SubCommand, c: Command):
+    commands = all_commands_for_subcommand_except(s, c)
+    for c in commands:
+        print(f"Sending {c}")
+        with pytest.raises(ExceptionRAPDU) as e:
+            ex._client.exchange(ex.CLA, c, p1=Rate.FIXED, p2=s, data=b"")
+        assert e.value.status == Errors.UNEXPECTED_INSTRUCTION
+
+@pytest.mark.parametrize("subcommand", ALL_SUBCOMMANDS)
+def test_wrong_flow_order(backend, subcommand, exchange_navigation_helper):
+    # Mutualize new and legacy snapshots. Eg SubCommand.SWAP_NG => "swap"
+    suffix = "_" + str(subcommand).split('.')[1].split('_')[0].lower()
+    exchange_navigation_helper.set_test_name_suffix(suffix)
+
+    ex = ExchangeClient(backend, Rate.FIXED, subcommand)
+    partner = SigningAuthority(curve=get_partner_curve(subcommand), name="Name")
+
+    try_all_commands_for_subcommand_except(ex, subcommand, Command.START_NEW_TRANSACTION)
+    transaction_id = ex.init_transaction().data
+
+    try_all_commands_for_subcommand_except(ex, subcommand, Command.SET_PARTNER_KEY)
+    ex.set_partner_key(partner.credentials)
+
+    try_all_commands_for_subcommand_except(ex, subcommand, Command.CHECK_PARTNER)
+    ex.check_partner_key(LEDGER_SIGNER.sign(partner.credentials))
+
+    try_all_commands_for_subcommand_except(ex, subcommand, Command.PROCESS_TRANSACTION_RESPONSE)
+    tx_infos = TX_INFOS[subcommand]
+    tx = craft_tx(subcommand, tx_infos, transaction_id)
+    ex.process_transaction(tx, FEES)
+
+    try_all_commands_for_subcommand_except(ex, subcommand, Command.CHECK_TRANSACTION_SIGNATURE)
+    signed_tx = encode_tx(subcommand, partner, tx)
+    ex.check_transaction_signature(signed_tx)
+
+    payout_ticker = extract_payout_ticker(subcommand, tx_infos)
+    payout_configuration = cal.get_conf_for_ticker(payout_ticker)
+
+    if subcommand == SubCommand.SWAP or subcommand == SubCommand.SWAP_NG:
+        try_all_commands_for_subcommand_except(ex, subcommand, Command.CHECK_PAYOUT_ADDRESS)
+        ex.check_payout_address(payout_configuration)
+
+        try_all_commands_for_subcommand_except(ex, subcommand, Command.CHECK_REFUND_ADDRESS)
+        refund_ticker = extract_refund_ticker(subcommand, tx_infos)
+        refund_configuration = cal.get_conf_for_ticker(refund_ticker)
+        with ex.check_refund_address(refund_configuration):
+            exchange_navigation_helper.simple_accept()
+    else:
+        if subcommand == SubCommand.FUND or subcommand == SubCommand.SELL:
+            try_all_commands_for_subcommand_except(ex, subcommand, Command.CHECK_ASSET_IN_LEGACY)
+        else:
+            try_all_commands_for_subcommand_except(ex, subcommand, Command.CHECK_ASSET_IN)
+        with ex.check_asset_in(payout_configuration):
+            exchange_navigation_helper.simple_accept()
+
+    try_all_commands_for_subcommand_except(ex, subcommand, Command.START_SIGNING_TRANSACTION)
+    ex.start_signing_transaction()