test(core): test UI abort after missing ButtonAck

[no changelog]

Author: romanz

tests/device_tests/test_msg_delayed_ack.py

@@ -14,7 +14,10 @@
 
 import time
 
-from trezorlib import messages
+import pytest
+
+from tests import translations as TR
+from trezorlib import messages, models
 from trezorlib.debuglink import DebugSession as Session
 
 
@@ -27,5 +30,25 @@ def test_delayed_ack(session: Session):
     # "waiting" screen should be shown after 2 seconds on Core models
     # (following https://github.com/trezor/trezor-firmware/issues/5884)
     time.sleep(2.5)
+    if session.model not in models.LEGACY_MODELS:
+        assert TR.words__comm_trouble in session.debug.read_layout().text_content()
     res = session.call_raw(messages.ButtonAck())
+    res = messages.Success.ensure_isinstance(res)
     assert res.message == "delayed"
+
+
+@pytest.mark.models("core")
+def test_delayed_ack_abort(session: Session):
+    br = session.call_raw(messages.Ping(message="delayed", button_protection=True))
+    assert isinstance(br, messages.ButtonRequest)
+    assert br.code == messages.ButtonRequestType.ProtectCall
+    # confirm layout instead of sending ButtonAck
+    session.debug.press_yes()
+    # "waiting" screen should be shown after 2 seconds on Core models
+    # (following https://github.com/trezor/trezor-firmware/issues/5884)
+    time.sleep(2.5)
+    assert TR.words__comm_trouble in session.debug.read_layout().text_content()
+    session.debug.press_yes()  # abort flow on device (instead of ButtonAck)
+    res = session.read()
+    res = messages.Failure.ensure_isinstance(res)
+    assert res.code == messages.FailureType.ActionCancelled