arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-23 05:40:57 +00:00
Code Issues Releases Wiki Activity

feat(tests): Add test for busy screen.

Browse Source
This commit is contained in:
Andrew Kozlik 2022-08-11 11:47:30 +02:00 committed by Andrew Kozlik
parent a3ac5d752e
commit f253d7e591
2 changed files with 70 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
tests/device_tests/test_busy_state.py Normal file
View File

@ -0,0 +1,68 @@
# This file is part of the Trezor project.
#
# Copyright (C) 2022 SatoshiLabs and contributors
#
# This library is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License version 3
# as published by the Free Software Foundation.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# 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>.
import time
import pytest
from trezorlib import btc, device
from trezorlib.debuglink import TrezorClientDebugLink as Client
from trezorlib.tools import parse_path
PIN = "1234"
pytestmark = pytest.mark.skip_t1
@pytest.mark.setup_client(pin=PIN)
def test_busy_state(client: Client):
assert client.features.unlocked is False
assert client.features.busy is False
# Show busy dialog for 1 minute.
device.set_busy(client, expiry_ms=60 * 1000)
assert client.features.unlocked is False
assert client.features.busy is True
with client:
client.use_pin_sequence([PIN])
btc.get_address(
client, "Bitcoin", parse_path("m/44h/0h/0h/0/0"), show_display=True
)
client.refresh_features()
assert client.features.unlocked is True
assert client.features.busy is True
# Hide the busy dialog.
device.set_busy(client, None)
assert client.features.unlocked is True
assert client.features.busy is False
def test_busy_expiry(client: Client):
expiry_ms = 100 # 100 milliseconds
# Show the busy dialog.
device.set_busy(client, expiry_ms=expiry_ms)
# Wait for it to expire.
time.sleep(expiry_ms / 1000)
# Check that the device is no longer busy.
client.refresh_features()
assert client.features.busy is False

2
tests/ui_tests/fixtures.json
View File

@ -1535,6 +1535,8 @@
"TT_test_basic.py::test_device_id_same": "c09de07fbbf1e047442180e2facb5482d06a1a428891b875b7dd93c9e4704ae1",
"TT_test_basic.py::test_features": "c09de07fbbf1e047442180e2facb5482d06a1a428891b875b7dd93c9e4704ae1",
"TT_test_basic.py::test_ping": "c09de07fbbf1e047442180e2facb5482d06a1a428891b875b7dd93c9e4704ae1",
"TT_test_busy_state.py::test_busy_expiry": "84cbf08ffe52f79e63d32ed2298dbdc60c74059bcf59eaa69c01f13acf61c8ae",
"TT_test_busy_state.py::test_busy_state": "0517aa3301d4f55068766dfc130139e45449d8f90213d4a51c49776e3bbb7c98",
"TT_test_cancel.py::test_cancel_message_via_cancel[message0]": "b014449cbf1a45739d64a370b30af75df2228f48c090a02227bac8ed20c7b2dc",
"TT_test_cancel.py::test_cancel_message_via_cancel[message1]": "b014449cbf1a45739d64a370b30af75df2228f48c090a02227bac8ed20c7b2dc",
"TT_test_cancel.py::test_cancel_message_via_initialize[message0]": "b014449cbf1a45739d64a370b30af75df2228f48c090a02227bac8ed20c7b2dc",