75 lines
3.5 KiB
Python
75 lines
3.5 KiB
Python
from unittest.mock import patch
|
|
from zulip_bots.test_lib import BotTestCase, DefaultTests
|
|
from requests.exceptions import ConnectionError
|
|
|
|
class TestFlockBot(BotTestCase, DefaultTests):
|
|
bot_name = "flock"
|
|
normal_config = {"token": "12345"}
|
|
|
|
message_config = {
|
|
"token": "12345",
|
|
"text": "Ricky: test message",
|
|
"to": "u:somekey"
|
|
}
|
|
|
|
help_message = '''
|
|
You can send messages to any Flock user associated with your account from Zulip.
|
|
*Syntax*: **@botname to: message** where `to` is **firstName** of recipient.
|
|
'''
|
|
|
|
def test_bot_responds_to_empty_message(self) -> None:
|
|
self.verify_reply('', self.help_message)
|
|
|
|
def test_help_message(self) -> None:
|
|
self.verify_reply('', self.help_message)
|
|
|
|
def test_fetch_id_connection_error(self) -> None:
|
|
with self.mock_config_info(self.normal_config), \
|
|
patch('requests.get', side_effect=ConnectionError()), \
|
|
patch('logging.exception'):
|
|
self.verify_reply('tyler: Hey tyler', "Uh-Oh, couldn\'t process the request \
|
|
right now.\nPlease try again later")
|
|
|
|
def test_response_connection_error(self) -> None:
|
|
with self.mock_config_info(self.message_config), \
|
|
patch('requests.get', side_effect=ConnectionError()), \
|
|
patch('logging.exception'):
|
|
self.verify_reply('Ricky: test message', "Uh-Oh, couldn\'t process the request \
|
|
right now.\nPlease try again later")
|
|
|
|
def test_no_recipient_found(self) -> None:
|
|
bot_response = "No user found. Make sure you typed it correctly."
|
|
with self.mock_config_info(self.normal_config), \
|
|
self.mock_http_conversation('test_no_recipient_found'):
|
|
self.verify_reply('david: hello', bot_response)
|
|
|
|
def test_found_invalid_recipient(self) -> None:
|
|
bot_response = "Found user is invalid."
|
|
with self.mock_config_info(self.normal_config), \
|
|
self.mock_http_conversation('test_found_invalid_recipient'):
|
|
self.verify_reply('david: hello', bot_response)
|
|
|
|
@patch('zulip_bots.bots.flock.flock.get_recipient_id')
|
|
def test_message_send_connection_error(self, get_recipient_id: str) -> None:
|
|
bot_response = "Uh-Oh, couldn't process the request right now.\nPlease try again later"
|
|
get_recipient_id.return_value = ["u:userid", None]
|
|
with self.mock_config_info(self.normal_config), \
|
|
patch('requests.get', side_effect=ConnectionError()), \
|
|
patch('logging.exception'):
|
|
self.verify_reply('Rishabh: hi there', bot_response)
|
|
|
|
@patch('zulip_bots.bots.flock.flock.get_recipient_id')
|
|
def test_message_send_success(self, get_recipient_id: str) -> None:
|
|
bot_response = "Message sent."
|
|
get_recipient_id.return_value = ["u:userid", None]
|
|
with self.mock_config_info(self.normal_config), \
|
|
self.mock_http_conversation('test_message_send_success'):
|
|
self.verify_reply('Rishabh: hi there', bot_response)
|
|
|
|
@patch('zulip_bots.bots.flock.flock.get_recipient_id')
|
|
def test_message_send_failed(self, get_recipient_id: str) -> None:
|
|
bot_response = "Message sending failed :slightly_frowning_face:. Please try again."
|
|
get_recipient_id.return_value = ["u:invalid", None]
|
|
with self.mock_config_info(self.normal_config), \
|
|
self.mock_http_conversation('test_message_send_failed'):
|
|
self.verify_reply('Rishabh: hi there', bot_response)
|