LBRYCommunity/lbrycrd
1
0
Fork 0
Code Issues 29 Pull requests 1 Projects Releases 32 Packages Wiki Activity

Add start and stop node methods to BitcoinTestFramework

Browse source
This commit is contained in:
John Newbery 2017-03-23 16:56:45 -04:00
parent b111324acd
commit b7dd44c528
1 changed files with 19 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
test/functional/test_framework/test_framework.py
View file

@ -31,6 +31,7 @@ from .util import (
p2p_port, p2p_port,
rpc_url, rpc_url,
set_node_times, set_node_times,
start_node,
start_nodes, start_nodes,
stop_node, stop_node,
stop_nodes, stop_nodes,
@ -64,9 +65,18 @@ class BitcoinTestFramework(object):
else: else:
self._initialize_chain(self.options.tmpdir, self.num_nodes, self.options.cachedir) self._initialize_chain(self.options.tmpdir, self.num_nodes, self.options.cachedir)
def start_node(self, i, dirname, extra_args=None, rpchost=None, timewait=None, binary=None, stderr=None):
return start_node(i, dirname, extra_args, rpchost, timewait, binary, stderr)
def start_nodes(self, num_nodes, dirname, extra_args=None, rpchost=None, timewait=None, binary=None):
return start_nodes(num_nodes, dirname, extra_args, rpchost, timewait, binary)
def stop_node(self, num_node): def stop_node(self, num_node):
stop_node(self.nodes[num_node], num_node) stop_node(self.nodes[num_node], num_node)
def stop_nodes(self):
stop_nodes(self.nodes)
def setup_nodes(self): def setup_nodes(self):
extra_args = None extra_args = None
if hasattr(self, "extra_args"): if hasattr(self, "extra_args"):
@ -165,7 +175,7 @@ class BitcoinTestFramework(object):
if not self.options.noshutdown: if not self.options.noshutdown:
self.log.info("Stopping nodes") self.log.info("Stopping nodes")
stop_nodes(self.nodes) self.stop_nodes()
else: else:
self.log.info("Note: bitcoinds were not stopped and may still be running") self.log.info("Note: bitcoinds were not stopped and may still be running")
@ -259,10 +269,10 @@ class BitcoinTestFramework(object):
wait_for_bitcoind_start(bitcoind_processes[i], rpc_url(i), i) wait_for_bitcoind_start(bitcoind_processes[i], rpc_url(i), i)
self.log.debug("initialize_chain: RPC successfully started") self.log.debug("initialize_chain: RPC successfully started")
rpcs = [] self.nodes = []
for i in range(MAX_NODES): for i in range(MAX_NODES):
try: try:
rpcs.append(get_rpc_proxy(rpc_url(i), i)) self.nodes.append(get_rpc_proxy(rpc_url(i), i))
except: except:
self.log.exception("Error connecting to node %d" % i) self.log.exception("Error connecting to node %d" % i)
sys.exit(1) sys.exit(1)
@ -279,14 +289,15 @@ class BitcoinTestFramework(object):
for i in range(2): for i in range(2):
for peer in range(4): for peer in range(4):
for j in range(25): for j in range(25):
set_node_times(rpcs, block_time) set_node_times(self.nodes, block_time)
rpcs[peer].generate(1) self.nodes[peer].generate(1)
block_time += 10 * 60 block_time += 10 * 60
# Must sync before next peer starts generating blocks # Must sync before next peer starts generating blocks
sync_blocks(rpcs) sync_blocks(self.nodes)
# Shut them down, and clean up cache directories: # Shut them down, and clean up cache directories:
stop_nodes(rpcs) self.stop_nodes()
self.nodes = []
disable_mocktime() disable_mocktime()
for i in range(MAX_NODES): for i in range(MAX_NODES):
os.remove(log_filename(cachedir, i, "debug.log")) os.remove(log_filename(cachedir, i, "debug.log"))
@ -330,7 +341,7 @@ class ComparisonTestFramework(BitcoinTestFramework):
help="bitcoind binary to use for reference nodes (if any)") help="bitcoind binary to use for reference nodes (if any)")
def setup_network(self): def setup_network(self):
self.nodes = start_nodes( self.nodes = self.start_nodes(
self.num_nodes, self.options.tmpdir, self.num_nodes, self.options.tmpdir,
extra_args=[['-whitelist=127.0.0.1']] * self.num_nodes, extra_args=[['-whitelist=127.0.0.1']] * self.num_nodes,
binary=[self.options.testbinary] + binary=[self.options.testbinary] +