git / moonbite-core / test/functional/wallet_importprunedfunds.py · raw

#!/usr/bin/env python3# Copyright (c) 2014-2018 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 Decimalfrom test_framework.address import key_to_p2wpkhfrom test_framework.key import ECKeyfrom test_framework.test_framework import BitcoinTestFrameworkfrom test_framework.util import (    assert_equal,    assert_raises_rpc_error,)from test_framework.wallet_util import bytes_to_wifclass 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(101)        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(), 101)        # 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()