tMerge pull request #2950 from SomberNight/tests_wallet_keystore - electrum - Electrum Bitcoin wallet
HTML git clone https://git.parazyd.org/electrum
DIR Log
DIR Files
DIR Refs
DIR Submodules
---
DIR commit 1daecf285d27f369eade81ba58f6d34528508c05
DIR parent c73819a3d4f4990f4e91bafbd875c48a46a69fa8
HTML Author: ThomasV <thomasv@electrum.org>
Date: Fri, 29 Sep 2017 20:12:42 +0200
Merge pull request #2950 from SomberNight/tests_wallet_keystore
ttests: wallet-keystore integrity (seed->address)
Diffstat:
A lib/tests/test_wallet_vertical.py | 135 +++++++++++++++++++++++++++++++
1 file changed, 135 insertions(+), 0 deletions(-)
---
DIR diff --git a/lib/tests/test_wallet_vertical.py b/lib/tests/test_wallet_vertical.py
t@@ -0,0 +1,135 @@
+import unittest
+from unittest import mock
+
+import lib.bitcoin as bitcoin
+import lib.keystore as keystore
+import lib.storage as storage
+import lib.wallet as wallet
+
+
+# TODO: 2fa
+# TODO: bip39 (standard, segwit_p2sh)
+class TestWalletKeystoreAddressIntegrity(unittest.TestCase):
+
+ def _check_seeded_keystore_sanity(self, ks):
+ self.assertTrue (ks.is_deterministic())
+ self.assertFalse(ks.is_watching_only())
+ self.assertFalse(ks.can_import())
+ self.assertTrue (ks.has_seed())
+
+ def _check_xpub_keystore_sanity(self, ks):
+ self.assertTrue (ks.is_deterministic())
+ self.assertTrue (ks.is_watching_only())
+ self.assertFalse(ks.can_import())
+ self.assertFalse(ks.has_seed())
+
+ @mock.patch.object(storage.WalletStorage, '_write')
+ def test_electrum_seed_standard(self, mock_write):
+ seed_words = 'cycle rocket west magnet parrot shuffle foot correct salt library feed song'
+ self.assertEqual(bitcoin.seed_type(seed_words), 'standard')
+
+ ks = keystore.from_seed(seed_words, '')
+
+ self._check_seeded_keystore_sanity(ks)
+ self.assertTrue(isinstance(ks, keystore.BIP32_KeyStore))
+
+ self.assertEqual(ks.xpub, 'xpub661MyMwAqRbcFWohJWt7PHsFEJfZAvw9ZxwQoDa4SoMgsDDM1T7WK3u9E4edkC4ugRnZ8E4xDZRpk8Rnts3Nbt97dPwT52CwBdDWroaZf8U')
+
+ store = storage.WalletStorage('if_this_exists_mocking_failed_648151893')
+ store.put('keystore', ks.dump())
+ w = wallet.Standard_Wallet(store)
+ w.synchronize()
+
+ self.assertEqual(w.get_receiving_addresses()[0], '1NNkttn1YvVGdqBW4PR6zvc3Zx3H5owKRf')
+ self.assertEqual(w.get_change_addresses()[0], '1KSezYMhAJMWqFbVFB2JshYg69UpmEXR4D')
+
+ @mock.patch.object(storage.WalletStorage, '_write')
+ def test_electrum_seed_segwit(self, mock_write):
+ seed_words = 'bitter grass shiver impose acquire brush forget axis eager alone wine silver'
+ self.assertEqual(bitcoin.seed_type(seed_words), 'segwit')
+
+ ks = keystore.from_seed(seed_words, '')
+
+ self._check_seeded_keystore_sanity(ks)
+ self.assertTrue(isinstance(ks, keystore.BIP32_KeyStore))
+
+ self.assertEqual(ks.xpub, 'zpub6jftahH18ngZyLeqfLBFAm7YaWFVttE9pku5pNMX2qPzTjoq1FVgZMmhjecyB2nqFb31gHE9vNvbaggU6vvWpNZbXEWLLUjYjFqG95LNyT8')
+
+ store = storage.WalletStorage('if_this_exists_mocking_failed_648151893')
+ store.put('keystore', ks.dump())
+ w = wallet.Standard_Wallet(store)
+ w.synchronize()
+
+ self.assertEqual(w.get_receiving_addresses()[0], 'bc1qtt5msqvcqyvuu7hq7urwgraqqyq2yhtuzdpwc4')
+ self.assertEqual(w.get_change_addresses()[0], 'bc1q9wlrynvj7qz7x4fs29d8dnje0zdevj5vmenter')
+
+ @mock.patch.object(storage.WalletStorage, '_write')
+ def test_electrum_seed_old(self, mock_write):
+ seed_words = 'powerful random nobody notice nothing important anyway look away hidden message over'
+ self.assertEqual(bitcoin.seed_type(seed_words), 'old')
+
+ ks = keystore.from_seed(seed_words, '')
+
+ self._check_seeded_keystore_sanity(ks)
+ self.assertTrue(isinstance(ks, keystore.Old_KeyStore))
+
+ self.assertEqual(ks.mpk, 'e9d4b7866dd1e91c862aebf62a49548c7dbf7bcc6e4b7b8c9da820c7737968df9c09d5a3e271dc814a29981f81b3faaf2737b551ef5dcc6189cf0f8252c442b3')
+
+ store = storage.WalletStorage('if_this_exists_mocking_failed_648151893')
+ store.put('keystore', ks.dump())
+ w = wallet.Standard_Wallet(store)
+ w.synchronize()
+
+ self.assertEqual(w.get_receiving_addresses()[0], '1FJEEB8ihPMbzs2SkLmr37dHyRFzakqUmo')
+ self.assertEqual(w.get_change_addresses()[0], '1KRW8pH6HFHZh889VDq6fEKvmrsmApwNfe')
+
+ @mock.patch.object(storage.WalletStorage, '_write')
+ def test_electrum_multisig_seed_standard(self, mock_write):
+ seed_words = 'blast uniform dragon fiscal ensure vast young utility dinosaur abandon rookie sure'
+ self.assertEqual(bitcoin.seed_type(seed_words), 'standard')
+
+ ks1 = keystore.from_seed(seed_words, '')
+ self._check_seeded_keystore_sanity(ks1)
+ self.assertTrue(isinstance(ks1, keystore.BIP32_KeyStore))
+ self.assertEqual(ks1.xpub, 'xpub661MyMwAqRbcGNEPu3aJQqXTydqR9t49Tkwb4Esrj112kw8xLthv8uybxvaki4Ygt9xiwZUQGeFTG7T2TUzR3eA4Zp3aq5RXsABHFBUrq4c')
+
+ ks2 = keystore.from_xpub('xpub661MyMwAqRbcGfCPEkkyo5WmcrhTq8mi3xuBS7VEZ3LYvsgY1cCFDbenT33bdD12axvrmXhuX3xkAbKci3yZY9ZEk8vhLic7KNhLjqdh5ec')
+ self._check_xpub_keystore_sanity(ks2)
+ self.assertTrue(isinstance(ks2, keystore.BIP32_KeyStore))
+
+ store = storage.WalletStorage('if_this_exists_mocking_failed_648151893')
+ multisig_type = "%dof%d" % (2, 2)
+ store.put('wallet_type', multisig_type)
+ store.put('x%d/' % 1, ks1.dump())
+ store.put('x%d/' % 2, ks2.dump())
+ w = wallet.Multisig_Wallet(store)
+ w.synchronize()
+
+ self.assertEqual(w.get_receiving_addresses()[0], '32ji3QkAgXNz6oFoRfakyD3ys1XXiERQYN')
+ self.assertEqual(w.get_change_addresses()[0], '36XWwEHrrVCLnhjK5MrVVGmUHghr9oWTN1')
+
+ @mock.patch.object(storage.WalletStorage, '_write')
+ def test_electrum_multisig_seed_segwit(self, mock_write):
+ seed_words = 'snow nest raise royal more walk demise rotate smooth spirit canyon gun'
+ self.assertEqual(bitcoin.seed_type(seed_words), 'segwit')
+
+ ks1 = keystore.from_seed(seed_words, '')
+ self._check_seeded_keystore_sanity(ks1)
+ self.assertTrue(isinstance(ks1, keystore.BIP32_KeyStore))
+ self.assertEqual(ks1.xpub, 'zpub6jftahH18ngZxwy83eiaWSH1ynYTbA4Ta5MR6JQb4TZCJLCbEzY15f6BCpiDtQeFkzni3v4tT5x6x6Lanvg1YbMZ7ePQmjqtbznPUcYA6mK')
+
+ ks2 = keystore.from_xpub('zpub6jftahH18ngZxE47EYpJWzwTF71AhtNW8ToREBZB7mu3BMdud8aG1tw5TtSqY4qPrhFK1NpKvYwJi9mnhLE8p57nx6929YguJ1Sf2VB8VGt')
+ self._check_xpub_keystore_sanity(ks2)
+ self.assertTrue(isinstance(ks2, keystore.BIP32_KeyStore))
+
+ store = storage.WalletStorage('if_this_exists_mocking_failed_648151893')
+ multisig_type = "%dof%d" % (2, 2)
+ store.put('wallet_type', multisig_type)
+ store.put('x%d/' % 1, ks1.dump())
+ store.put('x%d/' % 2, ks2.dump())
+ w = wallet.Multisig_Wallet(store)
+ w.synchronize()
+
+ self.assertEqual(w.get_receiving_addresses()[0], 'bc1qnvks7gfdu72de8qv6q6rhkkzu70fqz4wpjzuxjf6aydsx7wxfwcqnlxuv3')
+ self.assertEqual(w.get_change_addresses()[0], 'bc1qsvfq6ekp0paugjhfey38pt3nqyvs3tcxu5l00v56j3g6g5la004qw4d33a')
+