Differential D2251 Diff 6539 test/functional/test_framework/util.py

Changeset View

Standalone View

test/functional/test_framework/util.py

Show First 20 Lines • Show All 304 Lines • ▼ Show 20 Lines	def p2p_port(n):
assert(n <= MAX_NODES)		assert(n <= MAX_NODES)
return PORT_MIN + n + (MAX_NODES * PortSeed.n) % (PORT_RANGE - 1 - MAX_NODES)		return PORT_MIN + n + (MAX_NODES * PortSeed.n) % (PORT_RANGE - 1 - MAX_NODES)


def rpc_port(n):		def rpc_port(n):
return PORT_MIN + PORT_RANGE + n + (MAX_NODES * PortSeed.n) % (PORT_RANGE - 1 - MAX_NODES)		return PORT_MIN + PORT_RANGE + n + (MAX_NODES * PortSeed.n) % (PORT_RANGE - 1 - MAX_NODES)


def rpc_url(datadir, i, rpchost=None):		def rpc_url(datadir, host, port):
rpc_u, rpc_p = get_auth_cookie(datadir)		rpc_u, rpc_p = get_auth_cookie(datadir)
		if host == None:
host = '127.0.0.1'		host = '127.0.0.1'
port = rpc_port(i)
if rpchost:
parts = rpchost.split(':')
if len(parts) == 2:
host, port = parts
else:
host = rpchost
return "http://%s:%s@%s:%d" % (rpc_u, rpc_p, host, int(port))		return "http://%s:%s@%s:%d" % (rpc_u, rpc_p, host, int(port))

# Node functions		# Node functions
################		################


def initialize_datadir(dirname, n):		def initialize_datadir(dirname, n):
datadir = os.path.join(dirname, "node" + str(n))		datadir = os.path.join(dirname, "node" + str(n))
▲ Show 20 Lines • Show All 57 Lines • ▼ Show 20 Lines	for _ in range(50):
if [peer['id'] for peer in from_node.getpeerinfo() if to_node.name in peer['subver']] == []:		if [peer['id'] for peer in from_node.getpeerinfo() if to_node.name in peer['subver']] == []:
break		break
time.sleep(0.1)		time.sleep(0.1)
else:		else:
raise AssertionError("timed out waiting for disconnect")		raise AssertionError("timed out waiting for disconnect")


def connect_nodes(from_node, to_node):		def connect_nodes(from_node, to_node):
ip_port = "127.0.0.1:" + str(to_node.p2p_port)		host = to_node.host
		if host == None:
		host = '127.0.0.1'
		ip_port = host + ':' + str(to_node.p2p_port)
from_node.addnode(ip_port, "onetry")		from_node.addnode(ip_port, "onetry")
# poll until version handshake complete to avoid race conditions		# poll until version handshake complete to avoid race conditions
# with transaction relaying		# with transaction relaying
while any(peer['version'] == 0 for peer in from_node.getpeerinfo()):		while any(peer['version'] == 0 for peer in from_node.getpeerinfo()):
time.sleep(0.1)		time.sleep(0.1)


def connect_nodes_bi(a, b):		def connect_nodes_bi(a, b):
▲ Show 20 Lines • Show All 232 Lines • Show Last 20 Lines