text: #!/usr/bin/env python3
  # Copyright (c) 2014-2016 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.address import key_to_p2pkh
  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(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_p2pkh(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, 100000)
          self.nodes[0].generate(1)
          rawtxn1 = self.nodes[0].gettransaction(txnid1)['hex']
          proof1 = self.nodes[0].gettxoutproof([txnid1])
  
          txnid2 = self.nodes[0].sendtoaddress(address2, 50000)
          self.nodes[0].generate(1)
          rawtxn2 = self.nodes[0].gettransaction(txnid2)['hex']
          proof2 = self.nodes[0].gettxoutproof([txnid2])
  
          txnid3 = self.nodes[0].sendtoaddress(address3, 25000)
          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('25000'))
  
          # 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()
  