arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-11 16:00:57 +00:00
Code Issues Releases Wiki Activity

core/recovery: change exceptions types; modifications in tests

Browse Source
This commit is contained in:
Tomas Susanka 2019-12-30 11:35:56 +00:00
parent 7f730cb6f9
commit d8e956fc6d
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
core/src/apps/management/recovery_device/word_validity.py
View File

@ -8,7 +8,7 @@ if False:
from trezor.messages.ResetDevice import EnumTypeBackupType
class WordValidityResult(BaseException):
class WordValidityResult(Exception):
pass

8
core/tests/test_apps.management.recovery_device.py
View File

@ -151,10 +151,10 @@ class TestSlip39(unittest.TestCase):
check(BackupType.Slip39_Advanced, ["ocean"])
# if backup type is not set we can not do any checks
self.assertIsNone(check(None, ["ocean"]))
check(None, ["ocean"])
# BIP-39 has no "on-the-fly" checks
self.assertIsNone(check(BackupType.Bip39, ["ocean"]))
check(BackupType.Bip39, ["ocean"])
# let's store two shares in the storage
secret, share = process_slip39("trash smug adjust ambition criminal prisoner security math cover pecan response pharmacy center criminal salary elbow bracelet lunar briefing dragon")
@ -170,6 +170,10 @@ class TestSlip39(unittest.TestCase):
with self.assertRaises(IdentifierMismatch):
check(BackupType.Slip39_Advanced, ["trash", "slush"])
# same identifier but different group settings for Slip 39 Basic
with self.assertRaises(IdentifierMismatch):
check(BackupType.Slip39_Basic, ["trash", "smug", "slush"])
# same mnemonic found out using the index
with self.assertRaises(AlreadyAdded):
check(BackupType.Slip39_Advanced, ["trash", "smug", "adjust", "ambition"])