fixup! fixup! feat(core): show ETH account info on model T

trezor · Feb 11, 2025 · 9fc4846 · 9fc4846
1 parent 08b812e
commit 9fc4846
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/device_tests/ethereum/test_signtx.py b/tests/device_tests/ethereum/test_signtx.py
@@ -16,7 +16,7 @@
 
 import pytest
 
-from trezorlib import ethereum, exceptions, messages
+from trezorlib import ethereum, exceptions, messages, models
 from trezorlib.debuglink import TrezorClientDebugLink as Client
 from trezorlib.debuglink import message_filters
 from trezorlib.exceptions import TrezorFailure
@@ -220,10 +220,15 @@ def test_data_streaming(client: Client):
     checked in vectorized function above.
     """
     with client:
+        is_t1 = client.model is models.T1B1
         client.set_expected_responses(
             [
                 messages.ButtonRequest(code=messages.ButtonRequestType.SignTx),
-                messages.ButtonRequest(code=messages.ButtonRequestType.Other),
+                (is_t1, messages.ButtonRequest(code=messages.ButtonRequestType.SignTx)),
+                (
+                    not is_t1,
+                    messages.ButtonRequest(code=messages.ButtonRequestType.Other),
+                ),
                 messages.ButtonRequest(code=messages.ButtonRequestType.SignTx),
                 message_filters.EthereumTxRequest(
                     data_length=1_024,