Page MenuHomePhabricator
Differential D2789 Diff 8010 test/functional/feature_notifications.py

Changeset View

Standalone View

View Options

test/functional/feature_notifications.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
# Copyright (c) 2018 The Bitcoin developers # Copyright (c) 2018 The Bitcoin 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 the -alertnotify, -blocknotify and -walletnotify options.""" """Test the -blocknotify and -walletnotify options."""
import os import os
import mmap
from test_framework.test_framework import BitcoinTestFramework from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import assert_equal, connect_nodes_bi, wait_until from test_framework.util import assert_equal, connect_nodes_bi, wait_until
FORK_WARNING_MESSAGE = "Warning: Large-work fork detected, forking after block {}\n"
class NotificationsTest(BitcoinTestFramework): class NotificationsTest(BitcoinTestFramework):
def set_test_params(self): def set_test_params(self):
self.num_nodes = 2 self.num_nodes = 2
self.setup_clean_chain = True self.setup_clean_chain = True
def setup_network(self): def setup_network(self):
self.alert_filename = os.path.join(self.options.tmpdir, "alert.txt")
self.block_filename = os.path.join(self.options.tmpdir, "blocks.txt") self.block_filename = os.path.join(self.options.tmpdir, "blocks.txt")
self.tx_filename = os.path.join( self.tx_filename = os.path.join(
self.options.tmpdir, "transactions.txt") self.options.tmpdir, "transactions.txt")
# -alertnotify and -blocknotify on node0, walletnotify on node1 # -blocknotify on node0, walletnotify on node1
self.extra_args = [["-blockversion=2", self.extra_args = [["-blockversion=2",
"-alertnotify=echo %s >> {}".format(
self.alert_filename),
"-blocknotify=echo %s >> {}".format(self.block_filename)], "-blocknotify=echo %s >> {}".format(self.block_filename)],
["-blockversion=211", ["-blockversion=211",
"-rescan", "-rescan",
"-walletnotify=echo %s >> {}".format(self.tx_filename)]] "-walletnotify=echo %s >> {}".format(self.tx_filename)]]
super().setup_network() super().setup_network()
def run_test(self): def run_test(self):
self.log.info("test -blocknotify") self.log.info("test -blocknotify")
Show All 29 Lines def run_test(self):
self.tx_filename).st_size >= (block_count * 65), timeout=10) self.tx_filename).st_size >= (block_count * 65), timeout=10)
# file content should equal the generated transaction hashes # file content should equal the generated transaction hashes
txids_rpc = list( txids_rpc = list(
map(lambda t: t['txid'], self.nodes[1].listtransactions("*", block_count))) map(lambda t: t['txid'], self.nodes[1].listtransactions("*", block_count)))
with open(self.tx_filename, 'r') as f: with open(self.tx_filename, 'r') as f:
assert_equal(sorted(txids_rpc), sorted(f.read().splitlines())) assert_equal(sorted(txids_rpc), sorted(f.read().splitlines()))
# Mine another 41 up-version blocks. -alertnotify should trigger on the 51st. # Mine another 41 up-version blocks. Warning should appear on the 51st in the logs
self.log.info("test -alertnotify for bip9") self.log.info(
"test that previously defined alertnotify conditions are present in debug.log")
self.nodes[1].generate(41) self.nodes[1].generate(41)
self.sync_all() self.sync_all()
# Give bitcoind 10 seconds to write the alert notification log_file = os.path.join(self.nodes[0].datadir, "regtest", "debug.log")
wait_until(lambda: os.path.isfile(self.alert_filename) with open(log_file, 'rb', 0) as file, \
and os.path.getsize(self.alert_filename), timeout=10) mmap.mmap(file.fileno(), 0, access=mmap.ACCESS_READ) as s:
assert s.find(b'of last 100 blocks have unexpected version') != - \
with open(self.alert_filename, 'r', encoding='utf8') as f: 1, "Unknown version message warning not found in log file."
alert_text = f.read()
# Mine more up-version blocks, should not get more alerts:
self.nodes[1].generate(2)
self.sync_all()
with open(self.alert_filename, 'r', encoding='utf8') as f:
alert_text2 = f.read()
os.remove(self.alert_filename)
self.log.info(
"-alertnotify should not continue notifying for more unknown version blocks")
assert_equal(alert_text, alert_text2)
# Create an invalid chain and ensure the node warns. # Create an invalid chain and ensure the node warns.
self.log.info("test -alertnotify for forked chain") self.log.info("test log for forked chain")
fork_block = self.nodes[0].getbestblockhash() fork_block = self.nodes[0].getbestblockhash()
self.nodes[0].generate(1) self.nodes[0].generate(1)
invalid_block = self.nodes[0].getbestblockhash() invalid_block = self.nodes[0].getbestblockhash()
self.nodes[0].generate(7) self.nodes[0].generate(7)
# Invalidate a large branch, which should trigger an alert. # Invalidate a large branch, which should trigger an alert.
self.nodes[0].invalidateblock(invalid_block) self.nodes[0].invalidateblock(invalid_block)
# Give bitcoind 10 seconds to write the alert notification with open(log_file, 'rb', 0) as file, \
wait_until(lambda: os.path.isfile(self.alert_filename) mmap.mmap(file.fileno(), 0, access=mmap.ACCESS_READ) as s:
and os.path.getsize(self.alert_filename), timeout=10) assert s.find(b'Warning: Large fork found') != -1 and s.find(fork_block.encode('utf-8')), \
"Large fork warning not found in log file."
self.log.info(self.alert_filename)
with open(self.alert_filename, 'r', encoding='utf8') as f:
assert_equal(f.read(), (FORK_WARNING_MESSAGE.format(fork_block)))
if __name__ == '__main__': if __name__ == '__main__':
NotificationsTest().main() NotificationsTest().main()