From d1625ab4a82f7a0f159c4d30b738c6cb2ec4b88a Mon Sep 17 00:00:00 2001 From: matejcik Date: Fri, 19 Jul 2024 14:50:24 +0200 Subject: [PATCH] fixup! tests: fix small irregularities --- core/src/trezor/ui/layouts/tt/recovery.py | 2 +- tests/input_flows.py | 8 ++++---- tests/input_flows_helpers.py | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/core/src/trezor/ui/layouts/tt/recovery.py b/core/src/trezor/ui/layouts/tt/recovery.py index 13d04fcbd..6a6670b71 100644 --- a/core/src/trezor/ui/layouts/tt/recovery.py +++ b/core/src/trezor/ui/layouts/tt/recovery.py @@ -15,7 +15,7 @@ if TYPE_CHECKING: async def request_word_count(recovery_type: RecoveryType) -> int: count = await interact( trezorui2.select_word_count(recovery_type=recovery_type), - "word_count", + "recovery_word_count", ButtonRequestType.MnemonicWordCount, ) return int(count) diff --git a/tests/input_flows.py b/tests/input_flows.py index 2f48c47a0..2cff74823 100644 --- a/tests/input_flows.py +++ b/tests/input_flows.py @@ -461,11 +461,11 @@ class InputFlowShowMultisigXPUBs(InputFlowBase): # Three xpub pages with the same testing logic for xpub_num in range(3): - expected_title = f"multisig xpub #{xpub_num + 1}\n" + ( - "(yours)" if self.index == xpub_num else "(cosigner)" + expected_title = f"MULTISIG XPUB #{xpub_num + 1}\n" + ( + "(Yours)" if self.index == xpub_num else "(Cosigner)" ) layout = self.debug.swipe_left() - assert expected_title == layout.title().lower() + assert expected_title == layout.title() content = layout.text_content().replace(" ", "") assert self.xpubs[xpub_num] in content @@ -498,7 +498,7 @@ class InputFlowShowMultisigXPUBs(InputFlowBase): # Three xpub pages with the same testing logic for xpub_num in range(3): expected_title = f"MULTISIG XPUB #{xpub_num + 1} " + ( - "(YOURS)" if self.index == xpub_num else "(COSIGNER)" + "(Yours)" if self.index == xpub_num else "(Cosigner)" ) layout = self.debug.press_right() assert expected_title in layout.title() diff --git a/tests/input_flows_helpers.py b/tests/input_flows_helpers.py index b20fc4208..d2f505c78 100644 --- a/tests/input_flows_helpers.py +++ b/tests/input_flows_helpers.py @@ -189,7 +189,7 @@ class RecoveryFlow: def input_number_of_words(self, num_words: int) -> BRGeneratorType: br = yield assert br.code == B.MnemonicWordCount - assert br.name == "word_count" + assert br.name == "recovery_word_count" if self.client.model is models.T2B1: TR.assert_in(self.debug.read_layout().title(), "word_count__title") else: