diff --git a/tests/test_messages.py b/tests/test_messages.py index 0ef844f..555a79a 100644 --- a/tests/test_messages.py +++ b/tests/test_messages.py @@ -20,3 +20,27 @@ def test_message_from(): assert message.trade.currency == 'chaos' assert message.trade.item == 'level 21 23% Vaal Impurity of Lightning' assert message.trade.league == 'Ritual' + + +def test_message_to(): + text_to = '2021/01/24 23:11:23 17039703 bb2 [INFO Client 10144] @To EraseAndDelete: Thank you & good luck!' + message = trader.parse_log(text_to) + assert message.user == 'EraseAndDelete' + assert message.message == 'Thank you & good luck!' + assert message.date == datetime(2021, 1, 24, 23, 11, 23) + assert message.channel == Channel.WHISPER + assert message.guild is None + assert message.to_from == 'To' + assert message.trade is None + + +def test_message_global(): + text_global = '2021/02/10 19:01:56 2720500 bb2 [INFO Client 3464] #HarvestScarab: so by the time it was finally juiced you couldnt do it' + message = trader.parse_log(text_global) + assert message.user == 'HarvestScarab' + assert message.message == 'so by the time it was finally juiced you couldnt do it' + assert message.date == datetime(2021, 2, 10, 19, 1, 56) + assert message.channel == Channel.GLOBAL + assert message.guild is None + assert message.to_from is None + assert message.trade is None