arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-04 05:28:17 +00:00
Code Issues Releases Wiki Activity

test: fix legacy test_firmware_upgrades device tests

Browse Source 
[no changelog]
This commit is contained in:
M1nd3r 2024-11-27 16:22:12 +01:00
parent 987b92f888
commit cc49648cf0
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/upgrade_tests/test_firmware_upgrades.py
View File

@ -137,11 +137,11 @@ def test_storage_upgrade_progressive(gen: str, tags: List[str]):
assert client.features.initialized assert client.features.initialized
assert client.features.label == LABEL assert client.features.label == LABEL
client.use_pin_sequence([PIN]) client.use_pin_sequence([PIN])
assert btc.get_address(client, "Bitcoin", PATH) == ADDRESS assert btc.get_address(client.get_session(), "Bitcoin", PATH) == ADDRESS
with EmulatorWrapper(gen, tags[0]) as emu: with EmulatorWrapper(gen, tags[0]) as emu:
debuglink.load_device_by_mnemonic( debuglink.load_device_by_mnemonic(
emu.client, emu.client.get_management_session(),
mnemonic=MNEMONIC, mnemonic=MNEMONIC,
pin=PIN, pin=PIN,
passphrase_protection=False, passphrase_protection=False,
@ -171,10 +171,7 @@ def test_upgrade_wipe_code(gen: str, tag: str):
assert client.features.initialized assert client.features.initialized
assert client.features.label == LABEL assert client.features.label == LABEL
client.use_pin_sequence([PIN]) client.use_pin_sequence([PIN])
assert ( assert btc.get_address(client.get_session(), "Bitcoin", PATH) == ADDRESS
btc.get_address(client.get_session(passphrase=""), "Bitcoin", PATH)
== ADDRESS
)
with EmulatorWrapper(gen, tag) as emu: with EmulatorWrapper(gen, tag) as emu:
debuglink.load_device_by_mnemonic( debuglink.load_device_by_mnemonic(
@ -205,7 +202,7 @@ def test_upgrade_wipe_code(gen: str, tag: str):
exceptions.TrezorFailure, exceptions.TrezorFailure,
match="The new PIN must be different from your wipe code", match="The new PIN must be different from your wipe code",
): ):
return device.change_pin(emu.client) return device.change_pin(session)
@for_all("legacy") @for_all("legacy")