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

tests: improve Shamir persistence test

Browse Source
This commit is contained in:
matejcik 2019-10-22 13:55:17 +02:00
parent 5488270bc3
commit 9b827baca0
1 changed files with 32 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/upgrade_tests/test_shamir_persistence.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 pytest
from trezorlib import device from trezorlib import device
from .. import buttons from .. import buttons
@ -29,11 +31,16 @@ def enter_word(debug, word):
return debug.click(buttons.CONFIRM_WORD, wait=True) return debug.click(buttons.CONFIRM_WORD, wait=True)
@pytest.fixture
def emulator():
emu = EmulatorWrapper("core")
with emu:
yield emu
@core_only @core_only
def test_persistence(): def test_persistence(emulator):
with EmulatorWrapper("core") as emu, BackgroundDeviceHandler( device_handler = BackgroundDeviceHandler(emulator.client)
emu.client
) as device_handler:
debug = device_handler.debuglink() debug = device_handler.debuglink()
features = device_handler.features() features = device_handler.features()
@ -46,12 +53,7 @@ def test_persistence():
layout = debug.click(buttons.OK, wait=True) layout = debug.click(buttons.OK, wait=True)
assert "Select number of words" in layout.text assert "Select number of words" in layout.text
storage = emu.storage() device_handler.restart(emulator)
device_handler.kill_task()
with EmulatorWrapper("core", storage=storage) as emu, BackgroundDeviceHandler(
emu.client
) as device_handler:
debug = device_handler.debuglink() debug = device_handler.debuglink()
features = device_handler.features() features = device_handler.features()