try fixing test_regression_continue_after_reject

trilitech · Dec 13, 2024 · d204a1e · d204a1e
1 parent dce68a2
commit d204a1e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/integration/nano/test_wrong_apdu.py b/tests/integration/nano/test_wrong_apdu.py
@@ -34,8 +34,12 @@ def test_regression_continue_after_reject(
         snapshot_dir: Path):
     """Check the app still runs after rejects signing"""
 
+    backend.wait_for_home_screen()
+
     tezos_navigator.toggle_expert_mode()
 
+    backend.wait_for_home_screen()
+
     with StatusCode.REJECT.expected():
         tezos_navigator.reject_public_key(account, snap_path=snapshot_dir / "reject_public_key")
 
@@ -61,6 +65,8 @@ def test_regression_continue_after_reject(
             snap_path=snapshot_dir / "reject_signing"
         )
 
+    backend.wait_for_home_screen()
+
     backend.get_public_key(account, with_prompt=False)