_pytest/test_process_message.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

import json

from wee_slack import render

def test_process_message(monkeypatch, realish_eventrouter, mock_websocket):

    e = realish_eventrouter

    t = e.teams.keys()[0]
    u = e.teams[t].users.keys()[0]

    user = e.teams[t].users[u]
    #print user

    socket = mock_websocket
    e.teams[t].ws = socket

    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/1485976157.18-message.json', 'r').read()))
    messages.append(json.loads(open('_pytest/data/websocket/1485976151.6-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.id
        socket.add(m)

    e.receive_ws_callback(t)
    e.handle_next()

    e.receive_ws_callback(t)
    e.handle_next()

    e.receive_ws_callback(t)
    e.handle_next()

    e.receive_ws_callback(t)
    e.handle_next()