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

#!/usr/bin/env python3# Copyright (c) 2017-2019 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 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 osimport shutilfrom test_framework.test_framework import BitcoinTestFrameworkfrom test_framework.util import (    assert_equal,)class KeypoolRestoreTest(BitcoinTestFramework):    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, self.chain, "wallets", self.default_wallet_name, self.wallet_data_filename)        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])        self.connect_nodes(0, 1)        self.connect_nodes(0, 2)        self.connect_nodes(0, 3)        for i, output_type in enumerate(["legacy", "p2sh-segwit", "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"]            elif i == 1:                assert address_details["isscript"] and not address_details["iswitness"]            else:                assert not 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)            self.sync_blocks()            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])            self.connect_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            if self.options.descriptors:                if output_type == 'legacy':                    assert_equal(self.nodes[idx].getaddressinfo(self.nodes[idx].getnewaddress(address_type=output_type))['hdkeypath'], "m/44'/1'/0'/0/110")                elif output_type == 'p2sh-segwit':                    assert_equal(self.nodes[idx].getaddressinfo(self.nodes[idx].getnewaddress(address_type=output_type))['hdkeypath'], "m/49'/1'/0'/0/110")                elif output_type == 'bech32':                    assert_equal(self.nodes[idx].getaddressinfo(self.nodes[idx].getnewaddress(address_type=output_type))['hdkeypath'], "m/84'/1'/0'/0/110")            else:                assert_equal(self.nodes[idx].getaddressinfo(self.nodes[idx].getnewaddress(address_type=output_type))['hdkeypath'], "m/0'/0'/110'")if __name__ == '__main__':    KeypoolRestoreTest().main()