Page MenuHomePhabricator
Differential D2164 Diff 6210 test/functional/test_framework/util.py

Changeset View

Standalone View

View Options

test/functional/test_framework/util.py

Show First 20 Lines • Show All 263 Lines • ▼ Show 20 Lines
# The maximum number of nodes a single test can spawn # The maximum number of nodes a single test can spawn
MAX_NODES = 8 MAX_NODES = 8
# Don't assign rpc or p2p ports lower than this # Don't assign rpc or p2p ports lower than this
PORT_MIN = 11000 PORT_MIN = 11000
# The number of ports to "reserve" for p2p and rpc, each # The number of ports to "reserve" for p2p and rpc, each
PORT_RANGE = 5000 PORT_RANGE = 5000
class PortSeed:
# Must be initialized with a unique integer for each process
n = None
def get_rpc_proxy(url, node_number, timeout=None, coveragedir=None): def get_rpc_proxy(url, node_number, timeout=None, coveragedir=None):
""" """
Args: Args:
url (str): URL of the RPC server to call url (str): URL of the RPC server to call
node_number (int): the node number (or id) that this calls to node_number (int): the node number (or id) that this calls to
Kwargs: Kwargs:
timeout (int): HTTP timeout in seconds timeout (int): HTTP timeout in seconds
Show All 10 Linesdef get_rpc_proxy(url, node_number, timeout=None, coveragedir=None):
proxy.url = url # store URL on proxy for info proxy.url = url # store URL on proxy for info
coverage_logfile = coverage.get_filename( coverage_logfile = coverage.get_filename(
coveragedir, node_number) if coveragedir else None coveragedir, node_number) if coveragedir else None
return coverage.AuthServiceProxyWrapper(proxy, coverage_logfile) return coverage.AuthServiceProxyWrapper(proxy, coverage_logfile)
def p2p_port(n):
assert(n <= MAX_NODES)
return PORT_MIN + n + (MAX_NODES * PortSeed.n) % (PORT_RANGE - 1 - MAX_NODES)
def rpc_port(n):
return PORT_MIN + PORT_RANGE + n + (MAX_NODES * PortSeed.n) % (PORT_RANGE - 1 - MAX_NODES)
def rpc_url(datadir, i, rpchost, rpc_port): def rpc_url(datadir, i, rpchost, rpc_port):
rpc_u, rpc_p = get_auth_cookie(datadir) rpc_u, rpc_p = get_auth_cookie(datadir)
host = '127.0.0.1' host = '127.0.0.1'
if rpchost: if rpchost:
host = rpchost host = rpchost
return "http://%s:%s@%s:%d" % (rpc_u, rpc_p, host, int(rpc_port)) return "http://%s:%s@%s:%d" % (rpc_u, rpc_p, host, int(rpc_port))
# Node functions # Node functions
################ ################
def initialize_datadir(dirname, n, p2p_port, rpc_port): def initialize_datadir(dirname, n, p2p_port, rpc_port):
datadir = os.path.join(dirname, "node" + str(n)) datadir = os.path.join(dirname, "node" + str(n))
if not os.path.isdir(datadir): if not os.path.isdir(datadir):
os.makedirs(datadir) os.makedirs(datadir)
▲ Show 20 Lines • Show All 304 Lines • Show Last 20 Lines