arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 07:28:10 +00:00
Code Issues Releases Wiki Activity

tests: check all combination of Shamir shares in reset-recovery test

Browse Source
This commit is contained in:
matejcik 2019-09-10 11:52:59 +02:00
parent 7be9bcac79
commit e31a499702
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/device_tests/test_shamir_reset_recovery.py
View File

@ -1,3 +1,5 @@
import itertools
import pytest
from trezorlib import btc, device, messages
@ -13,10 +15,9 @@ def test_reset_recovery(client):
mnemonics = reset(client)
address_before = btc.get_address(client, "Bitcoin", parse_path("44'/0'/0'/0/0"))
for share_subset in ((0, 1, 2), (4, 3, 2), (2, 1, 3)):
# TODO: change the above to itertools.combinations(mnemonics, 3)
for share_subset in itertools.combinations(mnemonics, 3):
device.wipe(client)
selected_mnemonics = [mnemonics[i] for i in share_subset]
selected_mnemonics = share_subset
recover(client, selected_mnemonics)
address_after = btc.get_address(client, "Bitcoin", parse_path("44'/0'/0'/0/0"))
assert address_before == address_after