def testGetEventTags(self): """Tests the GetEventTags function.""" test_events = self._CreateTestEvents() with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, u'storage.plaso') storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file, read_only=False) for event in test_events: storage_file.AddEvent(event) test_event_tags = self._CreateTestEventTags(test_events) for event_tag in test_event_tags: storage_file.AddEventTag(event_tag) storage_file.Close() storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file) test_event_tags = list(storage_file.GetEventTags()) self.assertEqual(len(test_event_tags), 4) storage_file.Close()
def testGetEventSources(self): """Tests the GetEventSources function.""" event_source = event_sources.EventSource() with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, u'storage.plaso') storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file, read_only=False) storage_file.AddEventSource(event_source) storage_file.Close() storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file) test_event_sources = list(storage_file.GetEventSources()) self.assertEqual(len(test_event_sources), 1) storage_file.Close()
def testGetAnalysisReports(self): """Tests the GetAnalysisReports function.""" analysis_report = reports.AnalysisReport( plugin_name=u'test', text=u'test report') with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, u'storage.plaso') storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file, read_only=False) storage_file.AddAnalysisReport(analysis_report) storage_file.Close() storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file) test_reports = list(storage_file.GetAnalysisReports()) self.assertEqual(len(test_reports), 1) storage_file.Close()
def testAddEventSource(self): """Tests the AddEventSource function.""" event_source = event_sources.EventSource() with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, u'storage.plaso') storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file, read_only=False) storage_file.AddEventSource(event_source) storage_file.Close()
def testGetErrors(self): """Tests the GetErrors function.""" extraction_error = errors.ExtractionError( message=u'Test extraction error') with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, u'storage.plaso') storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file, read_only=False) storage_file.AddError(extraction_error) storage_file.Close() storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file) test_errors = list(storage_file.GetErrors()) self.assertEqual(len(test_errors), 1) storage_file.Close()
def testAddEvent(self): """Tests the AddEvent function.""" test_events = self._CreateTestEvents() with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, u'storage.plaso') storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file, read_only=False) for event in test_events: storage_file.AddEvent(event) storage_file.Close()
def testAddError(self): """Tests the AddError function.""" extraction_error = errors.ExtractionError( message=u'Test extraction error') with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, u'storage.plaso') storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file, read_only=False) storage_file.AddError(extraction_error) storage_file.Close()
def testAddAnalysisReport(self): """Tests the AddAnalysisReport function.""" analysis_report = reports.AnalysisReport(plugin_name='test', text='test report') with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, 'storage.plaso') storage_file = gzip_file.GZIPStorageFile() storage_file.Open(path=temp_file, read_only=False) storage_file.AddAnalysisReport(analysis_report) storage_file.Close()
def testWriteTaskStartAndCompletion(self): """Tests the WriteTaskStart and WriteTaskCompletion functions.""" session = sessions.Session() task_start = tasks.TaskStart(session_identifier=session.identifier) task_completion = tasks.TaskCompletion( identifier=task_start.identifier, session_identifier=session.identifier) with shared_test_lib.TempDirectory() as temp_directory: temp_file = os.path.join(temp_directory, u'storage.plaso') storage_file = gzip_file.GZIPStorageFile( storage_type=definitions.STORAGE_TYPE_TASK) storage_file.Open(path=temp_file, read_only=False) storage_file.WriteTaskStart(task_start) storage_file.WriteTaskCompletion(task_completion) storage_file.Close()