arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 07:28:10 +00:00
Code Issues Releases Wiki Activity

python/tezos: add input flow for paginators

Browse Source
This commit is contained in:
matejcik 2019-04-18 15:44:09 +02:00
parent 1e1e6513a0
commit 3e74ed927b
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
python/trezorlib/tests/device_tests/test_msg_tezos_sign_tx.py
View File

@ -14,6 +14,8 @@
# You should have received a copy of the License along with this library. # You should have received a copy of the License along with this library.
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>. # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
import time
import pytest import pytest
from trezorlib import messages, tezos from trezorlib import messages, tezos
@ -196,9 +198,19 @@ class TestMsgTezosSignTx(TrezorTest):
resp.operation_hash == "oocgc3hyKsGHPsw6WFWJpWT8jBwQLtebQAXF27KNisThkzoj635" resp.operation_hash == "oocgc3hyKsGHPsw6WFWJpWT8jBwQLtebQAXF27KNisThkzoj635"
) )
def input_flow(self, num_pages):
yield
time.sleep(1)
for _ in range(num_pages - 1):
self.client.debug.swipe_down()
time.sleep(1)
self.client.debug.press_yes()
def test_tezos_sign_tx_proposal(self): def test_tezos_sign_tx_proposal(self):
self.setup_mnemonic_allallall() self.setup_mnemonic_allallall()
self.client.set_input_flow(self.input_flow(num_pages=1))
resp = tezos.sign_tx( resp = tezos.sign_tx(
self.client, self.client,
TEZOS_PATH_10, TEZOS_PATH_10,
@ -231,6 +243,7 @@ class TestMsgTezosSignTx(TrezorTest):
def test_tezos_sign_tx_multiple_proposals(self): def test_tezos_sign_tx_multiple_proposals(self):
self.setup_mnemonic_allallall() self.setup_mnemonic_allallall()
self.client.set_input_flow(self.input_flow(num_pages=2))
resp = tezos.sign_tx( resp = tezos.sign_tx(
self.client, self.client,
TEZOS_PATH_10, TEZOS_PATH_10,