test adaptation for current qr capture changes

selfcustody · Aug 5, 2024 · 0ab2135 · 0ab2135
1 parent 4b0a528
commit 0ab2135
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/pages/home_pages/test_sign_message_ui.py b/tests/pages/home_pages/test_sign_message_ui.py
@@ -245,10 +245,11 @@ def test_sign_message_at_address(mocker, m5stickv, tdata):
     mocker.patch.object(ctx.display, "width", new=lambda: 135)
     message_signer = SignMessage(ctx)
     mocker.spy(message_signer, "display_qr_codes")
+
     mocker.patch.object(
-        message_signer,
-        "capture_qr_code",
-        new=lambda: (
+        QRCodeCapture,
+        "qr_capture_loop",
+        new=lambda self: (
             "signmessage m/84h/1h/0h/0/3 ascii:a test message with a colon ':' character.",
             FORMAT_NONE,
         ),