def testGetFormattedField(self):
"""Tests the GetFormattedField function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event = TestEvent()
zone_string = dynamic_fields_helper.GetFormattedField(event, 'zone')
self.assertEqual(zone_string, 'UTC')
def testFormatSourceShort(self):
"""Tests the _FormatSourceShort function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event = TestEvent()
source_short_string = dynamic_fields_helper._FormatSourceShort(event)
self.assertEqual(source_short_string, 'LOG')
def testFormatInode(self):
"""Tests the _FormatInode function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event = TestEvent()
inode_string = dynamic_fields_helper._FormatInode(event)
self.assertEqual(inode_string, '-')
def testFormatMACB(self):
"""Tests the _FormatMACB function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event = TestEvent()
macb_string = dynamic_fields_helper._FormatMACB(event)
self.assertEqual(macb_string, '..C.')
def testFormatInode(self):
"""Tests the _FormatInode function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event, event_data = containers_test_lib.CreateEventFromValues(
self._TEST_EVENTS[0])
inode_string = dynamic_fields_helper._FormatInode(event, event_data)
self.assertEqual(inode_string, '-')
def testFormatTimestampDescription(self):
"""Tests the _FormatTimestampDescription function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event = TestEvent()
timestamp_description_string = (
dynamic_fields_helper._FormatTimestampDescription(event))
self.assertEqual(timestamp_description_string,
'Metadata Modification Time')
def testGetFormattedField(self):
"""Tests the GetFormattedField function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event, event_data = containers_test_lib.CreateEventFromValues(
self._TEST_EVENTS[0])
zone_string = dynamic_fields_helper.GetFormattedField(
event, event_data, None, 'zone')
self.assertEqual(zone_string, 'UTC')
def testFormatDateTime(self):
"""Tests the _FormatDateTime function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event = TestEvent()
date_time_string = dynamic_fields_helper._FormatDateTime(event)
self.assertEqual(date_time_string, '2012-06-27T18:17:01+00:00')
event.timestamp = -9223372036854775808
date_time_string = dynamic_fields_helper._FormatDateTime(event)
self.assertEqual(date_time_string, '0000-00-00T00:00:00')
def testFormatTime(self):
"""Tests the _FormatTime function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event = TestEvent()
time_string = dynamic_fields_helper._FormatTime(event)
self.assertEqual(time_string, '18:17:01')
event.timestamp = -9223372036854775808
time_string = dynamic_fields_helper._FormatTime(event)
self.assertEqual(time_string, '--:--:--')
def testFormatTimestampDescription(self):
"""Tests the _FormatTimestampDescription function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event, event_data = containers_test_lib.CreateEventFromValues(
self._TEST_EVENTS[0])
timestamp_description_string = (
dynamic_fields_helper._FormatTimestampDescription(
event, event_data))
self.assertEqual(timestamp_description_string,
'Metadata Modification Time')
def testFormatTime(self):
"""Tests the _FormatTime function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event, event_data = containers_test_lib.CreateEventFromValues(
self._TEST_EVENTS[0])
time_string = dynamic_fields_helper._FormatTime(event, event_data)
self.assertEqual(time_string, '18:17:01')
event.timestamp = -9223372036854775808
time_string = dynamic_fields_helper._FormatTime(event, event_data)
self.assertEqual(time_string, '--:--:--')
def testFormatDate(self):
"""Tests the _FormatDate function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event, event_data = containers_test_lib.CreateEventFromValues(
self._TEST_EVENTS[0])
date_string = dynamic_fields_helper._FormatDate(event, event_data)
self.assertEqual(date_string, '2012-06-27')
event.timestamp = -9223372036854775808
date_string = dynamic_fields_helper._FormatDate(event, event_data)
self.assertEqual(date_string, '0000-00-00')
def testFormatTag(self):
"""Tests the _FormatTag function."""
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
tag_string = dynamic_fields_helper._FormatTag(None)
self.assertEqual(tag_string, '-')
event_tag = events.EventTag()
event_tag.AddLabel('one')
event_tag.AddLabel('two')
tag_string = dynamic_fields_helper._FormatTag(event_tag)
self.assertEqual(tag_string, 'one two')
def __init__(self, output_mediator):
"""Initializes an Excel Spreadsheet (XLSX) output module.
Args:
output_mediator (OutputMediator): output mediator.
"""
super(XLSXOutputModule, self).__init__(output_mediator)
self._column_widths = {}
self._current_row = 0
self._dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
self._fields = self._DEFAULT_FIELDS
self._filename = None
self._sheet = None
self._timestamp_format = self._DEFAULT_TIMESTAMP_FORMAT
self._workbook = None
def testFormatSource(self):
"""Tests the _FormatSource function."""
formatters_manager.FormattersManager.RegisterFormatter(
TestEventFormatter)
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event, event_data = containers_test_lib.CreateEventFromValues(
self._TEST_EVENTS[0])
source_string = dynamic_fields_helper._FormatSource(event, event_data)
self.assertEqual(source_string, 'Syslog')
formatters_manager.FormattersManager.DeregisterFormatter(
TestEventFormatter)
def testFormatMessageShort(self):
"""Tests the _FormatMessageShort function."""
formatters_manager.FormattersManager.RegisterFormatter(
TestEventFormatter)
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event = TestEvent()
message_short_string = dynamic_fields_helper._FormatMessageShort(event)
expected_message_short_string = (
'Reporter <CRON> PID: 8442 (pam_unix(cron:session): session closed '
'for user root)')
self.assertEqual(message_short_string, expected_message_short_string)
formatters_manager.FormattersManager.DeregisterFormatter(
TestEventFormatter)
def testFormatMessage(self):
"""Tests the _FormatMessage function."""
formatters_manager.FormattersManager.RegisterFormatter(
TestEventFormatter)
output_mediator = self._CreateOutputMediator()
dynamic_fields_helper = dynamic.DynamicFieldsHelper(output_mediator)
event, event_data = containers_test_lib.CreateEventFromValues(
self._TEST_EVENTS[0])
message_string = dynamic_fields_helper._FormatMessage(
event, event_data)
expected_message_string = (
'Reporter <CRON> PID: 8442 (pam_unix(cron:session): session closed '
'for user root)')
self.assertEqual(message_string, expected_message_string)
formatters_manager.FormattersManager.DeregisterFormatter(
TestEventFormatter)
