arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-04-26 20:19:02 +00:00
Code Issues Releases Wiki Activity

fix(tests): fix THP upgrade tests

Browse Source
This commit is contained in:
M1nd3r 2025-03-11 17:05:40 +01:00
parent 0efe6d53d9
commit bc9deff451
2 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/upgrade_tests/test_firmware_upgrades.py
View File

@ -375,9 +375,8 @@ def test_upgrade_shamir_recovery(gen: str, tag: Optional[str]):
emu.client.watch_layout(True) emu.client.watch_layout(True)
debug = device_handler.debuglink() debug = device_handler.debuglink()
session = emu.client.get_seedless_session() device_handler.run_with_session(
device_handler.run_with_provided_session( device.recover, seedless=True, pin_protection=False
session, device.recover, pin_protection=False
) )
recovery_old.confirm_recovery(debug) recovery_old.confirm_recovery(debug)

14
tests/upgrade_tests/test_passphrase_consistency.py
View File

@ -82,7 +82,7 @@ def test_passphrase_works(emulator: Emulator):
messages.PassphraseRequest, messages.PassphraseRequest,
messages.Address, messages.Address,
] ]
else: elif protocol_v1:
expected_responses = [ expected_responses = [
(protocol_v1, messages.Features), (protocol_v1, messages.Features),
messages.PassphraseRequest, messages.PassphraseRequest,
@ -90,6 +90,8 @@ def test_passphrase_works(emulator: Emulator):
messages.ButtonRequest, messages.ButtonRequest,
messages.Address, messages.Address,
] ]
else:
expected_responses = [messages.Address]
with emulator.client as client: with emulator.client as client:
client.set_expected_responses(expected_responses) client.set_expected_responses(expected_responses)
if protocol_v1: if protocol_v1:
@ -132,7 +134,7 @@ def test_init_device(emulator: Emulator):
messages.Features, messages.Features,
messages.Address, messages.Address,
] ]
else: elif protocol_v1:
expected_responses = [ expected_responses = [
(protocol_v1, messages.Features), (protocol_v1, messages.Features),
messages.PassphraseRequest, messages.PassphraseRequest,
@ -142,6 +144,12 @@ def test_init_device(emulator: Emulator):
messages.Features, messages.Features,
messages.Address, messages.Address,
] ]
else:
expected_responses = [
messages.Address,
messages.Features,
messages.Address,
]
with emulator.client as client: with emulator.client as client:
client.set_expected_responses(expected_responses) client.set_expected_responses(expected_responses)
@ -168,6 +176,8 @@ def test_init_device(emulator: Emulator):
session_id = session.id session_id = session.id
if protocol_v1: if protocol_v1:
session.call(messages.Initialize(session_id=session_id)) session.call(messages.Initialize(session_id=session_id))
else:
session.call(messages.GetFeatures())
btc.get_address( btc.get_address(
session, session,
"Testnet", "Testnet",