text: #!/usr/bin/env python3
  # Copyright (c) 2016-2017 The Crowncoin Core developers
  # Distributed under the MIT software license, see the accompanying
  # file COPYING or http://www.opensource.org/licenses/mit-license.php.
  """Test Hierarchical Deterministic wallet function."""
  
  from test_framework.test_framework import CrowncoinTestFramework
  from test_framework.util import (
      assert_equal,
      connect_nodes_bi,
  )
  import shutil
  import os
  
  class WalletHDTest(CrowncoinTestFramework):
      def set_test_params(self):
          self.setup_clean_chain = True
          self.num_nodes = 2
          self.extra_args = [[], ['-keypool=0']]
  
      def run_test (self):
          tmpdir = self.options.tmpdir
  
          # Make sure can't switch off usehd after wallet creation
          self.stop_node(1)
          self.assert_start_raises_init_error(1, ['-usehd=0'], 'already existing HD wallet')
          self.start_node(1)
          connect_nodes_bi(self.nodes, 0, 1)
  
          # Make sure we use hd, keep masterkeyid
          masterkeyid = self.nodes[1].getwalletinfo()['hdmasterkeyid']
          assert_equal(len(masterkeyid), 40)
  
          # create an internal key
          change_addr = self.nodes[1].getrawchangeaddress()
          change_addrV= self.nodes[1].validateaddress(change_addr)
          assert_equal(change_addrV["hdkeypath"], "m/0'/1'/0'") #first internal child key
  
          # Import a non-HD private key in the HD wallet
          non_hd_add = self.nodes[0].getnewaddress()
          self.nodes[1].importprivkey(self.nodes[0].dumpprivkey(non_hd_add))
  
          # This should be enough to keep the master key and the non-HD key
          self.nodes[1].backupwallet(tmpdir + "/hd.bak")
          #self.nodes[1].dumpwallet(tmpdir + "/hd.dump")
  
          # Derive some HD addresses and remember the last
          # Also send funds to each add
          self.nodes[0].generate(101)
          hd_add = None
          num_hd_adds = 300
          for i in range(num_hd_adds):
              hd_add = self.nodes[1].getnewaddress()
              hd_info = self.nodes[1].validateaddress(hd_add)
              assert_equal(hd_info["hdkeypath"], "m/0'/0'/"+str(i)+"'")
              assert_equal(hd_info["hdmasterkeyid"], masterkeyid)
              self.nodes[0].sendtoaddress(hd_add, 1)
              self.nodes[0].generate(1)
          self.nodes[0].sendtoaddress(non_hd_add, 1)
          self.nodes[0].generate(1)
  
          # create an internal key (again)
          change_addr = self.nodes[1].getrawchangeaddress()
          change_addrV= self.nodes[1].validateaddress(change_addr)
          assert_equal(change_addrV["hdkeypath"], "m/0'/1'/1'") #second internal child key
  
          self.sync_all()
          assert_equal(self.nodes[1].getbalance(), num_hd_adds + 1)
  
          self.log.info("Restore backup ...")
          self.stop_node(1)
          # we need to delete the complete regtest directory
          # otherwise node1 would auto-recover all funds in flag the keypool keys as used
          shutil.rmtree(os.path.join(tmpdir, "node1/regtest/blocks"))
          shutil.rmtree(os.path.join(tmpdir, "node1/regtest/chainstate"))
          shutil.copyfile(os.path.join(tmpdir, "hd.bak"), os.path.join(tmpdir, "node1/regtest/wallets/wallet.dat"))
          self.start_node(1)
  
          # Assert that derivation is deterministic
          hd_add_2 = None
          for _ in range(num_hd_adds):
              hd_add_2 = self.nodes[1].getnewaddress()
              hd_info_2 = self.nodes[1].validateaddress(hd_add_2)
              assert_equal(hd_info_2["hdkeypath"], "m/0'/0'/"+str(_)+"'")
              assert_equal(hd_info_2["hdmasterkeyid"], masterkeyid)
          assert_equal(hd_add, hd_add_2)
          connect_nodes_bi(self.nodes, 0, 1)
          self.sync_all()
  
          # Needs rescan
          self.stop_node(1)
          self.start_node(1, extra_args=self.extra_args[1] + ['-rescan'])
          assert_equal(self.nodes[1].getbalance(), num_hd_adds + 1)
  
          # Try a RPC based rescan
          self.stop_node(1)
          shutil.rmtree(os.path.join(tmpdir, "node1/regtest/blocks"))
          shutil.rmtree(os.path.join(tmpdir, "node1/regtest/chainstate"))
          shutil.copyfile(os.path.join(tmpdir, "hd.bak"), os.path.join(tmpdir, "node1/regtest/wallet.dat"))
          self.start_node(1, extra_args=self.extra_args[1])
          connect_nodes_bi(self.nodes, 0, 1)
          self.sync_all()
          out = self.nodes[1].rescanblockchain(0, 1)
          assert_equal(out['start_height'], 0)
          assert_equal(out['stop_height'], 1)
          out = self.nodes[1].rescanblockchain()
          assert_equal(out['start_height'], 0)
          assert_equal(out['stop_height'], self.nodes[1].getblockcount())
          assert_equal(self.nodes[1].getbalance(), num_hd_adds + 1)
  
          # send a tx and make sure its using the internal chain for the changeoutput
          txid = self.nodes[1].sendtoaddress(self.nodes[0].getnewaddress(), 1)
          outs = self.nodes[1].decoderawtransaction(self.nodes[1].gettransaction(txid)['hex'])['vout']
          keypath = ""
          for out in outs:
              if out['value'] != 1:
                  keypath = self.nodes[1].validateaddress(out['scriptPubKey']['addresses'][0])['hdkeypath']
  
          assert_equal(keypath[0:7], "m/0'/1'")
  
  if __name__ == '__main__':
      WalletHDTest().main ()
  