arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-07-06 14:52:33 +00:00
Code Issues Releases Wiki Activity

tests: make use of input_flow in test_cancel

Browse Source
This commit is contained in:
matejcik 2019-08-26 18:21:25 +02:00
parent 7f443778f1
commit 77d4f02b3a
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/device_tests/test_cancel.py
View File

@ -17,6 +17,7 @@
import pytest import pytest
import trezorlib.messages as m import trezorlib.messages as m
from trezorlib.exceptions import Cancelled
from .conftest import setup_client from .conftest import setup_client
@ -35,16 +36,14 @@ from .conftest import setup_client
], ],
) )
def test_cancel_message_via_cancel(client, message): def test_cancel_message_via_cancel(client, message):
resp = client.call_raw(message) def input_flow():
assert isinstance(resp, m.ButtonRequest) yield
client.cancel()
client.transport.write(m.ButtonAck()) with client, pytest.raises(Cancelled):
client.transport.write(m.Cancel()) client.set_expected_responses([m.ButtonRequest(), m.Failure()])
client.set_input_flow(input_flow)
resp = client.transport.read() client.call(message)
assert isinstance(resp, m.Failure)
assert resp.code == m.FailureType.ActionCancelled
@setup_client() @setup_client()