From 893849ab39780dbea84f8537b3dc7fb694c01862 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Mon, 24 Aug 2015 09:42:19 +0300
Subject: [PATCH] test_account: add test case for BIP32_Account

---
 lib/tests/test_account.py | 69 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 lib/tests/test_account.py

diff --git a/lib/tests/test_account.py b/lib/tests/test_account.py
new file mode 100644
index 00000000..4c01eda4
--- /dev/null
+++ b/lib/tests/test_account.py
@@ -0,0 +1,69 @@
+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])