commit 68d891941667f707474198b024d7d723ac0de405
parent e13183ea7a4d43c861a973f4792980d637005074
Author: SomberNight <somber.night@protonmail.com>
Date: Fri, 27 Apr 2018 21:43:46 +0200
tests: sends between wallets
Diffstat:
1 file changed, 214 insertions(+), 11 deletions(-)
diff --git a/lib/tests/test_wallet_vertical.py b/lib/tests/test_wallet_vertical.py
@@ -1,11 +1,11 @@
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
-from lib import constants
+import lib
+from lib import storage, bitcoin, keystore, constants
+from lib.transaction import Transaction
+from lib.simple_config import SimpleConfig
+from lib.wallet import TX_HEIGHT_UNCONFIRMED, TX_HEIGHT_UNCONF_PARENT
from plugins.trustedcoin import trustedcoin
@@ -31,16 +31,16 @@ class WalletIntegrityHelper:
test_obj.assertFalse(ks.has_seed())
@classmethod
- def create_standard_wallet(cls, ks):
+ def create_standard_wallet(cls, ks, gap_limit=None):
store = storage.WalletStorage('if_this_exists_mocking_failed_648151893')
store.put('keystore', ks.dump())
- store.put('gap_limit', cls.gap_limit)
- w = wallet.Standard_Wallet(store)
+ store.put('gap_limit', gap_limit or cls.gap_limit)
+ w = lib.wallet.Standard_Wallet(store)
w.synchronize()
return w
@classmethod
- def create_multisig_wallet(cls, ks1, ks2, ks3=None):
+ def create_multisig_wallet(cls, ks1, ks2, ks3=None, gap_limit=None):
"""Creates a 2-of-2 or 2-of-3 multisig wallet."""
store = storage.WalletStorage('if_this_exists_mocking_failed_648151893')
store.put('x%d/' % 1, ks1.dump())
@@ -51,8 +51,8 @@ class WalletIntegrityHelper:
multisig_type = "%dof%d" % (2, 3)
store.put('x%d/' % 3, ks3.dump())
store.put('wallet_type', multisig_type)
- store.put('gap_limit', cls.gap_limit)
- w = wallet.Multisig_Wallet(store)
+ store.put('gap_limit', gap_limit or cls.gap_limit)
+ w = lib.wallet.Multisig_Wallet(store)
w.synchronize()
return w
@@ -314,3 +314,206 @@ class TestWalletKeystoreAddressIntegrityForTestnet(TestCaseForTestnet):
self.assertEqual(w.get_receiving_addresses()[0], '2MzsfTfTGomPRne6TkctMmoDj6LwmVkDrMt')
self.assertEqual(w.get_change_addresses()[0], '2NFp9w8tbYYP9Ze2xQpeYBJQjx3gbXymHX7')
+
+
+class TestWalletSending(TestCaseForTestnet):
+
+ config = SimpleConfig()
+
+ def create_standard_wallet_from_seed(self, seed_words):
+ ks = keystore.from_seed(seed_words, '', False)
+ return WalletIntegrityHelper.create_standard_wallet(ks, gap_limit=2)
+
+ @mock.patch.object(storage.WalletStorage, '_write')
+ def test_sending_between_p2wpkh_and_compressed_p2pkh(self, mock_write):
+ wallet1 = self.create_standard_wallet_from_seed('bitter grass shiver impose acquire brush forget axis eager alone wine silver')
+ wallet2 = self.create_standard_wallet_from_seed('cycle rocket west magnet parrot shuffle foot correct salt library feed song')
+
+ # bootstrap wallet1
+ funding_tx = Transaction('01000000014576dacce264c24d81887642b726f5d64aa7825b21b350c7b75a57f337da6845010000006b483045022100a3f8b6155c71a98ad9986edd6161b20d24fad99b6463c23b463856c0ee54826d02200f606017fd987696ebbe5200daedde922eee264325a184d5bbda965ba5160821012102e5c473c051dae31043c335266d0ef89c1daab2f34d885cc7706b267f3269c609ffffffff0240420f00000000001600148a28bddb7f61864bdcf58b2ad13d5aeb3abc3c42a2ddb90e000000001976a914c384950342cb6f8df55175b48586838b03130fad88ac00000000')
+ funding_txid = funding_tx.txid()
+ funding_output_value = 1000000
+ self.assertEqual('add2535aedcbb5ba79cc2260868bb9e57f328738ca192937f2c92e0e94c19203', funding_txid)
+ wallet1.receive_tx_callback(funding_txid, funding_tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet1 -> wallet2
+ outputs = [(bitcoin.TYPE_ADDRESS, wallet2.get_receiving_address(), 250000)]
+ tx = wallet1.mktx(outputs=outputs, password=None, config=self.config, fee=5000)
+
+ self.assertTrue(tx.is_complete())
+ self.assertTrue(tx.is_segwit())
+ self.assertEqual(1, len(tx.inputs()))
+ tx_copy = Transaction(tx.serialize())
+ self.assertEqual(wallet1.is_mine(tx.inputs()[0]['address']), wallet1.is_mine(tx_copy.inputs()[0]['address']))
+ self.assertTrue(wallet1.is_mine(tx.inputs()[0]['address']))
+ self.assertEqual(wallet1.txin_type, tx_copy.inputs()[0]['type'])
+ self.assertEqual(tx.wtxid(), tx_copy.wtxid())
+ self.assertEqual('3c06ae4d9be8226a472b3e7f7c127c7e3016f525d658d26106b80b4c7e3228e2', tx_copy.txid())
+
+ wallet1.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED) # TX_HEIGHT_UNCONF_PARENT but nvm
+ wallet2.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet2 -> wallet1
+ outputs = [(bitcoin.TYPE_ADDRESS, wallet1.get_receiving_address(), 100000)]
+ tx = wallet2.mktx(outputs=outputs, password=None, config=self.config, fee=5000)
+
+ self.assertTrue(tx.is_complete())
+ self.assertFalse(tx.is_segwit())
+ self.assertEqual(1, len(tx.inputs()))
+ tx_copy = Transaction(tx.serialize())
+ self.assertEqual(wallet2.is_mine(tx.inputs()[0]['address']), wallet2.is_mine(tx_copy.inputs()[0]['address']))
+ self.assertTrue(wallet2.is_mine(tx.inputs()[0]['address']))
+ self.assertEqual(wallet2.txin_type, tx_copy.inputs()[0]['type'])
+ self.assertEqual(tx.wtxid(), tx_copy.wtxid())
+ self.assertEqual('5f25707571eb776bdf14142f9966bf2a681906e0a79501edbb99a972c2ceb972', tx_copy.txid())
+
+ wallet1.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+ wallet2.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet level checks
+ self.assertEqual((0, funding_output_value - 250000 - 5000 + 100000, 0), wallet1.get_balance())
+ self.assertEqual((0, 250000 - 5000 - 100000, 0), wallet2.get_balance())
+
+ @mock.patch.object(storage.WalletStorage, '_write')
+ def test_sending_between_p2sh_2of3_and_uncompressed_p2pkh(self, mock_write):
+ wallet1a = WalletIntegrityHelper.create_multisig_wallet(
+ keystore.from_seed('blast uniform dragon fiscal ensure vast young utility dinosaur abandon rookie sure', '', True),
+ keystore.from_xpub('tpubD6NzVbkrYhZ4YTPEgwk4zzr8wyo7pXGmbbVUnfYNtx6SgAMF5q3LN3Kch58P9hxGNsTmP7Dn49nnrmpE6upoRb1Xojg12FGLuLHkVpVtS44'),
+ keystore.from_xpub('tpubD6NzVbkrYhZ4XJzYkhsCbDCcZRmDAKSD7bXi9mdCni7acVt45fxbTVZyU6jRGh29ULKTjoapkfFsSJvQHitcVKbQgzgkkYsAmaovcro7Mhf'),
+ gap_limit=2
+ )
+ wallet1b = WalletIntegrityHelper.create_multisig_wallet(
+ keystore.from_seed('cycle rocket west magnet parrot shuffle foot correct salt library feed song', '', True),
+ keystore.from_xpub('tpubD6NzVbkrYhZ4YTPEgwk4zzr8wyo7pXGmbbVUnfYNtx6SgAMF5q3LN3Kch58P9hxGNsTmP7Dn49nnrmpE6upoRb1Xojg12FGLuLHkVpVtS44'),
+ keystore.from_xpub('tpubD6NzVbkrYhZ4YARFMEZPckrqJkw59GZD1PXtQnw14ukvWDofR7Z1HMeSCxfYEZVvg4VdZ8zGok5VxHwdrLqew5cMdQntWc5mT7mh1CSgrnX'),
+ gap_limit=2
+ )
+ # ^ third seed: ghost into match ivory badge robot record tackle radar elbow traffic loud
+ wallet2 = self.create_standard_wallet_from_seed('powerful random nobody notice nothing important anyway look away hidden message over')
+
+ # bootstrap wallet1
+ funding_tx = Transaction('010000000001014121f99dc02f0364d2dab3d08905ff4c36fc76c55437fd90b769c35cc18618280100000000fdffffff02d4c22d00000000001600143fd1bc5d32245850c8cb5be5b09c73ccbb9a0f75001bb7000000000017a91480c2353f6a7bc3c71e99e062655b19adb3dd2e4887024830450221008781c78df0c9d4b5ea057333195d5d76bc29494d773f14fa80e27d2f288b2c360220762531614799b6f0fb8d539b18cb5232ab4253dd4385435157b28a44ff63810d0121033de77d21926e09efd04047ae2d39dbd3fb9db446e8b7ed53e0f70f9c9478f735dac11300')
+ funding_txid = funding_tx.txid()
+ funding_output_value = 12000000
+ self.assertEqual('b25cd55687c9e528c2cfd546054f35fb6741f7cf32d600f07dfecdf2e1d42071', funding_txid)
+ wallet1a.receive_tx_callback(funding_txid, funding_tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet1 -> wallet2
+ outputs = [(bitcoin.TYPE_ADDRESS, wallet2.get_receiving_address(), 370000)]
+ tx = wallet1a.mktx(outputs=outputs, password=None, config=self.config, fee=5000)
+ tx = Transaction(tx.serialize()) # simulates moving partial txn between cosigners
+ self.assertFalse(tx.is_complete())
+ wallet1b.sign_transaction(tx, password=None)
+
+ self.assertTrue(tx.is_complete())
+ self.assertFalse(tx.is_segwit())
+ self.assertEqual(1, len(tx.inputs()))
+ tx_copy = Transaction(tx.serialize())
+ self.assertEqual(wallet1a.is_mine(tx.inputs()[0]['address']), wallet1a.is_mine(tx_copy.inputs()[0]['address']))
+ self.assertTrue(wallet1a.is_mine(tx.inputs()[0]['address']))
+ self.assertEqual(wallet1a.txin_type, tx_copy.inputs()[0]['type'])
+ self.assertEqual(tx.wtxid(), tx_copy.wtxid())
+ self.assertEqual('26f3bdd0402e1cff19126244ebe3d32722cef0db507c7229ca8754f5e06ef25d', tx_copy.txid())
+
+ wallet1a.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+ wallet2.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet2 -> wallet1
+ outputs = [(bitcoin.TYPE_ADDRESS, wallet1a.get_receiving_address(), 100000)]
+ tx = wallet2.mktx(outputs=outputs, password=None, config=self.config, fee=5000)
+
+ self.assertTrue(tx.is_complete())
+ self.assertFalse(tx.is_segwit())
+ self.assertEqual(1, len(tx.inputs()))
+ tx_copy = Transaction(tx.serialize())
+ self.assertEqual(wallet2.is_mine(tx.inputs()[0]['address']), wallet2.is_mine(tx_copy.inputs()[0]['address']))
+ self.assertTrue(wallet2.is_mine(tx.inputs()[0]['address']))
+ self.assertEqual(wallet2.txin_type, tx_copy.inputs()[0]['type'])
+ self.assertEqual(tx.wtxid(), tx_copy.wtxid())
+ self.assertEqual('c573b3f8464a4ed40dfc79d0889a780f44e917beef7a75883b2427c2987f3e95', tx_copy.txid())
+
+ wallet1a.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+ wallet2.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet level checks
+ self.assertEqual((0, funding_output_value - 370000 - 5000 + 100000, 0), wallet1a.get_balance())
+ self.assertEqual((0, 370000 - 5000 - 100000, 0), wallet2.get_balance())
+
+ @mock.patch.object(storage.WalletStorage, '_write')
+ def test_sending_between_p2wsh_2of3_and_p2wsh_p2sh_2of2(self, mock_write):
+ wallet1a = WalletIntegrityHelper.create_multisig_wallet(
+ keystore.from_seed('bitter grass shiver impose acquire brush forget axis eager alone wine silver', '', True),
+ keystore.from_xpub('Vpub5fcdcgEwTJmbmqAktuK8Kyq92fMf7sWkcP6oqAii2tG47dNbfkGEGUbfS9NuZaRywLkHE6EmUksrqo32ZL3ouLN1HTar6oRiHpDzKMAF1tf'),
+ keystore.from_xpub('Vpub5fjkKyYnvSS4wBuakWTkNvZDaBM2vQ1MeXWq368VJHNr2eT8efqhpmZ6UUkb7s2dwCXv2Vuggjdhk4vZVyiAQTwUftvff73XcUGq2NQmWra'),
+ gap_limit=2
+ )
+ wallet1b = WalletIntegrityHelper.create_multisig_wallet(
+ keystore.from_seed('snow nest raise royal more walk demise rotate smooth spirit canyon gun', '', True),
+ keystore.from_xpub('Vpub5fjkKyYnvSS4wBuakWTkNvZDaBM2vQ1MeXWq368VJHNr2eT8efqhpmZ6UUkb7s2dwCXv2Vuggjdhk4vZVyiAQTwUftvff73XcUGq2NQmWra'),
+ keystore.from_xpub('Vpub5gSKXzxK7FeKQedu2q1z9oJWxqvX72AArW3HSWpEhc8othDH8xMDu28gr7gf17sp492BuJod8Tn7anjvJrKpETwqnQqX7CS8fcYyUtedEMk'),
+ gap_limit=2
+ )
+ # ^ third seed: hedgehog sunset update estate number jungle amount piano friend donate upper wool
+ wallet2a = WalletIntegrityHelper.create_multisig_wallet(
+ # bip39: finish seminar arrange erosion sunny coil insane together pretty lunch lunch rose, der: m/1234'/1'/0', p2wsh-p2sh multisig
+ keystore.from_xprv('Uprv9CvELvByqm8k2dpecJVjgLMX1z5DufEjY4fBC5YvdGF5WjGCa7GVJJ2fYni1tyuF7Hw83E6W2ZBjAhaFLZv2ri3rEsubkCd5avg4EHKoDBN'),
+ keystore.from_xpub('Upub5Qb8ik4Cnu8g97KLXKgVXHqY6tH8emQvqtBncjSKsyfTZuorPtTZgX7ovKKZHuuVGBVd1MTTBkWez1XXt2weN1sWBz6SfgRPQYEkNgz81QF'),
+ gap_limit=2
+ )
+ wallet2b = WalletIntegrityHelper.create_multisig_wallet(
+ # bip39: square page wood spy oil story rebel give milk screen slide shuffle, der: m/1234'/1'/0', p2wsh-p2sh multisig
+ keystore.from_xprv('Uprv9BbnKEXJxXaNvdEsRJ9VA9toYrSeFJh5UfGBpM2iKe8Uh7UhrM9K8ioL53s8gvCoGfirHHaqpABDAE7VUNw8LNU1DMJKVoWyeNKu9XcDC19'),
+ keystore.from_xpub('Upub5RuakRisg8h3F7u7iL2k3UJFa1uiK7xauHamzTxYBbn4PXbM7eajr6M9Q2VCr6cVGhfhqWQqxnABvtSATuVM1xzxk4nA189jJwzaMn1QX7V'),
+ gap_limit=2
+ )
+
+ # bootstrap wallet1
+ funding_tx = Transaction('01000000000101a41aae475d026c9255200082c7fad26dc47771275b0afba238dccda98a597bd20000000000fdffffff02400d0300000000002200203c43ac80d6e3015cf378bf6bac0c22456723d6050bef324ec641e7762440c63c9dcd410000000000160014824626055515f3ed1d2cfc9152d2e70685c71e8f02483045022100b9f39fad57d07ce1e18251424034f21f10f20e59931041b5167ae343ce973cf602200fefb727fa0ffd25b353f1bcdae2395898fe407b692c62f5885afbf52fa06f5701210301a28f68511ace43114b674371257bb599fd2c686c4b19544870b1799c954b40e9c11300')
+ funding_txid = funding_tx.txid()
+ funding_output_value = 200000
+ self.assertEqual('d2bd6c9d332db8e2c50aa521cd50f963fba214645aab2f7556e061a412103e21', funding_txid)
+ wallet1a.receive_tx_callback(funding_txid, funding_tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet1 -> wallet2
+ outputs = [(bitcoin.TYPE_ADDRESS, wallet2a.get_receiving_address(), 165000)]
+ tx = wallet1a.mktx(outputs=outputs, password=None, config=self.config, fee=5000)
+ tx = Transaction(tx.serialize()) # simulates moving partial txn between cosigners
+ self.assertFalse(tx.is_complete())
+ wallet1b.sign_transaction(tx, password=None)
+
+ self.assertTrue(tx.is_complete())
+ self.assertTrue(tx.is_segwit())
+ self.assertEqual(1, len(tx.inputs()))
+ tx_copy = Transaction(tx.serialize())
+ self.assertEqual(wallet1a.is_mine(tx.inputs()[0]['address']), wallet1a.is_mine(tx_copy.inputs()[0]['address']))
+ self.assertTrue(wallet1a.is_mine(tx.inputs()[0]['address']))
+ self.assertEqual(wallet1a.txin_type, tx_copy.inputs()[0]['type'])
+ self.assertEqual(tx.wtxid(), tx_copy.wtxid())
+ self.assertEqual('6e9c3cd8788bdb970a124ea06136d52bc01cec4f9b1e217627d5e90ebe77d049', tx_copy.txid())
+
+ wallet1a.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+ wallet2a.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet2 -> wallet1
+ outputs = [(bitcoin.TYPE_ADDRESS, wallet1a.get_receiving_address(), 100000)]
+ tx = wallet2a.mktx(outputs=outputs, password=None, config=self.config, fee=5000)
+ tx = Transaction(tx.serialize()) # simulates moving partial txn between cosigners
+ self.assertFalse(tx.is_complete())
+ wallet2b.sign_transaction(tx, password=None)
+
+ self.assertTrue(tx.is_complete())
+ self.assertTrue(tx.is_segwit())
+ self.assertEqual(1, len(tx.inputs()))
+ tx_copy = Transaction(tx.serialize())
+ self.assertEqual(wallet2a.is_mine(tx.inputs()[0]['address']), wallet2a.is_mine(tx_copy.inputs()[0]['address']))
+ self.assertTrue(wallet2a.is_mine(tx.inputs()[0]['address']))
+ self.assertEqual(wallet2a.txin_type, tx_copy.inputs()[0]['type'])
+ self.assertEqual(tx.wtxid(), tx_copy.wtxid())
+ self.assertEqual('84b0dcb43022385f7a10e2710e5625a2be3cd6e390387b6100b55500d5eea8f6', tx_copy.txid())
+
+ wallet1a.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+ wallet2a.receive_tx_callback(tx.txid(), tx, TX_HEIGHT_UNCONFIRMED)
+
+ # wallet level checks
+ self.assertEqual((0, funding_output_value - 165000 - 5000 + 100000, 0), wallet1a.get_balance())
+ self.assertEqual((0, 165000 - 5000 - 100000, 0), wallet2a.get_balance())
