def import_csv_data(request, data, graph_spaces, session, pytestconfig): data_dir = os.path.join(DATA_DIR, normalize_outline_scenario(request, data)) space_desc = load_csv_data( session, data_dir, "I" + space_generator(), ) assert space_desc is not None graph_spaces["space_desc"] = space_desc graph_spaces["drop_space"] = True
def given_nebulacluster_with_param( request, params, graphd_num, metad_num, storaged_num, class_fixture_variables, pytestconfig, ): graphd_param, metad_param, storaged_param = {}, {}, {} if params is not None: for param in params.splitlines(): module, config = param.strip().split(":") assert module.lower() in ["graphd", "storaged", "metad"] key, value = config.strip().split("=") if module.lower() == "graphd": graphd_param[key] = value elif module.lower() == "storaged": storaged_param[key] = value else: metad_param[key] = value user = pytestconfig.getoption("user") password = pytestconfig.getoption("password") build_dir = pytestconfig.getoption("build_dir") src_dir = pytestconfig.getoption("src_dir") nebula_svc = NebulaService( build_dir, src_dir, int(metad_num), int(storaged_num), int(graphd_num), ) for process in nebula_svc.graphd_processes: process.update_param(graphd_param) for process in nebula_svc.storaged_processes: process.update_param(storaged_param) for process in nebula_svc.metad_processes: process.update_param(metad_param) work_dir = os.path.join( build_dir, "C" + space_generator() + time.strftime('%Y-%m-%dT%H-%M-%S', time.localtime()), ) nebula_svc.install(work_dir) nebula_svc.start() graph_ip = nebula_svc.graphd_processes[0].host graph_port = nebula_svc.graphd_processes[0].tcp_port # TODO add ssl pool if tests needed pool = get_conn_pool(graph_ip, graph_port, None) sess = pool.get_session(user, password) class_fixture_variables["current_session"] = sess class_fixture_variables["sessions"].append(sess) class_fixture_variables["cluster"] = nebula_svc class_fixture_variables["pool"] = pool
def clone_space(graph_spaces, session, request): space_desc = graph_spaces["space_desc"] current_space = space_desc._name new_space = "EmptyGraph_" + space_generator() space_desc._name = new_space resp_ok(session, space_desc.drop_stmt(), True) ngql = "create space " + new_space + " as " + current_space exec_query(request, ngql, session, graph_spaces) resp_ok(session, space_desc.use_stmt(), True) graph_spaces["space_desc"] = space_desc graph_spaces["drop_space"] = True
def import_csv_data(data, graph_spaces, session, pytestconfig): data_dir = os.path.join(DATA_DIR, data) space_desc = load_csv_data( pytestconfig, session, data_dir, "I" + space_generator(), ) assert space_desc is not None graph_spaces["space_desc"] = space_desc graph_spaces["drop_space"] = True
def new_space(request, session, graph_spaces): name = "EmptyGraph_" + space_generator() space_desc = SpaceDesc( name=name, partition_num=9, replica_factor=1, vid_type="FIXED_STRING(30)", charset="utf8", collate="utf8_bin", ) create_space(space_desc, session) graph_spaces["space_desc"] = space_desc graph_spaces["drop_space"] = True
def new_space(options, session, graph_spaces): lines = csv.reader(io.StringIO(options), delimiter="|") opts = {line[1].strip(): line[2].strip() for line in lines} name = "EmptyGraph_" + space_generator() space_desc = SpaceDesc( name=name, partition_num=int(opts.get("partition_num", 7)), replica_factor=int(opts.get("replica_factor", 1)), vid_type=opts.get("vid_type", "FIXED_STRING(30)"), charset=opts.get("charset", "utf8"), collate=opts.get("collate", "utf8_bin"), ) create_space(space_desc, session) graph_spaces["space_desc"] = space_desc graph_spaces["drop_space"] = True