text: #!/usr/bin/env python3
  # Copyright (c) 2017-2018 The Bitcoin Core developers
  # Copyright (c) 2010-2021 The Freicoin Developers
  #
  # This program is free software: you can redistribute it and/or modify it under
  # the terms of version 3 of the GNU Affero General Public License as published
  # by the Free Software Foundation.
  #
  # This program is distributed in the hope that it will be useful, but WITHOUT
  # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  # FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
  # details.
  #
  # You should have received a copy of the GNU Affero General Public License
  # along with this program.  If not, see <https://www.gnu.org/licenses/>.
  """Test HD Wallet keypool restore function.
  
  Two nodes. Node1 is under test. Node0 is providing transactions and generating blocks.
  
  - Start node1, shutdown and backup wallet.
  - Generate 110 keys (enough to drain the keypool). Store key 90 (in the initial keypool) and key 110 (beyond the initial keypool). Send funds to key 90 and key 110.
  - Stop node1, clear the datadir, move wallet file back into the datadir and restart node1.
  - connect node1 to node0. Verify that they sync and node1 receives its funds."""
  import os
  import shutil
  
  from test_framework.test_framework import FreicoinTestFramework
  from test_framework.util import (
      assert_equal,
      connect_nodes_bi,
      sync_blocks,
  )
  
  
  class KeypoolRestoreTest(FreicoinTestFramework):
      def set_test_params(self):
          self.setup_clean_chain = True
          self.num_nodes = 4
          self.extra_args = [[], ['-keypool=100'], ['-keypool=100'], ['-keypool=100']]
  
      def skip_test_if_missing_module(self):
          self.skip_if_no_wallet()
  
      def run_test(self):
          wallet_path = os.path.join(self.nodes[1].datadir, "regtest", "wallets", "wallet.dat")
          wallet_backup_path = os.path.join(self.nodes[1].datadir, "wallet.bak")
          self.nodes[0].generate(101)
  
          self.log.info("Make backup of wallet")
          self.stop_node(1)
          shutil.copyfile(wallet_path, wallet_backup_path)
          self.start_node(1, self.extra_args[1])
          connect_nodes_bi(self.nodes, 0, 1)
          connect_nodes_bi(self.nodes, 0, 2)
          connect_nodes_bi(self.nodes, 0, 3)
  
          for i, output_type in enumerate(["legacy", "bech32"]):
  
              self.log.info("Generate keys for wallet with address type: {}".format(output_type))
              idx = i+1
              for _ in range(90):
                  addr_oldpool = self.nodes[idx].getnewaddress(address_type=output_type)
              for _ in range(20):
                  addr_extpool = self.nodes[idx].getnewaddress(address_type=output_type)
  
              # Make sure we're creating the outputs we expect
              address_details = self.nodes[idx].validateaddress(addr_extpool)
              if i == 0:
                  assert(not address_details["isscript"] and not address_details["iswitness"])
              else:
                  assert(address_details["isscript"] and address_details["iswitness"])
  
  
              self.log.info("Send funds to wallet")
              self.nodes[0].sendtoaddress(addr_oldpool, 10)
              self.nodes[0].generate(1)
              self.nodes[0].sendtoaddress(addr_extpool, 5)
              self.nodes[0].generate(1)
              sync_blocks(self.nodes)
  
              self.log.info("Restart node with wallet backup")
              self.stop_node(idx)
              shutil.copyfile(wallet_backup_path, wallet_path)
              self.start_node(idx, self.extra_args[idx])
              connect_nodes_bi(self.nodes, 0, idx)
              self.sync_all()
  
              self.log.info("Verify keypool is restored and balance is correct")
              assert_equal(self.nodes[idx].getbalance(), 15)
              assert_equal(self.nodes[idx].listtransactions()[0]['category'], "receive")
              # Check that we have marked all keys up to the used keypool key as used
              assert_equal(self.nodes[idx].getaddressinfo(self.nodes[idx].getnewaddress())['hdkeypath'], "m/0'/0'/110'")
  
  
  if __name__ == '__main__':
      KeypoolRestoreTest().main()
  