def test_autodiscovery_failure_module():
apps = ["example1", "example2", "example3"]
with patch.object(core, "import_module") as m, patch.object(
core.logger, "error") as logger_mock:
m.side_effect = ImportError()
core.autodiscovery(apps)
assert m.call_count == 3
assert logger_mock.call_count == 3
def create_bot(self, app):
token = app.config.get("API_TOKEN")
autodiscovery(app.config.get("APPS", []))
self.bot = telegram.Bot(token=token)
self.dispatcher = Dispatcher(self.bot, None, workers=0)
for handler in get_handlers():
self.dispatcher.add_handler(handler)
# log all errors
self.dispatcher.add_error_handler(self.error)
def create_bot(self, app, persistence):
self.app = app
self.persistence = persistence or DictPersistence()
token = app.config.get("API_TOKEN")
autodiscovery(app.config.get("APPS", []))
self.bot = telegram.Bot(token=token)
self.dispatcher = Dispatcher(self.bot,
None,
workers=0,
use_context=True,
persistence=self.persistence)
setup_handlers(self.dispatcher)
# log all errors
self.dispatcher.add_error_handler(self.error)
def main():
if not settings.API_TOKEN:
logger.critical("Telegram API Token is missing(TELEGRAM_API_TOKEN)")
return 1
updater = Updater(settings.API_TOKEN)
autodiscovery(settings.APPS)
dp = updater.dispatcher
for handler in get_handlers():
dp.add_handler(handler)
# log all errors
dp.add_error_handler(error)
updater.start_polling()
updater.idle()
return 0
def test_autodiscovery():
apps = ["example1", "example2", "example3"]
with patch.object(core, "import_module") as m:
core.autodiscovery(apps)
assert m.call_count == 3