mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-03-05 14:06:27 -05:00
fa4595deb3
This is needed for the next commit. Also, it doesn't really test anything novel. wait_for_debug_log is inherently racy, so will randomly terminate at the exact point or later. So the randomization is already sufficiently covered by the existing test.
131 lines
4.7 KiB
Python
Executable file
131 lines
4.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# Copyright (c) 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.
|
|
"""Stress tests related to node initialization."""
|
|
import random
|
|
import time
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from test_framework.test_framework import BitcoinTestFramework, SkipTest
|
|
from test_framework.test_node import ErrorMatch
|
|
from test_framework.util import assert_equal
|
|
|
|
|
|
class InitStressTest(BitcoinTestFramework):
|
|
"""
|
|
Ensure that initialization can be interrupted at a number of points and not impair
|
|
subsequent starts.
|
|
"""
|
|
|
|
def set_test_params(self):
|
|
self.setup_clean_chain = False
|
|
self.num_nodes = 1
|
|
|
|
def run_test(self):
|
|
"""
|
|
- test terminating initialization after seeing a certain log line.
|
|
- test terminating init after seeing a random number of log lines.
|
|
- test removing certain essential files to test startup error paths.
|
|
"""
|
|
# TODO: skip Windows for now since it isn't clear how to SIGTERM.
|
|
#
|
|
# Windows doesn't support `process.terminate()`.
|
|
# and other approaches (like below) don't work:
|
|
#
|
|
# os.kill(node.process.pid, signal.CTRL_C_EVENT)
|
|
if os.name == 'nt':
|
|
raise SkipTest("can't SIGTERM on Windows")
|
|
|
|
self.stop_node(0)
|
|
node = self.nodes[0]
|
|
|
|
def sigterm_node():
|
|
node.process.terminate()
|
|
node.process.wait()
|
|
|
|
def check_clean_start():
|
|
"""Ensure that node restarts successfully after various interrupts."""
|
|
node.start()
|
|
node.wait_for_rpc_connection()
|
|
assert_equal(200, node.getblockcount())
|
|
|
|
lines_to_terminate_after = [
|
|
'Validating signatures for all blocks',
|
|
'scheduler thread start',
|
|
'Starting HTTP server',
|
|
'Loading P2P addresses',
|
|
'Loading banlist',
|
|
'Loading block index',
|
|
'Switching active chainstate',
|
|
'Checking all blk files are present',
|
|
'Loaded best chain:',
|
|
'init message: Verifying blocks',
|
|
'init message: Starting network threads',
|
|
'net thread start',
|
|
'addcon thread start',
|
|
'loadblk thread start',
|
|
'txindex thread start',
|
|
'msghand thread start',
|
|
'net thread start',
|
|
'addcon thread start',
|
|
]
|
|
if self.is_wallet_compiled():
|
|
lines_to_terminate_after.append('Verifying wallet')
|
|
|
|
for terminate_line in lines_to_terminate_after:
|
|
self.log.info(f"Starting node and will exit after line '{terminate_line}'")
|
|
node.start(extra_args=['-txindex=1'])
|
|
|
|
num_total_logs = node.wait_for_debug_log([terminate_line], ignore_case=True)
|
|
self.log.debug(f"Terminating node after {num_total_logs} log lines seen")
|
|
sigterm_node()
|
|
|
|
check_clean_start()
|
|
self.stop_node(0)
|
|
|
|
self.log.info("Test startup errors after removing certain essential files")
|
|
|
|
files_to_disturb = {
|
|
'blocks/index/*.ldb': 'Error opening block database.',
|
|
'chainstate/*.ldb': 'Error opening block database.',
|
|
'blocks/blk*.dat': 'Error loading block database.',
|
|
}
|
|
|
|
for file_patt, err_fragment in files_to_disturb.items():
|
|
target_files = list(node.chain_path.glob(file_patt))
|
|
|
|
for target_file in target_files:
|
|
self.log.info(f"Tweaking file to ensure failure {target_file}")
|
|
bak_path = str(target_file) + ".bak"
|
|
target_file.rename(bak_path)
|
|
|
|
# TODO: at some point, we should test perturbing the files instead of removing
|
|
# them, e.g.
|
|
#
|
|
# contents = target_file.read_bytes()
|
|
# tweaked_contents = bytearray(contents)
|
|
# tweaked_contents[50:250] = b'1' * 200
|
|
# target_file.write_bytes(bytes(tweaked_contents))
|
|
#
|
|
# At the moment I can't get this to work (bitcoind loads successfully?) so
|
|
# investigate doing this later.
|
|
|
|
node.assert_start_raises_init_error(
|
|
extra_args=['-txindex=1'],
|
|
expected_msg=err_fragment,
|
|
match=ErrorMatch.PARTIAL_REGEX,
|
|
)
|
|
|
|
for target_file in target_files:
|
|
bak_path = str(target_file) + ".bak"
|
|
self.log.debug(f"Restoring file from {bak_path} and restarting")
|
|
Path(bak_path).rename(target_file)
|
|
|
|
check_clean_start()
|
|
self.stop_node(0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
InitStressTest().main()
|