bots: Enable zulip_bot_output.py to understand default commands.

Author: neiljp

zulip_bots/zulip_bots/zulip_bot_output.py

@@ -12,6 +12,7 @@
 from mock import MagicMock, patch
 from zulip_bots.lib import StateHandler
 from zulip_bots.lib import ExternalBotHandler
+from zulip_bots.lib import get_bot_details, setup_default_commands, updated_default_commands
 from zulip_bots.provision import provision_bot
 from zulip_bots.run import import_module_from_source
 
@@ -66,18 +67,32 @@ def main():
     message = {'content': args.message, 'sender_email': '[email protected]'}
     message_handler = lib_module.handler_class()
 
+    bot_details = get_bot_details(message_handler, args.bot)
+
+    default_commands = setup_default_commands(bot_details, message_handler)
+    updated_defaults = updated_default_commands(default_commands, bot_details)
+
     with patch('zulip.Client') as mock_client:
         mock_bot_handler = ExternalBotHandler(mock_client, bot_dir)
         mock_bot_handler.send_reply = MagicMock()
         mock_bot_handler.send_message = MagicMock()
         mock_bot_handler.update_message = MagicMock()
         if hasattr(message_handler, 'initialize') and callable(message_handler.initialize):
             message_handler.initialize(mock_bot_handler)
-        message_handler.handle_message(
-            message=message,
-            bot_handler=mock_bot_handler,
-            state_handler=StateHandler()
-        )
+
+        # Check if command handled in library first; if not, then delegate to bot
+        handled = False
+        for command in updated_defaults:
+            if command == message['content']:
+                mock_bot_handler.send_reply(message,
+                                            updated_defaults[command]['action']())
+                handled = True
+        if not handled:
+            message_handler.handle_message(
+                message=message,
+                bot_handler=mock_bot_handler,
+                state_handler=StateHandler()
+            )
         print("On sending {} bot the message \"{}\"".format(bot_name, args.message))
         # send_reply and send_message have slightly arguments; the
         # following takes that into account.