You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
7.5 KiB
123 lines
7.5 KiB
import unittest
|
|
|
|
from lib import account
|
|
from lib import wallet
|
|
|
|
class Test_Account(unittest.TestCase):
|
|
|
|
def test_bip32_account(self):
|
|
v = {
|
|
'change': [
|
|
'02d2967089cbcecf308f133cdec7e97eeeb53a1d8d76fc3656eaa55dac67b7694c',
|
|
'023a667b846434d35fa76d5fe452c11a74504f5d6d551ad7fb9fe837041c3ef1de',
|
|
'036df6df47ad3dae0594ff6f470daa4bded564d8d7ad9420ce077fa3ded949ffa9',
|
|
'0211f0dce7457dc5f0a5ae4222552fea08f5fc587bff2fd2b9744c1cb5dc3a673e',
|
|
'0239706f85ef5564ca95e9328cbc652bfc4295a1a6940d6b589f857332c2c8b29c',
|
|
'030cf3e44121e753884d21daac49d3e47edc5fce1a2ad2f3c6ef296e349ca5d54b'
|
|
],
|
|
'receiving': [
|
|
'02f0eaac8dde84cf80ebdb3b136cb29d8c7954c869c6c8fdf9d72a82323a72a30e',
|
|
'02a6f4acc94dc4496a78fad745897fec3b334b182a376ac7abe40975b9333ef67c',
|
|
'03a47015d474ecc80fa20160d5b1bdb7696d020c6c220adeff51e616445aff1564',
|
|
'030c7a3c94425e11dba07bb7bd147451f1b412b4b6659e467f2047383daf330e83',
|
|
'026cc8f26926a4c86d12bb456f4d231c56b1fb3f4bd9f86f5e94f9cead5c28f09d',
|
|
'0358ad542c97f2908458cd5d064e538b57d9ce9af35461e514388da16fba74344f',
|
|
'029dae6d94528de02c51d1f61c48c1e5cd36399735a1de10e1beddd561f0938bca',
|
|
'0354c030301461884e7728e12374b98bb85a82f199676f0c222be7fa8b157a72c8',
|
|
'03d9c693eb4e9ce9b9d8ffebec8ac0c4dbcaf928f39fb838474143f171734e5789',
|
|
'03765cfe701e03dcb9be91d4b40fb6b0717a53000aabdbdea275290bddb627921e',
|
|
'0224104233a86b5f6381e12f2aea3121bc817029526440cdfcf83d6f5d7b7c11fb',
|
|
'03aa93f028c330e00b5b95c790672beab4d654b06b73b5d06b8eb1a1373ada76de',
|
|
'0223d0ffeebd7d081c7d62206f1b57a02323712d90e4b1e4444647e9946b6dedfe',
|
|
'032b1bfc1a500eed050c7ba946827d048d4bef7230a327916d33edf68eff43cf37',
|
|
'039bebf87a2dba3e2e842f7e56c131c843cacc1e4ca827d479cc87aba8ac74e38d',
|
|
'028de15894751cbe5916ebd38c1c22129983560a5b1cc95b6e215661ae0bb193e4',
|
|
'02ecb390d1a92af7193dce256b18af2b65def1398069eed58291d9d467421a54ff',
|
|
'029538f295c5ab4225f15ac5e35299eb56cb84ee13e28371ac50409a140f8834f2',
|
|
'029482ac8185bda04d771c008ca6979d8cb786d4f47398752826908e46c9cc33ab',
|
|
'02624a80408fc39db86848bb6995bdc52ccc9591cff4633a4536a60f270d502cd4'
|
|
],
|
|
'xpub': 'xpub661MyMwAqRbcF8M4CH68NvHEc6TUNaVhXwmGrsagNjrCja49H9L4ziJGe8YmaSBPbY4ZmQPQeW5CK6fiwx2EH6VxQab3zwDzZVWVApDSVNh'
|
|
}
|
|
a = account.BIP32_Account(v)
|
|
self.assertEquals(a.dump(), v)
|
|
self.assertEquals(a.get_master_pubkeys(), [v['xpub']])
|
|
self.assertEquals(a.first_address(),
|
|
('1EtJphMVpes4UKm8bYu5D1fGvNoTSJM3ZL', v['receiving'][0]))
|
|
|
|
xprv = 'xprv9s21ZrQH143K2eGb6FZ81nLW44cyy7mrAiqg4VB4pQKDrmizjc1pSuynnpeiaMPdZxvrfvdBi5oqFi9hmsV7MrsVquKkruQ7TJPCfVuPSdw'
|
|
storage = dict(
|
|
master_public_keys={0: a.xpub},
|
|
master_private_keys={0: xprv},
|
|
wallet_type='standard'
|
|
)
|
|
w = wallet.BIP32_Wallet(storage)
|
|
self.assertEquals(a.get_private_key(sequence=[0, 0], wallet=w, password=None),
|
|
['KxuBFG13CPUBwPAUWvZSQ3mjNNjHoDghfxnax6RbwS3Rw8tqSzCk'])
|
|
|
|
for for_change in [0, 1]:
|
|
for n in range(6):
|
|
label = ['receiving', 'change'][for_change]
|
|
expected = v[label][n]
|
|
self.assertEquals(expected, a.derive_pubkey_from_xpub(a.xpub, for_change, n))
|
|
self.assertEquals(expected, a.get_pubkey_from_xpub(a.xpub, for_change, n))
|
|
self.assertEquals(expected, a.derive_pubkeys(for_change, n))
|
|
|
|
pubkey, = a.get_xpubkeys(for_change, n)
|
|
xpub, seq = a.parse_xpubkey(pubkey)
|
|
self.assertEquals(xpub, a.xpub)
|
|
self.assertEquals(seq, [for_change, n])
|
|
|
|
def test_old_account(self):
|
|
v = {
|
|
'change': [
|
|
'04def41cbdc23003c5636fb3aafdbce23062a06bffeca3d02ea613a9516b1f884397a6718bc6eb45e1e58ee1bb16e13108b94061680b7930479bee7b45fbaec9b7',
|
|
'046570a52ad1d313821b22175b869171a118c7f8d032c6d4d8614a2672807b6a092cf9702dd457aa6dd1ee4cd66ff0593280d407e4dc2c7c8f8a78551390b6bb5c',
|
|
'04ee98d63800824486a1cf5b4376f2f574d86e0a3009a6448105703453f3368e8e1d8d090aaecdd626a45cc49876709a3bbb6dc96a4311b3cac03e225df5f63dfc',
|
|
'049251a067684656c18c2e009d8ecdf735515599d2ae838cf94646ed89a370e818d6653f5e1d2e3fb1c64f45875e717ba95ea3239d3f30854c9aeaefa33d5e2dc8',
|
|
'04c89a24f4ffc71f622e46ce35096c1bd31078b96805742edf94e6011a72e6cad5de3b9bef92cb37fbf7ca0680073aa1270feb274562212829199494330c588651',
|
|
'04b68706478edad4fb4bdac4fc3ddeacf337182ae291462b4f63b765bbb0eef12e96008a8682c3e6ee4e66380c9e80fd7ec4ca5423ebc179fd581281ad4c7bac47',
|
|
'0454bb2aee65d7aa78ee9c5ce5e6a4eee0f9c986114fafc59431cfaf51e503095c362afc011079aa6f9c123bce828c1fa8a19de06dcd0eb09725fc2350d2e30409'
|
|
],
|
|
'receiving': [
|
|
'040900f07c15d3fa441979e71d7ccdcca1afc30a28de07a0525a3d7655dc49cca0f844fb0903b3cccc4604107a9de6a0571c4a39996a9e4bd6ab596138ecae54f5',
|
|
'0478aa6e296340d15563b1af073df57319ff4ffc09b16ff1b0f7c00e7b41410fb1353f59f5b9ce72853a53cb5a31416ed747352ccb9d55557f5e740121d2b1354d',
|
|
'0493021f661df7e2af42d2a6d8ee18b4566e17072040e649c4a272de3692cbad388b679b96b5c4216146868baf4a31d9aa0c07ba375cc3b166fcd3a1e7151705a0',
|
|
'0496ba9dc1028e109d01ca82b57fc1efec7dcdca74b2afabad45cdfecc84cc72c658d9404f7c4b36e67960d91b50e1e19da4e2ceb9348995aea74fecd22172ae96',
|
|
'04534a39b2418bbde9b6841e7ba7ee863cc392276d393425c5019f3eb3f68f0a3a47509ccea9837e6718e512917791e456ba1600bc953d54c84b8826a568315a53'
|
|
],
|
|
'mpk': '4e13b0f311a55b8a5db9a32e959da9f011b131019d4cebe6141b9e2c93edcbfc0954c358b062a9f94111548e50bde5847a3096b8b7872dcffadb0e9579b9017b'
|
|
}
|
|
|
|
seed = '00000000000000000000000000000000'
|
|
self.assertEquals(account.OldAccount.mpk_from_seed(seed), v['mpk'])
|
|
|
|
a = account.OldAccount(v)
|
|
self.assertEquals(a.get_master_pubkeys(), [v['mpk']])
|
|
self.assertEquals(a.get_address(for_change=0, n=0), '1FHsTashEBUNPQwC1CwVjnKUxzwgw73pU4')
|
|
self.assertEquals(a.get_address(for_change=0, n=2), '1Got6wbjxQ592WfwLcfLLxn3aTetLzpTom')
|
|
self.assertEquals(a.get_address(for_change=1, n=0), '16RyjNDNEwwWkv6mvptvxT9qNN5shJxcxo')
|
|
self.assertEquals(a.get_address(for_change=1, n=3), '1M6kHXnzmiUNsoYKZgzPDVpsSmMcfKFiiM')
|
|
|
|
self.assertTrue(a.check_seed(seed))
|
|
with self.assertRaises(account.InvalidPassword):
|
|
a.check_seed('1' * len(seed))
|
|
|
|
storage = {
|
|
'seed': '00000000000000000000000000000000',
|
|
'wallet_type': 'old'
|
|
}
|
|
w = wallet.OldWallet(storage)
|
|
privkey = a.get_private_key(sequence=[0, 0], wallet=w, password=None)
|
|
self.assertEquals(privkey, ['5Khs7w6fBkogoj1v71Mdt4g8m5kaEyRaortmK56YckgTubgnrhz'])
|
|
|
|
for for_change in [0, 1]:
|
|
for n in range(5):
|
|
label = ['receiving', 'change'][for_change]
|
|
pubkey = a.derive_pubkeys(for_change, n)
|
|
self.assertEquals(pubkey, v[label][n])
|
|
|
|
pubkey, = a.get_xpubkeys(for_change, n)
|
|
mpk, seq = a.parse_xpubkey(pubkey)
|
|
self.assertEquals(mpk, v['mpk'])
|
|
self.assertEquals(seq, [for_change, n])
|
|
|