arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-14 03:30:02 +00:00
Code Issues Releases Wiki Activity

fixup! tests: fix small irregularities

Browse Source
This commit is contained in:
matejcik 2024-07-19 14:50:24 +02:00 committed by M1nd3r
parent 8f4cf99e48
commit d1625ab4a8
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
core/src/trezor/ui/layouts/tt/recovery.py
View File

@ -15,7 +15,7 @@ if TYPE_CHECKING:
async def request_word_count(recovery_type: RecoveryType) -> int: async def request_word_count(recovery_type: RecoveryType) -> int:
count = await interact( count = await interact(
trezorui2.select_word_count(recovery_type=recovery_type), trezorui2.select_word_count(recovery_type=recovery_type),
"word_count", "recovery_word_count",
ButtonRequestType.MnemonicWordCount, ButtonRequestType.MnemonicWordCount,
) )
return int(count) return int(count)

8
tests/input_flows.py
View File

@ -461,11 +461,11 @@ class InputFlowShowMultisigXPUBs(InputFlowBase):
# Three xpub pages with the same testing logic # Three xpub pages with the same testing logic
for xpub_num in range(3): for xpub_num in range(3):
expected_title = f"multisig xpub #{xpub_num + 1}\n" + ( expected_title = f"MULTISIG XPUB #{xpub_num + 1}\n" + (
"(yours)" if self.index == xpub_num else "(cosigner)" "(Yours)" if self.index == xpub_num else "(Cosigner)"
) )
layout = self.debug.swipe_left() layout = self.debug.swipe_left()
assert expected_title == layout.title().lower() assert expected_title == layout.title()
content = layout.text_content().replace(" ", "") content = layout.text_content().replace(" ", "")
assert self.xpubs[xpub_num] in content assert self.xpubs[xpub_num] in content
@ -498,7 +498,7 @@ class InputFlowShowMultisigXPUBs(InputFlowBase):
# Three xpub pages with the same testing logic # Three xpub pages with the same testing logic
for xpub_num in range(3): for xpub_num in range(3):
expected_title = f"MULTISIG XPUB #{xpub_num + 1} " + ( 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() layout = self.debug.press_right()
assert expected_title in layout.title() assert expected_title in layout.title()

2
tests/input_flows_helpers.py
View File

@ -189,7 +189,7 @@ class RecoveryFlow:
def input_number_of_words(self, num_words: int) -> BRGeneratorType: def input_number_of_words(self, num_words: int) -> BRGeneratorType:
br = yield br = yield
assert br.code == B.MnemonicWordCount assert br.code == B.MnemonicWordCount
assert br.name == "word_count" assert br.name == "recovery_word_count"
if self.client.model is models.T2B1: if self.client.model is models.T2B1:
TR.assert_in(self.debug.read_layout().title(), "word_count__title") TR.assert_in(self.debug.read_layout().title(), "word_count__title")
else: else: