bot tests: Add StubBotTestCase.mock_config_info().
We now auto-initialize bots in verify_reply() and have a mock_config_info() helper that can override the new StubBotHandler.get_config_info().
This commit is contained in:
parent
be9570c9c4
commit
87662da139
|
@ -59,6 +59,10 @@ class StubBotHandler:
|
|||
# type: (Dict[str, Any]) -> None
|
||||
self.message_server.update(message)
|
||||
|
||||
def get_config_info(self, bot_name, optional=False):
|
||||
# type: (str, bool) -> Dict[str, Any]
|
||||
return None
|
||||
|
||||
def unique_reply(self):
|
||||
# type: () -> Dict[str, Any]
|
||||
responses = [
|
||||
|
@ -103,6 +107,9 @@ class StubBotTestCase(TestCase):
|
|||
bot = get_bot_message_handler(self.bot_name)
|
||||
bot_handler = StubBotHandler()
|
||||
|
||||
if hasattr(bot, 'initialize'):
|
||||
bot.initialize(bot_handler)
|
||||
|
||||
message = dict(
|
||||
sender_email='foo@example.com',
|
||||
content=request,
|
||||
|
@ -140,6 +147,10 @@ class StubBotTestCase(TestCase):
|
|||
http_data = read_bot_fixture_data(self.bot_name, test_name)
|
||||
return mock_http_conversation(http_data)
|
||||
|
||||
def mock_config_info(self, config_info):
|
||||
# type: (Dict[str, str]) -> Any
|
||||
return patch('zulip_bots.test_lib.StubBotHandler.get_config_info', return_value=config_info)
|
||||
|
||||
def get_bot_message_handler(bot_name):
|
||||
# type: (str) -> Any
|
||||
# message_handler is of type 'Any', since it can contain any bot's
|
||||
|
|
Loading…
Reference in a new issue