Changeset View
Changeset View
Standalone View
Standalone View
qa/rpc-tests/forknotify.py
#!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
# Copyright (c) 2014-2016 The Bitcoin Core developers | # Copyright (c) 2014-2016 The Bitcoin Core developers | ||||
# Distributed under the MIT software license, see the accompanying | # Distributed under the MIT software license, see the accompanying | ||||
# file COPYING or http://www.opensource.org/licenses/mit-license.php. | # file COPYING or http://www.opensource.org/licenses/mit-license.php. | ||||
# | # | ||||
# Test -alertnotify | # Test -alertnotify | ||||
# | # | ||||
from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
from test_framework.util import * | from test_framework.util import * | ||||
class ForkNotifyTest(BitcoinTestFramework): | class ForkNotifyTest(BitcoinTestFramework): | ||||
def __init__(self): | def __init__(self): | ||||
super().__init__() | super().__init__() | ||||
self.num_nodes = 2 | self.num_nodes = 2 | ||||
self.setup_clean_chain = False | self.setup_clean_chain = False | ||||
alert_filename = None # Set by setup_network | alert_filename = None # Set by setup_network | ||||
def setup_network(self): | def setup_network(self): | ||||
self.nodes = [] | self.nodes = [] | ||||
self.alert_filename = os.path.join(self.options.tmpdir, "alert.txt") | self.alert_filename = os.path.join(self.options.tmpdir, "alert.txt") | ||||
with open(self.alert_filename, 'w', encoding='utf8'): | with open(self.alert_filename, 'w', encoding='utf8'): | ||||
pass # Just open then close to create zero-length file | pass # Just open then close to create zero-length file | ||||
self.nodes.append(start_node(0, self.options.tmpdir, | self.nodes.append(start_node(0, self.options.tmpdir, | ||||
["-blockversion=2", "-alertnotify=echo %s >> \"" + self.alert_filename + "\""])) | ["-blockversion=2", "-alertnotify=echo %s >> \"" + self.alert_filename + "\""])) | ||||
# Node1 mines block.version=211 blocks | # Node1 mines block.version=211 blocks | ||||
self.nodes.append(start_node(1, self.options.tmpdir, | self.nodes.append(start_node(1, self.options.tmpdir, | ||||
["-blockversion=211"])) | ["-blockversion=211"])) | ||||
connect_nodes(self.nodes[1], 0) | connect_nodes(self.nodes[1], 0) | ||||
self.is_network_split = False | self.is_network_split = False | ||||
self.sync_all() | self.sync_all() | ||||
def run_test(self): | def run_test(self): | ||||
# Mine 51 up-version blocks | # Mine 51 up-version blocks | ||||
self.nodes[1].generate(51) | self.nodes[1].generate(51) | ||||
self.sync_all() | self.sync_all() | ||||
# -alertnotify should trigger on the 51'st, | # -alertnotify should trigger on the 51'st, | ||||
# but mine and sync another to give | # but mine and sync another to give | ||||
# -alertnotify time to write | # -alertnotify time to write | ||||
self.nodes[1].generate(1) | self.nodes[1].generate(1) | ||||
self.sync_all() | self.sync_all() | ||||
with open(self.alert_filename, 'r', encoding='utf8') as f: | with open(self.alert_filename, 'r', encoding='utf8') as f: | ||||
alert_text = f.read() | alert_text = f.read() | ||||
if len(alert_text) == 0: | if len(alert_text) == 0: | ||||
raise AssertionError("-alertnotify did not warn of up-version blocks") | raise AssertionError( | ||||
"-alertnotify did not warn of up-version blocks") | |||||
# Mine more up-version blocks, should not get more alerts: | # Mine more up-version blocks, should not get more alerts: | ||||
self.nodes[1].generate(1) | self.nodes[1].generate(1) | ||||
self.sync_all() | self.sync_all() | ||||
self.nodes[1].generate(1) | self.nodes[1].generate(1) | ||||
self.sync_all() | self.sync_all() | ||||
with open(self.alert_filename, 'r', encoding='utf8') as f: | with open(self.alert_filename, 'r', encoding='utf8') as f: | ||||
alert_text2 = f.read() | alert_text2 = f.read() | ||||
if alert_text != alert_text2: | if alert_text != alert_text2: | ||||
raise AssertionError("-alertnotify excessive warning of up-version blocks") | raise AssertionError( | ||||
"-alertnotify excessive warning of up-version blocks") | |||||
if __name__ == '__main__': | if __name__ == '__main__': | ||||
ForkNotifyTest().main() | ForkNotifyTest().main() |