Refactor: Make StateHandler() function independent.
This refactor makes the nested class 'StateHandler' in the
file /contrib_bots/bot_lib.py independent class. It previously
was nested in 'run_message_handler_for_bot' function.
This is done to write a cleaner test file for contrib_bots using
mock library.
@@ -71,13 +71,6 @@ class BotHandlerApi(object):
content=response,
))
-def run_message_handler_for_bot(lib_module, quiet, config_file):
- # Make sure you set up your ~/.zuliprc
- client = Client(config_file=config_file)
- restricted_client = BotHandlerApi(client)
-
- message_handler = lib_module.handler_class()
-
class StateHandler(object):
def __init__(self):
self.state = None
@@ -88,6 +81,13 @@ def run_message_handler_for_bot(lib_module, quiet, config_file):
def get_state(self):
return self.state
+def run_message_handler_for_bot(lib_module, quiet, config_file):
+ # Make sure you set up your ~/.zuliprc
+ client = Client(config_file=config_file)
+ restricted_client = BotHandlerApi(client)
+
+ message_handler = lib_module.handler_class()
+
state_handler = StateHandler()
if not quiet:
