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.
130 lines
5.1 KiB
130 lines
5.1 KiB
#!/usr/bin/env python3 |
|
# Copyright (c) 2014-2020 The Bitcoin Core developers |
|
# Distributed under the MIT software license, see the accompanying |
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php. |
|
"""Test the importprunedfunds and removeprunedfunds RPCs.""" |
|
from decimal import Decimal |
|
|
|
from test_framework.blocktools import COINBASE_MATURITY |
|
from test_framework.address import key_to_p2wpkh |
|
from test_framework.key import ECKey |
|
from test_framework.test_framework import BitcoinTestFramework |
|
from test_framework.util import ( |
|
assert_equal, |
|
assert_raises_rpc_error, |
|
) |
|
from test_framework.wallet_util import bytes_to_wif |
|
|
|
class ImportPrunedFundsTest(BitcoinTestFramework): |
|
def set_test_params(self): |
|
self.setup_clean_chain = True |
|
self.num_nodes = 2 |
|
|
|
def skip_test_if_missing_module(self): |
|
self.skip_if_no_wallet() |
|
|
|
def run_test(self): |
|
self.log.info("Mining blocks...") |
|
self.nodes[0].generate(COINBASE_MATURITY + 1) |
|
|
|
self.sync_all() |
|
|
|
# address |
|
address1 = self.nodes[0].getnewaddress() |
|
# pubkey |
|
address2 = self.nodes[0].getnewaddress() |
|
# privkey |
|
eckey = ECKey() |
|
eckey.generate() |
|
address3_privkey = bytes_to_wif(eckey.get_bytes()) |
|
address3 = key_to_p2wpkh(eckey.get_pubkey().get_bytes()) |
|
self.nodes[0].importprivkey(address3_privkey) |
|
|
|
# Check only one address |
|
address_info = self.nodes[0].getaddressinfo(address1) |
|
assert_equal(address_info['ismine'], True) |
|
|
|
self.sync_all() |
|
|
|
# Node 1 sync test |
|
assert_equal(self.nodes[1].getblockcount(), COINBASE_MATURITY + 1) |
|
|
|
# Address Test - before import |
|
address_info = self.nodes[1].getaddressinfo(address1) |
|
assert_equal(address_info['iswatchonly'], False) |
|
assert_equal(address_info['ismine'], False) |
|
|
|
address_info = self.nodes[1].getaddressinfo(address2) |
|
assert_equal(address_info['iswatchonly'], False) |
|
assert_equal(address_info['ismine'], False) |
|
|
|
address_info = self.nodes[1].getaddressinfo(address3) |
|
assert_equal(address_info['iswatchonly'], False) |
|
assert_equal(address_info['ismine'], False) |
|
|
|
# Send funds to self |
|
txnid1 = self.nodes[0].sendtoaddress(address1, 0.1) |
|
self.nodes[0].generate(1) |
|
rawtxn1 = self.nodes[0].gettransaction(txnid1)['hex'] |
|
proof1 = self.nodes[0].gettxoutproof([txnid1]) |
|
|
|
txnid2 = self.nodes[0].sendtoaddress(address2, 0.05) |
|
self.nodes[0].generate(1) |
|
rawtxn2 = self.nodes[0].gettransaction(txnid2)['hex'] |
|
proof2 = self.nodes[0].gettxoutproof([txnid2]) |
|
|
|
txnid3 = self.nodes[0].sendtoaddress(address3, 0.025) |
|
self.nodes[0].generate(1) |
|
rawtxn3 = self.nodes[0].gettransaction(txnid3)['hex'] |
|
proof3 = self.nodes[0].gettxoutproof([txnid3]) |
|
|
|
self.sync_all() |
|
|
|
# Import with no affiliated address |
|
assert_raises_rpc_error(-5, "No addresses", self.nodes[1].importprunedfunds, rawtxn1, proof1) |
|
|
|
balance1 = self.nodes[1].getbalance() |
|
assert_equal(balance1, Decimal(0)) |
|
|
|
# Import with affiliated address with no rescan |
|
self.nodes[1].createwallet('wwatch', disable_private_keys=True) |
|
wwatch = self.nodes[1].get_wallet_rpc('wwatch') |
|
wwatch.importaddress(address=address2, rescan=False) |
|
wwatch.importprunedfunds(rawtransaction=rawtxn2, txoutproof=proof2) |
|
assert [tx for tx in wwatch.listtransactions(include_watchonly=True) if tx['txid'] == txnid2] |
|
|
|
# Import with private key with no rescan |
|
w1 = self.nodes[1].get_wallet_rpc(self.default_wallet_name) |
|
w1.importprivkey(privkey=address3_privkey, rescan=False) |
|
w1.importprunedfunds(rawtxn3, proof3) |
|
assert [tx for tx in w1.listtransactions() if tx['txid'] == txnid3] |
|
balance3 = w1.getbalance() |
|
assert_equal(balance3, Decimal('0.025')) |
|
|
|
# Addresses Test - after import |
|
address_info = w1.getaddressinfo(address1) |
|
assert_equal(address_info['iswatchonly'], False) |
|
assert_equal(address_info['ismine'], False) |
|
address_info = wwatch.getaddressinfo(address2) |
|
if self.options.descriptors: |
|
assert_equal(address_info['iswatchonly'], False) |
|
assert_equal(address_info['ismine'], True) |
|
else: |
|
assert_equal(address_info['iswatchonly'], True) |
|
assert_equal(address_info['ismine'], False) |
|
address_info = w1.getaddressinfo(address3) |
|
assert_equal(address_info['iswatchonly'], False) |
|
assert_equal(address_info['ismine'], True) |
|
|
|
# Remove transactions |
|
assert_raises_rpc_error(-8, "Transaction does not exist in wallet.", w1.removeprunedfunds, txnid1) |
|
assert not [tx for tx in w1.listtransactions(include_watchonly=True) if tx['txid'] == txnid1] |
|
|
|
wwatch.removeprunedfunds(txnid2) |
|
assert not [tx for tx in wwatch.listtransactions(include_watchonly=True) if tx['txid'] == txnid2] |
|
|
|
w1.removeprunedfunds(txnid3) |
|
assert not [tx for tx in w1.listtransactions(include_watchonly=True) if tx['txid'] == txnid3] |
|
|
|
if __name__ == '__main__': |
|
ImportPrunedFundsTest().main()
|
|
|