arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 15:38:11 +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 import pytest
from trezorlib import btc, device, messages from trezorlib import btc, device, messages
@ -13,10 +15,9 @@ def test_reset_recovery(client):
mnemonics = reset(client) mnemonics = reset(client)
address_before = btc.get_address(client, "Bitcoin", parse_path("44'/0'/0'/0/0")) 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)): for share_subset in itertools.combinations(mnemonics, 3):
# TODO: change the above to itertools.combinations(mnemonics, 3)
device.wipe(client) device.wipe(client)
selected_mnemonics = [mnemonics[i] for i in share_subset] selected_mnemonics = share_subset
recover(client, selected_mnemonics) recover(client, selected_mnemonics)
address_after = btc.get_address(client, "Bitcoin", parse_path("44'/0'/0'/0/0")) address_after = btc.get_address(client, "Bitcoin", parse_path("44'/0'/0'/0/0"))
assert address_before == address_after assert address_before == address_after