path: root/_pytest/test_process_message.py

from __future__ import print_function, unicode_literals

import json

from wee_slack import SlackTS


def test_process_message(realish_eventrouter, team, user_alice):
    messages = []
    messages.append(
        json.loads(
            open("_pytest/data/websocket/1485975421.33-message.json", "r").read()
        )
    )

    # test message and then change
    messages.append(
        json.loads(open("_pytest/data/websocket/1485976151.6-message.json", "r").read())
    )
    messages.append(
        json.loads(
            open("_pytest/data/websocket/1485976157.18-message.json", "r").read()
        )
    )

    # test message then deletion
    messages.append(
        json.loads(
            open("_pytest/data/websocket/1485975698.45-message.json", "r").read()
        )
    )
    messages.append(
        json.loads(
            open("_pytest/data/websocket/1485975723.85-message.json", "r").read()
        )
    )

    for m in messages:
        m["user"] = user_alice.id
        team.ws.add(m)

    realish_eventrouter.receive_ws_callback(team.team_hash, None)
    realish_eventrouter.handle_next()
    realish_eventrouter.handle_next()
    realish_eventrouter.handle_next()
    realish_eventrouter.handle_next()
    realish_eventrouter.handle_next()

    assert sum([len(channel.messages) for channel in team.channels.values()]) == 3

    unchanged_message_channel = team.channels["D3ZEQULHZ"]
    unchanged_message_ts = SlackTS("1485975421.000002")
    assert list(unchanged_message_channel.messages.keys()) == [unchanged_message_ts]
    assert (
        unchanged_message_channel.messages[unchanged_message_ts].message_json["text"]
        == "hi bob"
    )
    assert (
        "edited"
        not in unchanged_message_channel.messages[unchanged_message_ts].message_json
    )

    changed_message_channel = team.channels["C407ABS94"]
    changed_message_ts = SlackTS("1485976151.000016")
    assert list(changed_message_channel.messages.keys()) == [changed_message_ts]
    assert (
        changed_message_channel.messages[changed_message_ts].message_json["text"]
        == "referencing a <#C407ABS94|general>"
    )
    assert "edited" in changed_message_channel.messages[changed_message_ts].message_json

    deleted_message_channel = team.channels["G3ZGMF4RZ"]
    deleted_message_ts = SlackTS("1485975698.000002")
    assert list(deleted_message_channel.messages.keys()) == [deleted_message_ts]
    deleted_str = "<[color red]>(deleted)<[color reset]>"
    assert (
        deleted_message_channel.messages[deleted_message_ts].message_json["text"]
        == deleted_str
    )