diff options
Diffstat (limited to '_pytest/test_process_message.py')
-rw-r--r-- | _pytest/test_process_message.py | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/_pytest/test_process_message.py b/_pytest/test_process_message.py new file mode 100644 index 0000000..6c58336 --- /dev/null +++ b/_pytest/test_process_message.py @@ -0,0 +1,40 @@ + +import wee_slack +import pytest +import json +from collections import defaultdict + + +def test_process_message(slack_debug, monkeypatch, myservers, mychannels, myusers): + called = defaultdict(int) + wee_slack.servers = myservers + wee_slack.channels = mychannels + wee_slack.users = myusers + wee_slack.message_cache = {} + wee_slack.servers[0].users = myusers + + def mock_buffer_prnt(*args): + called['buffer_prnt'] += 1 + monkeypatch.setattr(wee_slack.Channel, 'buffer_prnt', mock_buffer_prnt) + +# def mock_buffer_prnt_changed(*args): +# called['buffer_prnt_changed'] += 1 +# print args +# monkeypatch.setattr(wee_slack.Channel, 'buffer_prnt_changed', mock_buffer_prnt_changed) + + + messages = [] + messages.append( json.loads(open('_pytest/data/message-normal.json', 'r').read()) ) + messages.append( json.loads(open('_pytest/data/message-normal2.json', 'r').read()) ) + messages.append( json.loads(open('_pytest/data/message-changed.json', 'r').read()) ) + messages.append( json.loads(open('_pytest/data/message-deleted.json', 'r').read()) ) + for m in messages: + wee_slack.process_message(m) + print called +# assert called['buffer_prnt'] == 2 +# assert called['buffer_prnt_changed'] == 1 + + + + + |