paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-03 05:23:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

bots: Add additional service bot tests.

Browse Source
This commit is contained in:
Elliott Jin
2017-05-25 11:02:13 -07:00
committed by Tim Abbott
parent af49a23013
commit 4907f24603
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
zerver/tests/test_service_bot_system.py
View File

@@ -124,6 +124,19 @@ class TestServiceBotEventTriggers(ZulipTestCase):
self.send_message(sender_email, recipient_email, message_type, subject='', content='test')
self.assertTrue(mock_queue_json_publish.called)
@mock.patch('zerver.lib.actions.queue_json_publish')
def test_no_trigger_on_personal_message_from_bot(self, mock_queue_json_publish):
# type: (mock.Mock) -> None
for bot_type in BOT_TYPE_TO_QUEUE_NAME:
self.bot_profile.bot_type = bot_type
self.bot_profile.save()
sender_email = self.second_bot_profile.email
recipient_email = self.bot_profile.email
message_type = Recipient.PERSONAL
self.send_message(sender_email, recipient_email, message_type)
self.assertFalse(mock_queue_json_publish.called)
@mock.patch('zerver.lib.actions.queue_json_publish')
def test_trigger_on_huddle_message_from_user(self, mock_queue_json_publish):
# type: (mock.Mock) -> None
@@ -153,3 +166,16 @@ class TestServiceBotEventTriggers(ZulipTestCase):
self.send_message(sender_email, recipient_emails, message_type, subject='', content='test')
self.assertEqual(mock_queue_json_publish.call_count, 2)
mock_queue_json_publish.reset_mock()
@mock.patch('zerver.lib.actions.queue_json_publish')
def test_no_trigger_on_huddle_message_from_bot(self, mock_queue_json_publish):
# type: (mock.Mock) -> None
for bot_type in BOT_TYPE_TO_QUEUE_NAME:
self.bot_profile.bot_type = bot_type
self.bot_profile.save()
sender_email = self.second_bot_profile.email
recipient_emails = [self.user_profile.email, self.bot_profile.email]
message_type = Recipient.HUDDLE
self.send_message(sender_email, recipient_emails, message_type)
self.assertFalse(mock_queue_json_publish.called)