1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-06-28 02:42:34 +00:00

fixup! test: update device tests

This commit is contained in:
M1nd3r 2025-03-24 15:49:37 +01:00
parent 272f024c25
commit 0e9eed8a39

View File

@ -19,7 +19,6 @@ from mnemonic import Mnemonic
from trezorlib import device, messages from trezorlib import device, messages
from trezorlib.btc import get_public_node from trezorlib.btc import get_public_node
from trezorlib.client import ProtocolVersion
from trezorlib.debuglink import LayoutType from trezorlib.debuglink import LayoutType
from trezorlib.debuglink import SessionDebugWrapper as Session from trezorlib.debuglink import SessionDebugWrapper as Session
from trezorlib.exceptions import TrezorFailure from trezorlib.exceptions import TrezorFailure
@ -156,11 +155,7 @@ def test_reset_entropy_check(session: Session):
assert IF.mnemonic == expected_mnemonic assert IF.mnemonic == expected_mnemonic
# Check that the device is properly initialized. # Check that the device is properly initialized.
if session.client.protocol_version is ProtocolVersion.V1: features = session.refresh_features()
features = session.call_raw(messages.Initialize())
else:
session.refresh_features()
features = session.features
assert features.initialized is True assert features.initialized is True
assert features.backup_availability == messages.BackupAvailability.NotAvailable assert features.backup_availability == messages.BackupAvailability.NotAvailable