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()
