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)