def parse_conversations(messages, time_range):
for conversation in group_into_conversations(messages):
try:
gp2gp_conversation = parse_conversation(conversation)
if time_range.contains(gp2gp_conversation.request_started.time):
yield gp2gp_conversation
except ConversationMissingStart:
pass
def test_group_into_conversations_produces_correct_conversations():
message_one = build_message(conversation_id="abc")
message_two = build_message(conversation_id="xyz")
messages = [message_one, message_two]
expected = [Conversation("abc", [message_one]), Conversation("xyz", [message_two])]
actual = group_into_conversations(messages)
assert list(actual) == expected
def test_group_into_conversations_sorts_messages_within_conversations():
message_one = build_message(conversation_id="abc", time=datetime(year=2020, month=6, day=6))
message_two = build_message(conversation_id="abc", time=datetime(year=2020, month=6, day=5))
messages = [message_one, message_two]
expected = [Conversation("abc", [message_two, message_one])]
actual = group_into_conversations(messages)
assert list(actual) == expected
def process_messages(messages):
return group_into_conversations(messages)