arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-01 20:08:26 +00:00
Code Issues Releases Wiki Activity

fix(python): fix test_msg_backup_device interruption

Browse Source 
[no changelog]
This commit is contained in:
M1nd3r 2024-11-25 10:45:58 +01:00
parent d058648de8
commit 79af52366d
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/device_tests/test_msg_backup_device.py
View File

@ -171,8 +171,7 @@ def test_no_backup_fails(session: Session):
# we only test this with bip39 because the code path is always the same # we only test this with bip39 because the code path is always the same
@pytest.mark.setup_client(needs_backup=True) @pytest.mark.setup_client(needs_backup=True)
def test_interrupt_backup_fails(session: Session): def test_interrupt_backup_fails(session: Session):
raise Exception("TEST DOESN'T WORK BECAUSE OF CHANGED INITIALIZE BEHAVIOUR") session.ensure_unlocked()
# TODO ?? session.ensure_unlocked()
assert session.features.initialized is True assert session.features.initialized is True
assert session.features.backup_availability == messages.BackupAvailability.Required assert session.features.backup_availability == messages.BackupAvailability.Required
assert session.features.unfinished_backup is False assert session.features.unfinished_backup is False
@ -182,16 +181,18 @@ def test_interrupt_backup_fails(session: Session):
session.call_raw(messages.BackupDevice()) session.call_raw(messages.BackupDevice())
# interupt backup by sending initialize # interupt backup by sending initialize
# TODO remove session.init_device() session2 = session.client.resume_session(session)
session2.refresh_features()
# check that device state is as expected # check that device state is as expected
assert session.features.initialized is True assert session2.features.initialized is True
assert ( assert (
session.features.backup_availability == messages.BackupAvailability.NotAvailable session2.features.backup_availability
== messages.BackupAvailability.NotAvailable
) )
assert session.features.unfinished_backup is True assert session2.features.unfinished_backup is True
assert session.features.no_backup is False assert session2.features.no_backup is False
# Second attempt at backup should fail # Second attempt at backup should fail
with pytest.raises(TrezorFailure, match=r".*Seed already backed up"): with pytest.raises(TrezorFailure, match=r".*Seed already backed up"):
device.backup(session) device.backup(session2)