forked from LedgerHQ/app-symbol
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_sign_cmd.py
76 lines (64 loc) · 3.72 KB
/
test_sign_cmd.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import pytest
from json import load
from ragger.backend.interface import RaisePolicy
from ragger.navigator import NavInsID, NavIns
from apps.dHealth import dHealthClient, ErrorType
from apps.dHealth_transaction_builder import encode_txn_context
from utils import ROOT_SCREENSHOT_PATH, CORPUS_DIR, CORPUS_FILES
# Proposed DHP derivation paths for tests ###
DHEALTH_PATH = "m/44'/1'/0'/0'/0'"
def load_transaction_from_file(transaction_filename):
with open(CORPUS_DIR / transaction_filename, "r") as f:
transaction = load(f)
return encode_txn_context(transaction)
def check_transaction(test_name, firmware, backend, navigator, transaction_filename):
transaction = load_transaction_from_file(transaction_filename)
nem = dHealthClient(backend)
with nem.send_async_sign_message(DHEALTH_PATH, transaction):
if firmware.device.startswith("nano"):
navigator.navigate_until_text_and_compare(NavInsID.RIGHT_CLICK,
[NavInsID.BOTH_CLICK,
NavIns(NavInsID.WAIT, ([0]))],
"Approve",
ROOT_SCREENSHOT_PATH,
test_name)
else:
navigator.navigate_until_text_and_compare(NavInsID.USE_CASE_REVIEW_TAP,
[NavInsID.USE_CASE_REVIEW_CONFIRM,
NavInsID.USE_CASE_STATUS_DISMISS],
"Hold to sign",
ROOT_SCREENSHOT_PATH,
test_name)
# Missing signature verification
@pytest.mark.parametrize("transaction_filename", CORPUS_FILES)
def test_sign_tx_accepted(test_name, firmware, backend, navigator, transaction_filename):
folder_name = test_name + "/" + transaction_filename.replace(".json", "")
check_transaction(folder_name, firmware, backend, navigator, transaction_filename)
def test_sign_tx_refused(test_name, firmware, backend, navigator):
transaction = load_transaction_from_file("transfer_transaction.json")
client = dHealthClient(backend)
if firmware.device.startswith("nano"):
with client.send_async_sign_message(DHEALTH_PATH, transaction):
backend.raise_policy = RaisePolicy.RAISE_NOTHING
navigator.navigate_until_text_and_compare(NavInsID.RIGHT_CLICK,
[NavInsID.BOTH_CLICK],
"Reject",
ROOT_SCREENSHOT_PATH,
test_name)
rapdu = client.get_async_response()
assert rapdu.status == ErrorType.TRANSACTION_REJECTED
assert len(rapdu.data) == 0
else:
for i in range(4):
instructions = [NavInsID.USE_CASE_REVIEW_TAP] * i
instructions += [NavInsID.USE_CASE_REVIEW_REJECT,
NavInsID.USE_CASE_CHOICE_CONFIRM,
NavInsID.USE_CASE_STATUS_DISMISS]
with client.send_async_sign_message(DHEALTH_PATH, transaction):
backend.raise_policy = RaisePolicy.RAISE_NOTHING
navigator.navigate_and_compare(ROOT_SCREENSHOT_PATH,
test_name + f"/part{i}",
instructions)
rapdu = client.get_async_response()
assert rapdu.status == ErrorType.TRANSACTION_REJECTED
assert len(rapdu.data) == 0