def get_fuzzer_directory(fuzzer_name):
"""Return directory used by a fuzzer."""
builtin_fuzzer = builtin_fuzzers.get(fuzzer_name)
if builtin_fuzzer:
return builtin_fuzzer.fuzzer_directory
fuzzer_directory = environment.get_value('FUZZERS_DIR')
fuzzer_directory = os.path.join(fuzzer_directory, fuzzer_name)
return fuzzer_directory
def test_all(self):
"""Tests all()."""
self.assertItemsEqual([
('afl', builtin_fuzzers.get('afl')),
('libFuzzer', builtin_fuzzers.get('libFuzzer')),
], builtin_fuzzers.all())
def test_get(self):
"""Tests get()."""
self.assertIsInstance(builtin_fuzzers.get('libFuzzer'),
builtin.BuiltinFuzzer)
self.assertIsNone(builtin_fuzzers.get('does_not_exist'))
