Example #1
0
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
Example #2
0
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
Example #3
0
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
Example #4
0
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
Example #5
0
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
Example #6
0
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())