coldcard/firmware
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improve test_usb.py::test_xpub_good

Browse Source
This commit is contained in:
scgbckbone 2024-02-22 15:44:55 +01:00 committed by doc-hex
parent 0a38118f51
commit 8c15ff6ae6
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
testing/test_usb.py
View File

@ -7,6 +7,7 @@
import pytest, struct
from pycoin.key.BIP32Node import BIP32Node
from binascii import b2a_hex
from constants import simulator_fixed_tprv
from ckcc_protocol.protocol import MAX_MSG_LEN, CCProtocolPacker, CCProtoError
@pytest.mark.skip
@ -84,9 +85,11 @@ def test_xpub_good(dev, master_xpub, path):
assert k.hwif() == xpub
mk = BIP32Node.from_wallet_key(master_xpub)
sk = mk.subkey_for_path(path[2:].replace("h", "'"))
assert sk.hwif() == xpub
is_hard = ("'" in path) or ("h" in path)
if not is_hard or dev.is_simulator:
mk = BIP32Node.from_wallet_key(simulator_fixed_tprv if is_hard else master_xpub)
sk = mk.subkey_for_path(path[2:].replace("h", "'"))
assert sk.hwif() == xpub
if len(path) <= 2:
assert mk.fingerprint() == struct.pack('<I', dev.master_fingerprint)