def trial_runner(fs, environ):
"""Fixture that creates a TrialRunner object."""
os.environ.update({
'BENCHMARK': BENCHMARK,
'EXPERIMENT': EXPERIMENT,
'TRIAL_ID': str(TRIAL_NUM),
'FUZZER': FUZZER,
'EXPERIMENT_FILESTORE': EXPERIMENT_FILESTORE,
'MAX_TOTAL_TIME': str(MAX_TOTAL_TIME)
})
with mock.patch('common.filestore_utils.rm'):
trial_runner = runner.TrialRunner()
trial_runner.initialize_directories()
yield trial_runner
def trial_runner(fs, environ):
"""Fixture that creates a TrialRunner object."""
os.environ.update({
'BENCHMARK': BENCHMARK,
'FUZZER_VARIANT_NAME': FULL_FUZZER_NAME,
'EXPERIMENT': EXPERIMENT,
'TRIAL_ID': str(TRIAL_NUM),
'FUZZER': FUZZER,
'CLOUD_EXPERIMENT_BUCKET': CLOUD_EXPERIMENT_BUCKET,
'MAX_TOTAL_TIME': str(MAX_TOTAL_TIME)
})
with mock.patch('common.gsutil.rm'):
trial_runner = runner.TrialRunner()
trial_runner.initialize_directories()
yield trial_runner
