else:
assert False, f'Unsupported --script={args.script}'
if not args.skip_load:
logger.info('Starting transaction spamming scripts.')
mocknet.start_load_test_helpers(validator_nodes,
script,
rpc_nodes,
num_nodes,
max_tps,
get_node_key=True)
initial_epoch_height = mocknet.get_epoch_height(rpc_nodes, -1)
logger.info(f'initial_epoch_height: {initial_epoch_height}')
assert initial_epoch_height >= 0
initial_metrics = mocknet.get_metrics(archival_node)
start_time = time.monotonic()
logger.info(
f'Waiting for contracts to be deployed for {deploy_time} seconds.')
prev_epoch_height = initial_epoch_height
EPOCH_HEIGHT_CHECK_DELAY = 30
while time.monotonic() - start_time < deploy_time:
epoch_height = mocknet.get_epoch_height(rpc_nodes,
prev_epoch_height)
if epoch_height > prev_epoch_height:
mocknet.upgrade_nodes(epoch_height - initial_epoch_height,
upgrade_schedule, all_nodes)
prev_epoch_height = epoch_height
time.sleep(EPOCH_HEIGHT_CHECK_DELAY)
logger.info(
def check_memory_usage(node):
metrics = mocknet.get_metrics(node)
mem_usage = metrics.memory_usage / 1e6
print(f'INFO: Memory usage (MB) = {mem_usage}')
return mem_usage < 4500