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

Extract get_bot_message_handler().

Browse source
This commit is contained in:
Steve Howell 2017-11-30 08:15:15 -08:00 committed by showell
parent 4ca11fdd67
commit 0a477b0736
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
zulip_bots/zulip_bots/test_lib.py
View file

@ -26,19 +26,19 @@ from types import ModuleType
from copy import deepcopy
def get_bot_message_handler(bot_name):
# type: (str) -> Any
# message_handler is of type 'Any', since it can contain any bot's
# handler class. Eventually, we want bot's handler classes to
# inherit from a common prototype specifying the handle_message
# function.
lib_module = import_module('zulip_bots.bots.{bot}.{bot}'.format(bot=bot_name)) # type: Any
return lib_module.handler_class()
class BotTestCaseBase(TestCase):
"""Test class for common Bot test helper methods"""
bot_name = '' # type: str
def get_bot_message_handler(self):
# type: () -> Any
# message_handler is of type 'Any', since it can contain any bot's
# handler class. Eventually, we want bot's handler classes to
# inherit from a common prototype specifying the handle_message
# function.
lib_module = import_module('zulip_bots.bots.{bot}.{bot}'.format(bot=self.bot_name)) # type: Any
return lib_module.handler_class()
def setUp(self):
# type: () -> None
# Mocking ExternalBotHandler
@ -51,7 +51,7 @@ class BotTestCaseBase(TestCase):
self.mock_bot_handler.storage = StateHandler(self.mock_client)
self.mock_bot_handler.send_message.return_value = {'id': 42}
self.mock_bot_handler.send_reply.return_value = {'id': 42}
self.message_handler = self.get_bot_message_handler()
self.message_handler = get_bot_message_handler(self.bot_name)
def tearDown(self):
# type: () -> None