def test_threadMessages(self):
"""Test MessageSet.threadMessages."""
messages = self.createTestMessages()
message1, message2, message3, message4 = messages
threads = MessageSet.threadMessages(messages)
self.assertEqual([(message1, [(message2, [(message4, [])]),
(message3, [])])], threads)
def test_flattenThreads(self):
"""Test MessageSet.flattenThreads."""
messages = self.createTestMessages()
message1, message2, message3, message4 = messages
threads = MessageSet.threadMessages(messages)
flattened = list(MessageSet.flattenThreads(threads))
expected = [(0, message1), (1, message2), (2, message4), (1, message3)]
self.assertEqual(expected, flattened)
def test_flattenThreads(self):
"""Test MessageSet.flattenThreads."""
messages = self.createTestMessages()
message1, message2, message3, message4 = messages
threads = MessageSet.threadMessages(messages)
flattened = list(MessageSet.flattenThreads(threads))
expected = [(0, message1), (1, message2), (2, message4), (1, message3)]
self.assertEqual(expected, flattened)
def test_threadMessages(self):
"""Test MessageSet.threadMessages."""
messages = self.createTestMessages()
message1, message2, message3, message4 = messages
threads = MessageSet.threadMessages(messages)
self.assertEqual(
[(message1, [(message2, [(message4, [])]), (message3, [])])],
threads)
