xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

bot lib tests: Add test_send_reply.

Browse source
This commit is contained in:
neiljp (Neil Pilgrim) 2017-11-29 15:00:02 -08:00 committed by showell
parent efeac92d42
commit 39601d190c
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
zulip_bots/zulip_bots/lib_tests.py
View file

@ -1,4 +1,5 @@
from unittest import TestCase from unittest import TestCase
from mock import MagicMock
from zulip_bots.lib import ( from zulip_bots.lib import (
ExternalBotHandler, ExternalBotHandler,
StateHandler, StateHandler,
@ -58,3 +59,25 @@ class LibTest(TestCase):
state_handler = StateHandler(client) state_handler = StateHandler(client)
val = state_handler.get('key') val = state_handler.get('key')
self.assertEqual(val, [1, 2, 3]) self.assertEqual(val, [1, 2, 3])
def test_send_reply(self):
client = FakeClient()
profile = client.get_profile()
handler = ExternalBotHandler(
client=client,
root_dir=None,
bot_details=None,
bot_config_file=None
)
to = {'email': 'Some@User'}
expected = [({'type': 'private', 'display_recipient': [to]},
{'type': 'private', 'to': [to['email']]}),
({'type': 'private', 'display_recipient': [to, profile]},
{'type': 'private', 'to': [to['email']]}),
({'type': 'stream', 'display_recipient': 'Stream name', 'subject': 'Topic'},
{'type': 'stream', 'to': 'Stream name', 'subject': 'Topic'})]
response_text = "Response"
for test in expected:
client.send_message = MagicMock()
handler.send_reply(test[0], response_text)
client.send_message.assert_called_once_with(dict(test[1], content=response_text))