def testFormatVersion(self):
"""Tests the _FormatVersion function."""
output_mediator = self._CreateOutputMediator()
formatting_helper = l2t_csv.L2TCSVFieldFormattingHelper(output_mediator)
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
version_string = formatting_helper._FormatVersion(
event, event_data, event_data_stream)
self.assertEqual(version_string, '2')
def testFormatType(self):
"""Tests the _FormatType function."""
output_mediator = self._CreateOutputMediator()
formatting_helper = l2t_csv.L2TCSVFieldFormattingHelper(output_mediator)
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
type_string = formatting_helper._FormatType(
event, event_data, event_data_stream)
self.assertEqual(type_string, 'Content Modification Time')
def testFormatDate(self):
"""Tests the _FormatDate function."""
output_mediator = self._CreateOutputMediator()
formatting_helper = l2t_csv.L2TCSVFieldFormattingHelper(
output_mediator)
# Test with event.date_time
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
date_string = formatting_helper._FormatDate(event, event_data,
event_data_stream)
self.assertEqual(date_string, '06/27/2012')
output_mediator.SetTimezone('Australia/Sydney')
date_string = formatting_helper._FormatDate(event, event_data,
event_data_stream)
self.assertEqual(date_string, '06/28/2012')
output_mediator.SetTimezone('UTC')
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
event.date_time._time_zone_offset = 600
date_string = formatting_helper._FormatDate(event, event_data,
event_data_stream)
self.assertEqual(date_string, '06/27/2012')
# Test with event.is_local_time
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
event.timestamp += 600 * 60 * 1000000
event.date_time.is_local_time = True
date_string = formatting_helper._FormatDate(event, event_data,
event_data_stream)
self.assertEqual(date_string, '06/28/2012')
# Test with event.timestamp
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
event.date_time = None
date_string = formatting_helper._FormatDate(event, event_data,
event_data_stream)
self.assertEqual(date_string, '06/27/2012')
event.timestamp = -9223372036854775808
date_string = formatting_helper._FormatDate(event, event_data,
event_data_stream)
self.assertEqual(date_string, '00/00/0000')
def testFormatDate(self):
"""Tests the _FormatDate function."""
output_mediator = self._CreateOutputMediator()
formatting_helper = l2t_csv.L2TCSVFieldFormattingHelper(output_mediator)
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
date_string = formatting_helper._FormatDate(
event, event_data, event_data_stream)
self.assertEqual(date_string, '06/27/2012')
event.timestamp = -9223372036854775808
date_string = formatting_helper._FormatDate(
event, event_data, event_data_stream)
self.assertEqual(date_string, '00/00/0000')
def testFormatExtraAttributes(self):
"""Tests the _FormatExtraAttributes function."""
output_mediator = self._CreateOutputMediator()
formatters_directory_path = self._GetTestFilePath(['formatters'])
output_mediator.ReadMessageFormattersFromDirectory(
formatters_directory_path)
formatting_helper = l2t_csv.L2TCSVFieldFormattingHelper(output_mediator)
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
extra_attributes_string = formatting_helper._FormatExtraAttributes(
event, event_data, event_data_stream)
expected_extra_attributes_string = (
'a_binary_field: binary; '
'my_number: 123; '
'some_additional_foo: True')
self.assertEqual(extra_attributes_string, expected_extra_attributes_string)
def testFormatExtraAttributes(self):
"""Tests the _FormatExtraAttributes function."""
output_mediator = self._CreateOutputMediator()
formatting_helper = l2t_csv.L2TCSVFieldFormattingHelper(output_mediator)
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
formatters_manager.FormattersManager.RegisterFormatter(
formatters_test_lib.TestEventFormatter)
try:
extra_attributes_string = formatting_helper._FormatExtraAttributes(
event, event_data, event_data_stream)
finally:
formatters_manager.FormattersManager.DeregisterFormatter(
formatters_test_lib.TestEventFormatter)
expected_extra_attributes_string = (
'a_binary_field: binary; '
'my_number: 123; '
'some_additional_foo: True')
self.assertEqual(extra_attributes_string, expected_extra_attributes_string)