diff --git a/zulip_bots/zulip_bots/test_lib.py b/zulip_bots/zulip_bots/test_lib.py index 2b80906..f1f61fb 100755 --- a/zulip_bots/zulip_bots/test_lib.py +++ b/zulip_bots/zulip_bots/test_lib.py @@ -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