From b59e14cc742e58d91b02863f9c10c838a00466f6 Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Tue, 11 Mar 2025 16:10:45 +0100 Subject: [PATCH] fix(python): partially fix shamir persistence tests --- tests/device_handler.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/device_handler.py b/tests/device_handler.py index a3dc61314d..3197ea5f7e 100644 --- a/tests/device_handler.py +++ b/tests/device_handler.py @@ -59,6 +59,7 @@ class BackgroundDeviceHandler: def run_with_session( self, function: t.Callable[tx.Concatenate["Session", P], t.Any], + seedless: bool = False, *args: P.args, **kwargs: P.kwargs, ) -> None: @@ -70,7 +71,10 @@ class BackgroundDeviceHandler: raise RuntimeError("Wait for previous task first") # wait for the first UI change triggered by the task running in the background - session = self.client.get_session() + if seedless: + session = self.client.get_seedless_session() + else: + session = self.client.get_session() with self.debuglink().wait_for_layout_change(): self.task = self._pool.submit(function, session, *args, **kwargs)