def test_with_rebased_files_returns_same_args_when_there_are_no_files(
self):
args = IDAPluginArguments()
rebased_args = args.with_rebased_files(
Directory(os.path.join(ROOT_DIR, 'inputs')),
Directory(os.path.join(ROOT_DIR, 'outputs')))
self.assertEqual(args, rebased_args)
def test_two_args_having_same_data_are_equal(self):
args1 = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ),
idb_file=StandaloneFile('file.idb'),
args='--arg')
args2 = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ),
idb_file=StandaloneFile('file.idb'),
args='--arg')
self.assertEqual(args1, args2)
def test_clone_returns_other_args_equal_to_original_args(self):
args = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ),
idb_file=StandaloneFile('file.idb'),
output_file=StandaloneFile('file.c'),
args='--arg')
cloned_args = args.clone()
self.assertIsNot(args, cloned_args)
self.assertEqual(args, cloned_args)
def test_with_rebased_files_returns_correct_args_when_there_are_files(
self):
args = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ),
idb_file=StandaloneFile('file.idb'),
output_file=StandaloneFile('file.c'))
rebased_args = args.with_rebased_files(
Directory(os.path.join(ROOT_DIR, 'inputs')),
Directory(os.path.join(ROOT_DIR, 'outputs')))
self.assertEqual(len(rebased_args.input_files), 1)
self.assertEqual(rebased_args.input_files[0].path,
os.path.join(ROOT_DIR, 'inputs', 'file.exe'))
self.assertEqual(rebased_args.idb_file.path,
os.path.join(ROOT_DIR, 'inputs', 'file.idb'))
self.assertEqual(rebased_args.output_file.path,
os.path.join(ROOT_DIR, 'outputs', 'file.c'))
def setUp(self):
super().setUp()
self.name = 'idaplugin'
self.args = IDAPluginArguments(
input_files=(File('prog.exe', Directory('/test')), ),
output_file=File('prog.c', Directory('/test/outputs')))
self.idaplugin = self.create_idaplugin()
def test_without_paths_and_output_files_returns_correct_args_when_there_are_files(
self):
args = IDAPluginArguments(
input_files=(File('file.exe',
Directory(os.path.join(ROOT_DIR, 'inputs'))), ),
idb_file=File('file.idb',
Directory(os.path.join(ROOT_DIR, 'inputs'))),
output_file=File('file.c',
Directory(os.path.join(ROOT_DIR, 'outputs'))))
stripped_args = args.without_paths_and_output_files
self.assertEqual(len(stripped_args.input_files), 1)
self.assertEqual(stripped_args.input_files[0].path, 'file.exe')
self.assertIsNone(stripped_args.output_file)
def test_as_list_returns_correct_list_when_just_args_is_set(self):
args = IDAPluginArguments(args=' --arg1 --arg2 ')
self.assertEqual(args.as_list, ['--arg1', '--arg2'])
def test_as_list_returns_correct_list_when_output_file_is_set(self):
args = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ),
output_file=StandaloneFile('file.c'))
self.assertEqual(args.as_list, ['file.exe', '-o', 'file.c'])
def test_as_list_returns_correct_list_when_just_input_files_are_set(self):
args = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ))
self.assertEqual(args.as_list, ['file.exe'])
def test_as_list_returns_empty_list_when_nothing_is_set(self):
args = IDAPluginArguments()
self.assertEqual(args.as_list, [])
def test_without_paths_and_output_files_returns_same_args_when_there_are_no_files(
self):
args = IDAPluginArguments()
self.assertEqual(args, args.without_paths_and_output_files)
def test_from_test_settings_args_is_present_when_set(self):
test_settings = IDAPluginTestSettings(input='test.exe',
args='--arg1 --arg2')
args = IDAPluginArguments.from_test_settings(test_settings)
self.assertEqual(args.args, test_settings.args)
def test_from_test_settings_output_file_is_automatically_set(self):
test_settings = IDAPluginTestSettings(input='test.exe')
args = IDAPluginArguments.from_test_settings(test_settings)
self.assertIsNotNone(args.output_file)
def test_two_args_having_different_output_file_are_not_equal(self):
args1 = IDAPluginArguments(output_file=StandaloneFile('file1.c'))
args2 = IDAPluginArguments(output_file=StandaloneFile('file2.c'))
self.assertNotEqual(args1, args2)
def test_from_test_settings_input_files_are_present_when_set(self):
test_settings = IDAPluginTestSettings(input='test.exe')
args = IDAPluginArguments.from_test_settings(test_settings)
self.assertEqual(len(args.input_files), 1)
self.assertEqual(args.input_files[0].name, test_settings.input)
def test_from_test_settings_idb_file_is_present_when_set(self):
test_settings = IDAPluginTestSettings(input='test.exe', idb='file.idb')
args = IDAPluginArguments.from_test_settings(test_settings)
self.assertEqual(args.idb_file.name, test_settings.idb)
def test_two_args_having_different_args_are_not_equal(self):
args1 = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ),
args='--arg')
args2 = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ),
args='--other-arg')
self.assertNotEqual(args1, args2)
def test_from_test_settings_output_file_has_correct_name_when_it_does_not_end_with_exe(
self):
test_settings = IDAPluginTestSettings(input='test.elf')
args = IDAPluginArguments.from_test_settings(test_settings)
self.assertEqual(args.output_file.name, 'test.elf.c')
def test_input_file_returns_file_with_correct_name(self):
args = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ))
self.assertEqual(args.input_file.name, 'file.exe')
def scenario_invalid_settings_error_is_raised(self, test_settings,
ref_exc_substr):
with self.assertRaises(InvalidTestSettingsError) as cm:
IDAPluginArguments.from_test_settings(test_settings)
self.assertIn(ref_exc_substr, str(cm.exception))
def test_repr_returns_executable_repr_that_creates_original_args(self):
args = IDAPluginArguments(input_files=(StandaloneFile('file.exe'), ),
idb_file=StandaloneFile('file.idb'),
output_file=StandaloneFile('file.c'),
args='--arg')
self.assertEqual(args, eval(repr(args)))