def testAddDescriptiveResults(self): reference_file_path = os.path.join( self.base_dir2, 'network_listener_analyzer_descriptive_results') analyzer = NetworkListenerAnalyzer() analyzer._analysis_results = self.expected_result analysis_result = wheelbarrow_pb2.AnalysisResult() analysis_result.analysis_name = 'test' analyzer.AddDescriptiveResults(wheelbarrow_pb2.RUN_BINARIES, analysis_result) test_utils.CheckResultProtobufFromFile( self, analysis_result, reference_file_path, wheelbarrow_pb2.AnalysisResult())
def testAddDescriptiveResults(self): reference_file_name = os.path.join( self.base_dir, 'permission_analyzer_descriptive_results') permission_analyzer = PermissionAnalyzer() permission_analyzer._file_types = self.file_types permission_analyzer._AddAnalysisResult(self.trigger1, self.result1) analysis_result = wheelbarrow_pb2.AnalysisResult() analysis_result.analysis_name = self.analysis_name permission_analyzer.AddDescriptiveResults(self.trigger1, analysis_result) test_utils.CheckResultProtobufFromFile( self, analysis_result, reference_file_name, wheelbarrow_pb2.AnalysisResult())
def testAddDiffResults(self): reference_file_name = os.path.join(self.base_dir, 'permission_analyzer_diff_results') permission_analyzer = PermissionAnalyzer() permission_analyzer._file_types = self.file_types permission_analyzer._AddAnalysisResult(self.trigger1, self.result1) permission_analyzer._AddAnalysisResult(self.trigger2, self.result2) analysis_result = wheelbarrow_pb2.AnalysisResult() analysis_result.analysis_name = self.analysis_name diff_pair = wheelbarrow_pb2.AnalysisDescriptor.DiffPair() diff_pair.before = self.trigger1 diff_pair.after = self.trigger2 permission_analyzer.AddDiffResults(diff_pair, analysis_result) test_utils.CheckResultProtobufFromFile( self, analysis_result, reference_file_name, wheelbarrow_pb2.AnalysisResult())
def testPrepareDescriptiveFileResult(self): file_analyzer = FileAnalyzer() file_analyzer._RecordFileType(self.script, self.test_path) pb = file_analyzer._PrepareDescriptiveFileResult( self.test_path, wheelbarrow_pb2.AnalysisResult(), wheelbarrow_pb2.REMOVE) reference_file_path = os.path.join(self.base_dir, 'descriptive_file_result') test_utils.CheckResultProtobufFromFile(self, pb, reference_file_path, wheelbarrow_pb2.FileResult())
def testAddDiffResults(self): reference_file_name = os.path.join( WHEELBARROW_HOME, TEST_PATH, 'inotify_file_analyzer_diff_results') file_analyzer = inotify_file_analyzer.InotifyFileAnalyzer() file_analyzer._file_types = { 'file1': wheelbarrow_pb2.FileResult.TEXT, 'file2': wheelbarrow_pb2.FileResult.SCRIPT } result1 = Counter({'file1': 2}) result2 = Counter({'file1': 5, 'file2': 21}) file_analyzer._analysis_results[self.trigger1] = result1 file_analyzer._analysis_results[self.trigger3] = result2 analysis_result = wheelbarrow_pb2.AnalysisResult() analysis_result.analysis_name = 'test' diff_pair = wheelbarrow_pb2.AnalysisDescriptor.DiffPair() diff_pair.before = self.trigger1 diff_pair.after = self.trigger3 file_analyzer.AddDiffResults(diff_pair, analysis_result) test_utils.CheckResultProtobufFromFile( self, analysis_result, reference_file_name, wheelbarrow_pb2.AnalysisResult())
def testPrepareDiffFileResult(self): diff_pair = wheelbarrow_pb2.AnalysisDescriptor.DiffPair() diff_pair.before = wheelbarrow_pb2.EXTRACT diff_pair.after = wheelbarrow_pb2.INSTALL file_analyzer = FileAnalyzer() file_analyzer._RecordFileType(self.script, self.test_path) pb = file_analyzer._PrepareDiffFileResult( self.test_path, wheelbarrow_pb2.AnalysisResult(), wheelbarrow_pb2.ADD, diff_pair) reference_file_path = os.path.join(self.base_dir, 'diff_file_result') test_utils.CheckResultProtobufFromFile(self, pb, reference_file_path, wheelbarrow_pb2.FileResult())
def testPrepareFileResultWithBadArguments(self): self.assertRaises(analysis.FatalAnalysisError, FileAnalyzer()._PrepareFileResult, self.test_path, wheelbarrow_pb2.AnalysisResult(), wheelbarrow_pb2.ADD, None, None)