def test_is_reply_to_bot_message_handler(self, mocker, fake_message_dict, fake_user_dict):
fake_reply_message_dict = fake_message_dict.copy()
bot = TululBot('TOKEN')
bot_id = 12345
class FakeUser:
def __init__(self, id):
self.id = id
bot.user = FakeUser(bot_id)
mock_message_handler = mocker.patch.object(bot._telebot, 'message_handler',
autospec=True)
fake_user_dict['id'] = bot_id
bot_message = 'Hah?'
fake_message_dict['text'] = bot_message
fake_message_dict['from'] = fake_user_dict
fake_reply_message_dict['reply_to_message'] = fake_message_dict
fake_reply_message = Message.from_dict(fake_reply_message_dict)
@bot.message_handler(is_reply_to_bot=bot_message)
def handle(message):
pass
args, kwargs = mock_message_handler.call_args
assert len(args) == 0
assert len(kwargs) == 1
assert 'func' in kwargs
func = kwargs['func']
assert func(fake_reply_message)
def test_handle_new_message(self, mocker, fake_message):
bot = TululBot('TOKEN')
mock_process_new_messages = mocker.patch.object(bot._telebot, 'process_new_messages',
autospec=True)
bot.handle_new_message(fake_message)
mock_process_new_messages.assert_called_once_with([fake_message])
def test_reply_to_with_preview_disabled(self, mocker, fake_message):
bot = TululBot('TOKEN')
mock_reply_to = mocker.patch.object(bot._telebot, 'reply_to', autospec=True)
text = 'Hello world'
bot.reply_to(fake_message, text, disable_preview=True)
mock_reply_to.assert_called_once_with(fake_message, text,
disable_web_page_preview=True,
reply_markup=None)
def test_get_me(self, mocker):
bot = TululBot('TOKEN')
return_value = 'askldfjlkf'
mock_get_me = mocker.patch.object(bot._telebot, 'get_me', autospec=True,
return_value=return_value)
rv = bot.get_me()
assert rv == return_value
mock_get_me.assert_called_once_with()
def test_set_webhook(self, mocker):
bot = TululBot('TOKEN')
return_value = 'some return value'
webhook_url = 'some url'
mock_set_webhook = mocker.patch.object(bot._telebot, 'set_webhook',
return_value=return_value, autospec=True)
rv = bot.set_webhook(webhook_url)
assert rv == return_value
mock_set_webhook.assert_called_once_with(webhook_url)
def test_send_message(self, mocker):
bot = TululBot('TOKEN')
return_value = 'some return value'
mock_send_message = mocker.patch.object(bot._telebot, 'send_message',
return_value=return_value,
autospec=True)
chat_id = 12345
text = 'Hello world'
rv = bot.send_message(chat_id, text)
assert rv == return_value
mock_send_message.assert_called_once_with(chat_id, text)
def test_forward_message(self, mocker):
bot = TululBot('TOKEN')
return_value = 'some return value'
mock_forward_message = mocker.patch.object(bot._telebot, 'forward_message',
return_value=return_value,
autospec=True)
chat_id = 12345
from_chat_id = 67890
message_id = 42
rv = bot.forward_message(chat_id, from_chat_id, message_id)
assert rv == return_value
mock_forward_message.assert_called_once_with(chat_id, from_chat_id, message_id)
def test_reply_to_with_force_reply(self, mocker, fake_message):
bot = TululBot('TOKEN')
mock_reply_to = mocker.patch.object(bot._telebot, 'reply_to', autospec=True)
text = 'dummy text'
bot.reply_to(fake_message, text, force_reply=True)
args, kwargs = mock_reply_to.call_args
assert args == (fake_message, text)
assert len(kwargs) == 2
assert 'disable_web_page_preview' in kwargs
assert not kwargs['disable_web_page_preview']
assert 'reply_markup' in kwargs
assert isinstance(kwargs['reply_markup'], types.ForceReply)
def test_reply_to(self, mocker, fake_message):
bot = TululBot('TOKEN')
return_value = 'some return value'
mock_reply_to = mocker.patch.object(bot._telebot, 'reply_to',
return_value=return_value,
autospec=True)
text = 'Hello world'
rv = bot.reply_to(fake_message, text)
assert rv == return_value
mock_reply_to.assert_called_once_with(fake_message, text,
disable_web_page_preview=False,
reply_markup=None)
def test_create_is_reply_to_filter(self, mocker, fake_message_dict, fake_user_dict):
fake_replied_message_dict = fake_message_dict.copy()
fake_message = types.Message.de_json(fake_message_dict)
fake_replied_message = types.Message.de_json(fake_replied_message_dict)
bot_user = types.User.de_json(fake_user_dict)
bot_message = 'Message text from bot goes here'
fake_replied_message.text = bot_message
fake_replied_message.from_user = bot_user
fake_message.reply_to_message = fake_replied_message
bot = TululBot('TOKEN')
bot.user = bot_user
assert bot.create_is_reply_to_filter(bot_message)(fake_message)
assert not bot.create_is_reply_to_filter('foo bar')(fake_message)
def test_create_is_reply_to_filter(self, mocker, fake_message_dict,
fake_user_dict):
fake_replied_message_dict = fake_message_dict.copy()
fake_message = types.Message.de_json(fake_message_dict)
fake_replied_message = types.Message.de_json(fake_replied_message_dict)
bot_user = types.User.de_json(fake_user_dict)
bot_message = 'Message text from bot goes here'
fake_replied_message.text = bot_message
fake_replied_message.from_user = bot_user
fake_message.reply_to_message = fake_replied_message
bot = TululBot('TOKEN')
bot.user = bot_user
assert bot.create_is_reply_to_filter(bot_message)(fake_message)
assert not bot.create_is_reply_to_filter('foo bar')(fake_message)
def test_user_property(self, mocker, fake_user):
bot = TululBot('TOKEN')
mock_get_me = mocker.patch.object(bot,
'get_me',
autospec=True,
return_value=fake_user)
rv = bot.user
assert rv == fake_user
mock_get_me.assert_called_once_with()
import traceback
from flask import Flask, abort, request
app = Flask(__name__)
app.config.from_object('{}.config'.format(__name__))
from telebot import types # noqa: E402
from tululbot.utils import TululBot # noqa: E402
bot = TululBot(app.config['TELEGRAM_BOT_TOKEN']) # Must be before importing commands
from tululbot import commands # noqa
# Why do this? See https://core.telegram.org/bots/api#setwebhook
webhook_url_path = '/{}'.format(app.config['TELEGRAM_BOT_TOKEN'])
webhook_url_base = app.config['WEBHOOK_HOST']
app.logger.setLevel(app.config['LOG_LEVEL'])
@app.route(webhook_url_path, methods=['POST'])
def main():
json_data = request.get_json(silent=True)
if json_data is not None:
update = types.Update.de_json(json_data)
app.logger.debug('Get an update with id %s', update.update_id)
if update.message is not None:
bot.process_new_messages([update.message])
return 'OK'
else:
import traceback
from flask import Flask, abort, request
app = Flask(__name__)
app.config.from_object('{}.config'.format(__name__))
from telebot import types
from tululbot.utils import TululBot
bot = TululBot(app.config['TELEGRAM_BOT_TOKEN']) # Must be before importing commands
from tululbot import commands # noqa
# Why do this? See https://core.telegram.org/bots/api#setwebhook
webhook_url_path = '/{}'.format(app.config['TELEGRAM_BOT_TOKEN'])
webhook_url_base = app.config['WEBHOOK_HOST']
app.logger.setLevel(app.config['LOG_LEVEL'])
@app.route(webhook_url_path, methods=['POST'])
def main():
json_data = request.get_json(silent=True)
if json_data is not None:
update = types.Update.de_json(json_data)
app.logger.debug('Get an update with id %s', update.update_id)
if update.message is not None:
bot.process_new_messages([update.message])
return 'OK'
else:
