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