From 7a0f29e26c076432bdb99dea741f41e4834c58c8 Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Mon, 24 Mar 2025 13:40:20 +0100 Subject: [PATCH] fixup! test: update upgrade tests --- tests/upgrade_tests/test_firmware_upgrades.py | 2 +- tests/upgrade_tests/test_passphrase_consistency.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/upgrade_tests/test_firmware_upgrades.py b/tests/upgrade_tests/test_firmware_upgrades.py index 64606eb9b6..af43ce0cde 100644 --- a/tests/upgrade_tests/test_firmware_upgrades.py +++ b/tests/upgrade_tests/test_firmware_upgrades.py @@ -382,7 +382,7 @@ def test_upgrade_shamir_backup(gen: str, tag: Optional[str]): # Get a passphrase-less and a passphrased address. address = btc.get_address(session, "Bitcoin", PATH) - if session.protocol_version == ProtocolVersion.PROTOCOL_V1: + if session.protocol_version == ProtocolVersion.V1: session.call(messages.Initialize(new_session=True)) new_session = emu.client.get_session(passphrase="TREZOR") address_passphrase = btc.get_address(new_session, "Bitcoin", PATH) diff --git a/tests/upgrade_tests/test_passphrase_consistency.py b/tests/upgrade_tests/test_passphrase_consistency.py index cc785e4dbf..03be5f1ddb 100644 --- a/tests/upgrade_tests/test_passphrase_consistency.py +++ b/tests/upgrade_tests/test_passphrase_consistency.py @@ -141,7 +141,7 @@ def test_init_device(emulator: Emulator): btc.get_address(session, "Testnet", parse_path("44h/1h/0h/0/0")) # in TT < 2.3.0 session_id will only be available after PassphraseStateRequest session_id = session.id - if session.protocol_version == ProtocolVersion.PROTOCOL_V1: + if session.protocol_version == ProtocolVersion.V1: session.call(messages.Initialize(session_id=session_id)) btc.get_address( session,