def setUp(self):
os.environ['BUILD_DIR'] = DATA_DIRECTORY
os.environ['FUZZ_INPUTS_DISK'] = TEMP_DIRECTORY
os.environ['FAIL_RETRIES'] = '1'
test_helpers.patch_environ(self)
dont_use_strategies(self)
# Make it easy to assert if things were logged.
test_helpers.patch(self, [
'metrics.logs.log', 'os.getpid',
'bot.fuzzers.afl.launcher.rand_cmplog_level',
'bot.fuzzers.afl.launcher.rand_schedule'
])
self.mock.rand_cmplog_level.return_value = '2'
self.mock.rand_schedule.return_value = 'fast'
self.logged_messages = []
def mocked_log(message, **kwargs): # pylint: disable=unused-argument
self.logged_messages.append(message)
self.mock.log.side_effect = mocked_log
self.mock.getpid.return_value = 1337
clear_temp_dir()
create_temp_dir()
def setUp(self):
def get_data_path(filename, is_in_output=True):
"""Returns absolute path of data files used in unittests."""
input_or_output_dir = 'output' if is_in_output else 'input'
return os.path.join(self.data_dir, input_or_output_dir, filename)
self.data_dir = os.path.join(
os.path.abspath(os.path.dirname(__file__)), 'stats_data')
self.fuzzer_stats_path = get_data_path('fuzzer_stats')
self.fuzzer_stats_invalid_path = get_data_path('fuzzer_stats_invalid')
self.dict_path = get_data_path('sample.dict', False)
self.stats_getter = stats.StatsGetter(self.fuzzer_stats_path,
self.dict_path)
self.stats_getter.set_afl_stats()
dont_use_strategies(self)
self.strategies = launcher.FuzzingStrategies()
self.maxDiff = None # pylint: disable=invalid-name
