From ea3519cf935e5d8df00138fd4cb8ff68ecdd7e0c Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Fri, 28 Mar 2025 15:15:09 +0100 Subject: [PATCH] fixup! test: update persistence tests --- .../test_shamir_persistence.py | 20 +++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/tests/persistence_tests/test_shamir_persistence.py b/tests/persistence_tests/test_shamir_persistence.py index 37e0cb67c5..d39bf18323 100644 --- a/tests/persistence_tests/test_shamir_persistence.py +++ b/tests/persistence_tests/test_shamir_persistence.py @@ -45,7 +45,10 @@ def test_abort(core_emulator: Emulator): assert features.recovery_status == RecoveryStatus.Nothing - device_handler.run_with_session(device.recover, pin_protection=False) + session = device_handler.client.get_seedless_session() + device_handler.run_with_provided_session( + session, device.recover, pin_protection=False + ) recovery.confirm_recovery(debug) layout = debug.read_layout() @@ -82,7 +85,10 @@ def test_recovery_single_reset(core_emulator: Emulator): assert features.initialized is False assert features.recovery_status == RecoveryStatus.Nothing - device_handler.run_with_session(device.recover, pin_protection=False) + session = device_handler.client.get_seedless_session() + device_handler.run_with_provided_session( + session, device.recover, pin_protection=False + ) recovery.confirm_recovery(debug) @@ -130,7 +136,10 @@ def test_recovery_on_old_wallet(core_emulator: Emulator): assert features.recovery_status == RecoveryStatus.Nothing # enter recovery mode - device_handler.run_with_session(device.recover, pin_protection=False) + session = device_handler.client.get_seedless_session() + device_handler.run_with_provided_session( + session, device.recover, pin_protection=False + ) recovery.confirm_recovery(debug) @@ -208,7 +217,10 @@ def test_recovery_multiple_resets(core_emulator: Emulator): assert features.recovery_status == RecoveryStatus.Nothing # start device and recovery - device_handler.run_with_session(device.recover, pin_protection=False) + session = device_handler.client.get_seedless_session() + device_handler.run_with_provided_session( + session, device.recover, pin_protection=False + ) recovery.confirm_recovery(debug)