mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-01-11 04:12:36 -03:00
aaaa3aefbd
Seems odd to place the burden on test writers to hardcode the chain or datadir path for the nodes under test.
83 lines
3.4 KiB
Python
Executable file
83 lines
3.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# Copyright (c) 2017-2021 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 loadblock option
|
|
|
|
Test the option to start a node with the option loadblock which loads
|
|
a serialized blockchain from a file (usually called bootstrap.dat).
|
|
To generate that file this test uses the helper scripts available
|
|
in contrib/linearize.
|
|
"""
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import urllib
|
|
|
|
from test_framework.blocktools import COINBASE_MATURITY
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import assert_equal
|
|
|
|
|
|
class LoadblockTest(BitcoinTestFramework):
|
|
def set_test_params(self):
|
|
self.setup_clean_chain = True
|
|
self.num_nodes = 2
|
|
self.supports_cli = False
|
|
|
|
def run_test(self):
|
|
self.nodes[1].setnetworkactive(state=False)
|
|
self.generate(self.nodes[0], COINBASE_MATURITY, sync_fun=self.no_op)
|
|
|
|
# Parsing the url of our node to get settings for config file
|
|
data_dir = self.nodes[0].datadir
|
|
node_url = urllib.parse.urlparse(self.nodes[0].url)
|
|
cfg_file = os.path.join(data_dir, "linearize.cfg")
|
|
bootstrap_file = os.path.join(self.options.tmpdir, "bootstrap.dat")
|
|
genesis_block = self.nodes[0].getblockhash(0)
|
|
blocks_dir = self.nodes[0].chain_path / "blocks"
|
|
hash_list = tempfile.NamedTemporaryFile(dir=data_dir,
|
|
mode='w',
|
|
delete=False,
|
|
encoding="utf-8")
|
|
|
|
self.log.info("Create linearization config file")
|
|
with open(cfg_file, "a", encoding="utf-8") as cfg:
|
|
cfg.write(f"datadir={data_dir}\n")
|
|
cfg.write(f"rpcuser={node_url.username}\n")
|
|
cfg.write(f"rpcpassword={node_url.password}\n")
|
|
cfg.write(f"port={node_url.port}\n")
|
|
cfg.write(f"host={node_url.hostname}\n")
|
|
cfg.write(f"output_file={bootstrap_file}\n")
|
|
cfg.write(f"max_height=100\n")
|
|
cfg.write(f"netmagic=fabfb5da\n")
|
|
cfg.write(f"input={blocks_dir}\n")
|
|
cfg.write(f"genesis={genesis_block}\n")
|
|
cfg.write(f"hashlist={hash_list.name}\n")
|
|
|
|
base_dir = self.config["environment"]["SRCDIR"]
|
|
linearize_dir = os.path.join(base_dir, "contrib", "linearize")
|
|
|
|
self.log.info("Run linearization of block hashes")
|
|
linearize_hashes_file = os.path.join(linearize_dir, "linearize-hashes.py")
|
|
subprocess.run([sys.executable, linearize_hashes_file, cfg_file],
|
|
stdout=hash_list,
|
|
check=True)
|
|
|
|
self.log.info("Run linearization of block data")
|
|
linearize_data_file = os.path.join(linearize_dir, "linearize-data.py")
|
|
subprocess.run([sys.executable, linearize_data_file, cfg_file],
|
|
check=True)
|
|
|
|
self.log.info("Restart second, unsynced node with bootstrap file")
|
|
self.restart_node(1, extra_args=[f"-loadblock={bootstrap_file}"])
|
|
assert_equal(self.nodes[1].getblockcount(), 100) # start_node is blocking on all block files being imported
|
|
|
|
assert_equal(self.nodes[1].getblockchaininfo()['blocks'], 100)
|
|
assert_equal(self.nodes[0].getbestblockhash(), self.nodes[1].getbestblockhash())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
LoadblockTest().main()
|