def create_config(testvar): if 'data-directory' in testvar: # We need to create a tempfile to hold the config tmpconfig = conftest._create_context_config() tmpconfig['data-directory'] = testvar['data-directory'] tf = tempfile.NamedTemporaryFile(delete=False) with open(tf.name, 'w') as f: f.write(yaml.dump(tmpconfig)) return tf.name
def create_config(testvar): if 'data-directory' in testvar: # We need to create a tempfile to hold the config tmpconfig = conftest._create_context_config() tmpconfig['data-directory'] = testvar['data-directory'] tmpfname = '/tmp/test-sq.cfg' with open(tmpfname, 'w') as f: f.write(yaml.dump(tmpconfig)) return tmpfname
def create_config(testvar): if 'data-directory' in testvar: # We need to create a tempfile to hold the config tmpconfig = conftest._create_context_config() tmpconfig['data-directory'] = testvar['data-directory'] tf = tempfile.NamedTemporaryFile(delete=False) tf.write(b'{yaml.dump(tmpconfig)}') tf.close return tf.name
def create_config(): # We need to create a tempfile to hold the config tmpconfig = conftest._create_context_config() tmpconfig['data-directory'] = './tests/data/multidc/parquet-out' r_int = random.randint(17, 2073) fname = f'/tmp/suzieq-cfg-{r_int}.yml' with open(fname, 'w') as f: f.write(yaml.dump(tmpconfig)) return fname
def create_config(t_dir, suzieq_dir): # We need to create a tempfile to hold the config tmpconfig = conftest._create_context_config() tmpconfig['data-directory'] = f"{t_dir}/parquet-out" tmpconfig['service-directory'] = f"{suzieq_dir}/{tmpconfig['service-directory']}" tmpconfig['schema-directory'] = f"{suzieq_dir}/{tmpconfig['schema-directory']}" fname = f'{t_dir}/suzieq-cfg.yml' with open(fname, 'w') as f: f.write(yaml.dump(tmpconfig)) return fname
def _test_data(topology, proto, scenario, testvar): name = f'{topology}_{proto}_{scenario}' testvar['data-directory'] = f"{parquet_dir}/{name}/parquet-out" _test_sqcmds(testvar, conftest._create_context_config())