def test_with_rebased_files_returns_same_args_when_there_are_no_files( self): args = FileinfoArguments() 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_clone_returns_other_args_equal_to_original_args(self): args = FileinfoArguments(input_files=(StandaloneFile('file.exe'), ), config_file=StandaloneFile('file.json'), args='--arg') cloned_args = args.clone() self.assertIsNot(args, cloned_args) self.assertEqual(args, cloned_args)
def _run_fileinfo_if_requested(self, dir, unpacker_args): """Runs fileinfo over the output from the unpacker, but only when such a run was requested by the user. """ if not self._test_settings.run_fileinfo: return None fileinfo_args = FileinfoArguments( input_files=(unpacker_args.output_file,), args=self._test_settings.fileinfo_args ) output, return_code, timeouted = self._run_tool( 'fileinfo', fileinfo_args, self._test_settings.fileinfo_timeout ) return Fileinfo( 'fileinfo', dir, fileinfo_args, self._cmd_runner, output, return_code, timeouted )
def test_create_log_returns_combined_log_when_fileinfo_run(self): self.test_settings.run_fileinfo = True self.test_settings.fileinfo_timeout = 60 unpacker = mock.Mock() unpacker.name = 'unpacker' unpacker.output = 'unpacker output\n' unpacker.timeouted = False unpacker.return_code = 0 unpacker.args = UnpackerArguments(args='--unpacker-arg') unpacker.fileinfo = mock.Mock() unpacker.fileinfo.name = 'fileinfo' unpacker.fileinfo.output = 'fileinfo output\n' unpacker.fileinfo.timeouted = True unpacker.fileinfo.return_code = 1 unpacker.fileinfo.args = FileinfoArguments(args='--fileinfo-arg') log = self.unpacker_runner._create_log(unpacker, timeout=100) expected_log = '\n'.join([ '# Command: unpacker --unpacker-arg', '# Timeout: 100 seconds', '', 'unpacker output', '', '# Return code: 0', '# Timeouted: no', '', '# ------------------------------------------------------------------------------', '', '# Command: fileinfo --fileinfo-arg', '# Timeout: 60 seconds', '', 'fileinfo output', '', '# Return code: 1', '# Timeouted: yes', '' ]) self.assertEqual(log, expected_log)
def test_without_paths_and_output_files_returns_correct_args_when_there_are_files( self): args = FileinfoArguments( input_files=(File('file.exe', Directory(os.path.join(ROOT_DIR, 'inputs'))), ), config_file=File('file.json', 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.config_file)
def test_with_rebased_files_returns_correct_args_when_there_are_files( self): args = FileinfoArguments(input_files=(StandaloneFile('file.exe'), ), config_file=StandaloneFile('file.json')) 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.config_file.path, os.path.join(ROOT_DIR, 'outputs', 'file.json'))
def test_without_paths_and_output_files_returns_same_args_when_there_are_no_files( self): args = FileinfoArguments() self.assertEqual(args, args.without_paths_and_output_files)
def test_as_list_returns_correct_list_when_just_args_is_set(self): args = FileinfoArguments(args=' --arg1 --arg2 ') self.assertEqual(args.as_list, ['--arg1', '--arg2'])
def test_as_list_returns_correct_list_when_config_file_is_set(self): args = FileinfoArguments(input_files=(StandaloneFile('file.exe'), ), config_file=StandaloneFile('file.json')) self.assertEqual(args.as_list, ['file.exe', '-c', 'file.json'])
def test_as_list_returns_correct_list_when_just_input_files_are_set(self): args = FileinfoArguments(input_files=(StandaloneFile('file.exe'), )) self.assertEqual(args.as_list, ['file.exe'])
def test_input_file_returns_file_with_correct_name(self): args = FileinfoArguments(input_files=(StandaloneFile('file.exe'), )) self.assertEqual(args.input_file.name, 'file.exe')
def test_repr_returns_executable_repr_that_creates_original_args(self): args = FileinfoArguments(input_files=(StandaloneFile('file.exe'), ), config_file=StandaloneFile('file.json'), args='--arg') self.assertEqual(args, eval(repr(args)))
def test_two_args_having_different_args_are_not_equal(self): args1 = FileinfoArguments(input_files=(StandaloneFile('file.exe'), ), args='--arg') args2 = FileinfoArguments(input_files=(StandaloneFile('file.exe'), ), args='--other-arg') self.assertNotEqual(args1, args2)
def test_two_args_having_different_config_file_are_not_equal(self): args1 = FileinfoArguments(config_file=StandaloneFile('file1.json')) args2 = FileinfoArguments(config_file=StandaloneFile('file2.json')) self.assertNotEqual(args1, args2)
def test_two_args_having_different_input_files_are_not_equal(self): args1 = FileinfoArguments(input_files=(StandaloneFile('file1.exe'), )) args2 = FileinfoArguments(input_files=(StandaloneFile('file2.exe'), )) self.assertNotEqual(args1, args2)
def test_two_args_having_same_data_are_equal(self): args1 = FileinfoArguments(input_files=(StandaloneFile('file.exe'), ), args='--arg') args2 = FileinfoArguments(input_files=(StandaloneFile('file.exe'), ), args='--arg') self.assertEqual(args1, args2)
def test_as_list_returns_empty_list_when_nothing_is_set(self): args = FileinfoArguments() self.assertEqual(args.as_list, [])
def setUp(self): super().setUp() self.name = 'fileinfo' self.args = FileinfoArguments( input_files=(File('prog.exe', Directory('/test')), )) self.fileinfo = self.create_fileinfo()