arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-07-04 05:42:34 +00:00
Code Issues Releases Wiki Activity

tests: update test_cancel to test using Initialize

Browse Source
This commit is contained in:
Pavol Rusnak 2018-06-22 13:27:56 +02:00
parent f9f79b745c
commit 3c1456939c
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
1 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
trezorlib/tests/device_tests/test_cancel.py
View File

@ -30,7 +30,7 @@ import trezorlib.messages as m
show_display=True show_display=True
), ),
]) ])
def test_cancel_message(client, message): def test_cancel_message_via_cancel(client, message):
resp = client.call_raw(message) resp = client.call_raw(message)
assert isinstance(resp, m.ButtonRequest) assert isinstance(resp, m.ButtonRequest)
@ -41,3 +41,25 @@ def test_cancel_message(client, message):
assert isinstance(resp, m.Failure) assert isinstance(resp, m.Failure)
assert resp.code == m.FailureType.ActionCancelled assert resp.code == m.FailureType.ActionCancelled
@setup_client()
@pytest.mark.parametrize("message", [
m.Ping(message="hello", button_protection=True),
m.GetAddress(
address_n=[0],
coin_name="Bitcoin",
script_type=m.InputScriptType.SPENDADDRESS,
show_display=True
),
])
def test_cancel_message_via_initialize(client, message):
resp = client.call_raw(message)
assert isinstance(resp, m.ButtonRequest)
client.transport.write(m.ButtonAck())
client.transport.write(m.Initialize())
resp = client.transport.read()
assert isinstance(resp, m.Features)