def test_last_start_single(self):
"""
Ensure the last line is included as a record
if it is a start line.
"""
records = log_reader(self.last_start_single)
results = list(islice(records, 3))
self.assertEqual(deque([u'record 4\n']), next(records))
def test_read_logs(self):
"""
Ensure a deque is yielded for each log record.
Ensure last line of last record is included
when not a start line.
"""
records = log_reader(self.input_data_path)
results = list(records)
self.assertEqual(4, len(results))
self.assertEqual(
deque([u'record 1\n', u'\tline 1\n',
u'\tline 2\n', u'\tline 3\n']),
results[0])
self.assertEqual(
deque([u'record 4\n', u'\tline 1\n', u'\tline 2\n']),
results[3])
