77 lines
3.1 KiB
Python
77 lines
3.1 KiB
Python
import mock
|
|
from typing import Any, Dict
|
|
import unittest
|
|
from .server_test_lib import BotServerTestCase
|
|
import six
|
|
|
|
|
|
class BotServerTests(BotServerTestCase):
|
|
class MockMessageHandler(object):
|
|
def handle_message(self, message: Dict[str, str], bot_handler: Any) -> None:
|
|
assert message == {'key': "test message"}
|
|
|
|
class MockLibModule(object):
|
|
def handler_class(self) -> Any:
|
|
return BotServerTests.MockMessageHandler()
|
|
|
|
@mock.patch('zulip_botserver.server.ExternalBotHandler')
|
|
def test_successful_request(self, mock_ExternalBotHandler: mock.Mock) -> None:
|
|
available_bots = ['helloworld']
|
|
bots_config = {
|
|
'helloworld': {
|
|
'email': 'helloworld-bot@zulip.com',
|
|
'key': '123456789qwertyuiop',
|
|
'site': 'http://localhost',
|
|
}
|
|
}
|
|
self.assert_bot_server_response(available_bots=available_bots,
|
|
bots_config=bots_config,
|
|
check_success=True)
|
|
|
|
@mock.patch('zulip_botserver.server.ExternalBotHandler')
|
|
def test_successful_request_from_two_bots(self, mock_ExternalBotHandler: mock.Mock) -> None:
|
|
available_bots = ['helloworld', 'help']
|
|
bots_config = {
|
|
'helloworld': {
|
|
'email': 'helloworld-bot@zulip.com',
|
|
'key': '123456789qwertyuiop',
|
|
'site': 'http://localhost',
|
|
},
|
|
'help': {
|
|
'email': 'help-bot@zulip.com',
|
|
'key': '123456789qwertyuiop',
|
|
'site': 'http://localhost',
|
|
}
|
|
}
|
|
self.assert_bot_server_response(available_bots=available_bots,
|
|
bots_config=bots_config,
|
|
check_success=True)
|
|
|
|
def test_bot_module_not_exists(self) -> None:
|
|
self.assert_bot_server_response(available_bots=[],
|
|
payload_url="/bots/not_supported_bot",
|
|
check_success=False)
|
|
|
|
@mock.patch('logging.error')
|
|
@mock.patch('zulip_bots.lib.StateHandler')
|
|
def test_wrong_bot_credentials(self, mock_StateHandler: mock.Mock, mock_LoggingError: mock.Mock) -> None:
|
|
available_bots = ['nonexistent-bot']
|
|
bots_config = {
|
|
'nonexistent-bot': {
|
|
'email': 'helloworld-bot@zulip.com',
|
|
'key': '123456789qwertyuiop',
|
|
'site': 'http://localhost',
|
|
}
|
|
}
|
|
# TODO: The following passes mypy, though the six stubs don't match the
|
|
# unittest ones, so we could file a mypy bug to improve this.
|
|
six.assertRaisesRegex(self,
|
|
ImportError,
|
|
"Bot \"nonexistent-bot\" doesn't exists. Please "
|
|
"make sure you have set up the flaskbotrc file correctly.",
|
|
lambda: self.assert_bot_server_response(available_bots=available_bots,
|
|
bots_config=bots_config))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|