def setUp(self):
"""Makes preparations before running an individual test."""
output_mediator = self._CreateOutputMediator()
self._output_writer = cli_test_lib.TestOutputWriter()
self._output_module = kml.KMLOutputModule(output_mediator)
self._output_module.SetOutputWriter(self._output_writer)
self._event_object = test_lib.TestEventObject()
def testWriteFooter(self):
"""Tests the WriteFooter function."""
test_file_object = io.StringIO()
output_mediator = self._CreateOutputMediator()
output_module = kml.KMLOutputModule(output_mediator)
output_module._file_object = test_file_object
output_module.WriteFooter()
footer = test_file_object.getvalue()
self.assertEqual(footer, '</Document></kml>')
def testWriteHeader(self):
"""Tests the WriteHeader function."""
test_file_object = io.StringIO()
output_mediator = self._CreateOutputMediator()
output_module = kml.KMLOutputModule(output_mediator)
output_module._file_object = test_file_object
output_module.WriteHeader()
expected_header = (
'<?xml version="1.0" encoding="utf-8"?>'
'<kml xmlns="http://www.opengis.net/kml/2.2"><Document>')
header = test_file_object.getvalue()
self.assertEqual(header, expected_header)
def testWriteEventBody(self):
"""Tests the WriteEventBody function."""
# Test event without geo-location.
test_file_object = io.StringIO()
output_mediator = self._CreateOutputMediator()
output_module = kml.KMLOutputModule(output_mediator)
output_module._file_object = test_file_object
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[0]))
output_module.WriteEventBody(event, event_data, event_data_stream,
None)
event_body = test_file_object.getvalue()
self.assertEqual(event_body, '')
# Test event with geo-location.
test_file_object = io.StringIO()
output_mediator = self._CreateOutputMediator()
output_module = kml.KMLOutputModule(output_mediator)
output_module._file_object = test_file_object
event, event_data, event_data_stream = (
containers_test_lib.CreateEventFromValues(self._TEST_EVENTS[1]))
output_module.WriteEventBody(event, event_data, event_data_stream,
None)
event_body = test_file_object.getvalue()
event_identifier = event.GetIdentifier()
event_identifier_string = event_identifier.CopyToString()
if sys.platform.startswith('win'):
# The dict comparison is very picky on Windows hence we
# have to make sure the drive letter is in the same case.
expected_os_location = os.path.abspath('\\{0:s}'.format(
os.path.join('cases', 'image.dd')))
else:
expected_os_location = '{0:s}{1:s}'.format(
os.path.sep, os.path.join('cases', 'image.dd'))
expected_event_body = (
'<Placemark><name>{0:s}</name><description>'
'+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-'
'+-+-+-+-+-+-\n'
'[Timestamp]:\n'
' 2012-06-27T18:17:01.000000Z\n'
'\n'
'[Pathspec]:\n'
' type: OS, location: {1:s}\n'
' type: TSK, inode: 15, location: /var/log/syslog.1\n'
'\n'
'[Reserved attributes]:\n'
' {{data_type}} test:output\n'
' {{display_name}} TSK:/var/log/syslog.1\n'
' {{filename}} /var/log/syslog.1\n'
' {{hostname}} ubuntu\n'
' {{inode}} 15\n'
' {{username}} root\n'
'\n'
'[Additional attributes]:\n'
' {{latitude}} 37.4222899014\n'
' {{longitude}} -122.082203543\n'
' {{text}} Reporter <CRON> PID: |8442| '
'(pam_unix(cron:session): session\n'
' closed for user root)\n'
'\n'
'</description>'
'<Point><coordinates>-122.082203543,37.4222899014</coordinates>'
'</Point></Placemark>').format(event_identifier_string,
expected_os_location)
self.assertEqual(event_body.split('\n'),
expected_event_body.split('\n'))