1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-14 03:30:02 +00:00

tests: update test_cancel to test using Initialize

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

View File

@ -30,7 +30,7 @@ import trezorlib.messages as m
show_display=True
),
])
def test_cancel_message(client, message):
def test_cancel_message_via_cancel(client, message):
resp = client.call_raw(message)
assert isinstance(resp, m.ButtonRequest)
@ -41,3 +41,25 @@ def test_cancel_message(client, message):
assert isinstance(resp, m.Failure)
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)