def test(self):
xpub_str = 'xpub661MyMwAqRbcFW31YEwpkMuc5THy2PSt5bDMsktWQcFF8syAmRUapSCGu8ED9W6oDMSgv6Zz8idoc4a6mr8BDzTJY47LJhkJ8UB7WEGuduB'
xpub = CBitcoinExtPubKey(xpub_str)
xpub2 = CExtPubKey(xpub)
self.assertEqual(xpub_str, str(CBitcoinExtPubKey.from_bytes(xpub2)))
self.assertEqual(xpub, CBitcoinExtPubKey.from_bytes(xpub2))
self.assertEqual(xpub.derive(0), xpub2.derive(0))
def test_invalid_xpubkey(self):
invalid_xpub_str = 'xpub661MyMwAqRbcEYS8w7XLSVeEsBXy79zSzH1J8vCdxAZningWLdN3zgtU6Txnt3siSujt9RCVYsx4qHZGc62TG4McvMGcAUjeuwZdcaHQwzT'
with self.assertRaises(ValueError):
CBitcoinExtPubKey(invalid_xpub_str)
valid_xpubkey = CBitcoinExtPubKey('xpub661MyMwAqRbcG2veuy7DxC7yfKodTbY46UKYgnrERu9WADL5pBjRzthJtxMpWYeofw5rHWtemkgJAcSCup19Vorze6H3etYGbiEU4MumRV7')
with self.assertRaises(ValueError):
CExtPubKey(valid_xpubkey[:-1]) # short length
with self.assertRaises(ValueError):
CExtPubKey(valid_xpubkey + b'\x00') # long length
def test_extkey(self):
xprv_str = 'xprv9s21ZrQH143K3QTDL4LXw2F7HEK3wJUD2nW2nRk4stbPy6cq3jPPqjiChkVvvNKmPGJxWUtg6LnF5kejMRNNU3TGtRBeJgk33yuGBxrMPHi'
xprv = CBitcoinExtKey(xprv_str)
xprv2 = CExtKey(xprv)
self.assertEqual(xprv_str, str(CBitcoinExtKey.from_bytes(xprv2)))
self.assertEqual(bytes(xprv.derive(BIP32_HARDENED_KEY_LIMIT)),
bytes(xprv2.derive(BIP32_HARDENED_KEY_LIMIT)))
self.assertEqual(str(xprv.neuter()), str(CBitcoinExtPubKey.from_bytes(xprv2.neuter())))
def test_invalid_derivation(self):
xpub = CBitcoinExtPubKey(
'xpub661MyMwAqRbcFW31YEwpkMuc5THy2PSt5bDMsktWQcFF8syAmRUapSCGu8ED9W6oDMSgv6Zz8idoc4a6mr8BDzTJY47LJhkJ8UB7WEGuduB'
)
with self.assertRaises(ValueError):
xpub.derive(1 << 31)
final_xpub_str = 'xpubEPPCAoZp7t6CN5GGoyYTEr91FCaPpQonRouneRKmRCzgfcWNHnyHMuQPCDn8wLv1vYyPrFpSK26VeA9dDXTKMCLm7FaSY9aVTWw5mTZLC7F'
for _ in range(255):
xpub = xpub.derive(0)
self.assertEqual(str(CBitcoinExtPubKey.from_bytes(xpub)), final_xpub_str)
with self.assertRaises(ValueError):
xpub.derive(0) # depth > 255
def T(base_xpub, expected_child, path):
xpub = CBitcoinExtPubKey(base_xpub)
for child_num in path:
xpub = xpub.derive(child_num)
self.assertEqual(str(CBitcoinExtPubKey.from_bytes(xpub)),
expected_child)
xpub = CBitcoinExtPubKey(base_xpub).derive_path(
str(BIP32Path(path)))
self.assertEqual(str(CBitcoinExtPubKey.from_bytes(xpub)),
expected_child)
def T(base_xpub: str, expected_child: str, path: List[int]) -> None:
xpub = CBitcoinExtPubKey(base_xpub)
self.assertEqual(xpub.parent_fp, b'\x00\x00\x00\x00')
for child_num in path:
parent_fp = xpub.fingerprint
xpub = xpub.derive(child_num)
self.assertEqual(xpub.parent_fp, parent_fp)
self.assertEqual(str(CBitcoinExtPubKey.from_bytes(xpub)),
expected_child)
xpub = CBitcoinExtPubKey(base_xpub).derive_path(
str(BIP32Path(path)))
self.assertEqual(str(CBitcoinExtPubKey.from_bytes(xpub)),
expected_child)
def T(base_xpub, expected_child, path):
xpub = CBitcoinExtPubKey(base_xpub)
for child_num in path:
xpub = xpub.derive(child_num)
self.assertTrue(str(CBitcoinExtPubKey.from_bytes(xpub)) == expected_child)
def T(base58_xpubkey, expected_hex_xpubkey):
key = CBitcoinExtPubKey(base58_xpubkey)
self.assertEqual(b2x(key), expected_hex_xpubkey)
def test_from_xpub(self) -> None:
xpub_str = 'xpub661MyMwAqRbcFW31YEwpkMuc5THy2PSt5bDMsktWQcFF8syAmRUapSCGu8ED9W6oDMSgv6Zz8idoc4a6mr8BDzTJY47LJhkJ8UB7WEGuduB'
xpub = CBitcoinExtPubKey(xpub_str)
self.assertEqual(xpub_str, str(CBitcoinExtPubKey.from_bytes(xpub)))
def T(base58_xpubkey: str, expected_hex_xpubkey: str) -> None:
key = CBitcoinExtPubKey(base58_xpubkey)
self.assertEqual(b2x(key), expected_hex_xpubkey)
key2 = CBitcoinExtPubKey.from_bytes(x(expected_hex_xpubkey))
self.assertEqual(b2x(key), b2x(key2))
