2017-08-11 23:58:41 +02:00
|
|
|
#!/usr/bin/env python
|
2015-09-07 18:55:28 +02:00
|
|
|
#
|
|
|
|
# A library that provides a Python interface to the Telegram Bot API
|
2020-02-02 22:08:54 +01:00
|
|
|
# Copyright (C) 2015-2020
|
2016-01-05 14:12:03 +01:00
|
|
|
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
|
2015-09-07 18:55:28 +02:00
|
|
|
#
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
2017-08-11 23:58:41 +02:00
|
|
|
# it under the terms of the GNU Lesser Public License as published by
|
2015-09-07 18:55:28 +02:00
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
|
# (at your option) any later version.
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
2017-08-11 23:58:41 +02:00
|
|
|
# GNU Lesser Public License for more details.
|
2015-09-07 18:55:28 +02:00
|
|
|
#
|
2017-08-11 23:58:41 +02:00
|
|
|
# You should have received a copy of the GNU Lesser Public License
|
2015-09-07 18:55:28 +02:00
|
|
|
# along with this program. If not, see [http://www.gnu.org/licenses/].
|
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
import pytest
|
2016-04-24 04:11:25 +02:00
|
|
|
|
2020-10-09 17:22:07 +02:00
|
|
|
from telegram import (
|
|
|
|
Message,
|
|
|
|
User,
|
|
|
|
Update,
|
|
|
|
Chat,
|
|
|
|
CallbackQuery,
|
|
|
|
InlineQuery,
|
|
|
|
ChosenInlineResult,
|
|
|
|
ShippingQuery,
|
|
|
|
PreCheckoutQuery,
|
|
|
|
Poll,
|
|
|
|
PollOption,
|
|
|
|
)
|
2020-03-29 09:52:30 +02:00
|
|
|
from telegram.poll import PollAnswer
|
2015-09-07 18:55:28 +02:00
|
|
|
|
2020-10-06 19:28:40 +02:00
|
|
|
message = Message(1, None, Chat(1, ''), from_user=User(1, '', False), text='Text')
|
2015-09-07 18:55:28 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
params = [
|
|
|
|
{'message': message},
|
|
|
|
{'edited_message': message},
|
2017-09-01 08:40:05 +02:00
|
|
|
{'callback_query': CallbackQuery(1, User(1, '', False), 'chat', message=message)},
|
2017-08-11 23:58:41 +02:00
|
|
|
{'channel_post': message},
|
|
|
|
{'edited_channel_post': message},
|
2017-09-01 08:40:05 +02:00
|
|
|
{'inline_query': InlineQuery(1, User(1, '', False), '', '')},
|
|
|
|
{'chosen_inline_result': ChosenInlineResult('id', User(1, '', False), '')},
|
|
|
|
{'shipping_query': ShippingQuery('id', User(1, '', False), '', None)},
|
|
|
|
{'pre_checkout_query': PreCheckoutQuery('id', User(1, '', False), '', 0, '')},
|
2019-08-23 21:20:41 +02:00
|
|
|
{'callback_query': CallbackQuery(1, User(1, '', False), 'chat')},
|
2020-03-29 09:52:30 +02:00
|
|
|
{'poll': Poll('id', '?', [PollOption('.', 1)], False, False, False, Poll.REGULAR, True)},
|
2020-10-09 17:22:07 +02:00
|
|
|
{'poll_answer': PollAnswer("id", User(1, '', False), [1])},
|
2017-08-11 23:58:41 +02:00
|
|
|
]
|
2015-09-07 18:55:28 +02:00
|
|
|
|
2020-10-09 17:22:07 +02:00
|
|
|
all_types = (
|
|
|
|
'message',
|
|
|
|
'edited_message',
|
|
|
|
'callback_query',
|
|
|
|
'channel_post',
|
|
|
|
'edited_channel_post',
|
|
|
|
'inline_query',
|
|
|
|
'chosen_inline_result',
|
|
|
|
'shipping_query',
|
|
|
|
'pre_checkout_query',
|
|
|
|
'poll',
|
|
|
|
'poll_answer',
|
|
|
|
)
|
2015-09-07 18:55:28 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
ids = all_types + ('callback_query_without_message',)
|
2015-09-07 18:55:28 +02:00
|
|
|
|
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
@pytest.fixture(params=params, ids=ids)
|
|
|
|
def update(request):
|
|
|
|
return Update(update_id=TestUpdate.update_id, **request.param)
|
2015-09-07 18:55:28 +02:00
|
|
|
|
|
|
|
|
2020-06-15 18:20:51 +02:00
|
|
|
class TestUpdate:
|
2017-08-11 23:58:41 +02:00
|
|
|
update_id = 868573637
|
2016-04-27 03:48:02 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
@pytest.mark.parametrize('paramdict', argvalues=params, ids=ids)
|
|
|
|
def test_de_json(self, bot, paramdict):
|
|
|
|
json_dict = {'update_id': TestUpdate.update_id}
|
|
|
|
# Convert the single update 'item' to a dict of that item and apply it to the json_dict
|
|
|
|
json_dict.update({k: v.to_dict() for k, v in paramdict.items()})
|
|
|
|
update = Update.de_json(json_dict, bot)
|
2016-04-27 03:48:02 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert update.update_id == self.update_id
|
2015-09-07 18:55:28 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
# Make sure only one thing in the update (other than update_id) is not None
|
|
|
|
i = 0
|
|
|
|
for type in all_types:
|
|
|
|
if getattr(update, type) is not None:
|
|
|
|
i += 1
|
|
|
|
assert getattr(update, type) == paramdict[type]
|
|
|
|
assert i == 1
|
2015-09-07 18:55:28 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_update_de_json_empty(self, bot):
|
|
|
|
update = Update.de_json(None, bot)
|
2015-09-07 18:55:28 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert update is None
|
2015-09-07 18:55:28 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_to_dict(self, update):
|
|
|
|
update_dict = update.to_dict()
|
2017-03-26 14:36:34 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert isinstance(update_dict, dict)
|
|
|
|
assert update_dict['update_id'] == update.update_id
|
|
|
|
for type in all_types:
|
|
|
|
if getattr(update, type) is not None:
|
|
|
|
assert update_dict[type] == getattr(update, type).to_dict()
|
2017-01-20 20:13:58 +01:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_effective_chat(self, update):
|
|
|
|
# Test that it's sometimes None per docstring
|
|
|
|
chat = update.effective_chat
|
2020-10-09 17:22:07 +02:00
|
|
|
if not (
|
|
|
|
update.inline_query is not None
|
|
|
|
or update.chosen_inline_result is not None
|
|
|
|
or (update.callback_query is not None and update.callback_query.message is None)
|
|
|
|
or update.shipping_query is not None
|
|
|
|
or update.pre_checkout_query is not None
|
|
|
|
or update.poll is not None
|
|
|
|
or update.poll_answer is not None
|
|
|
|
):
|
2017-08-11 23:58:41 +02:00
|
|
|
assert chat.id == 1
|
|
|
|
else:
|
|
|
|
assert chat is None
|
|
|
|
|
|
|
|
def test_effective_user(self, update):
|
|
|
|
# Test that it's sometimes None per docstring
|
|
|
|
user = update.effective_user
|
2020-10-09 17:22:07 +02:00
|
|
|
if not (
|
|
|
|
update.channel_post is not None
|
|
|
|
or update.edited_channel_post is not None
|
|
|
|
or update.poll is not None
|
|
|
|
):
|
2017-08-11 23:58:41 +02:00
|
|
|
assert user.id == 1
|
|
|
|
else:
|
|
|
|
assert user is None
|
|
|
|
|
|
|
|
def test_effective_message(self, update):
|
|
|
|
# Test that it's sometimes None per docstring
|
|
|
|
eff_message = update.effective_message
|
2020-10-09 17:22:07 +02:00
|
|
|
if not (
|
|
|
|
update.inline_query is not None
|
|
|
|
or update.chosen_inline_result is not None
|
|
|
|
or (update.callback_query is not None and update.callback_query.message is None)
|
|
|
|
or update.shipping_query is not None
|
|
|
|
or update.pre_checkout_query is not None
|
|
|
|
or update.poll is not None
|
|
|
|
or update.poll_answer is not None
|
|
|
|
):
|
2017-08-11 23:58:41 +02:00
|
|
|
assert eff_message.message_id == message.message_id
|
|
|
|
else:
|
|
|
|
assert eff_message is None
|
2017-01-20 20:13:58 +01:00
|
|
|
|
2017-05-14 23:29:31 +02:00
|
|
|
def test_equality(self):
|
2017-08-11 23:58:41 +02:00
|
|
|
a = Update(self.update_id, message=message)
|
|
|
|
b = Update(self.update_id, message=message)
|
|
|
|
c = Update(self.update_id)
|
|
|
|
d = Update(0, message=message)
|
2017-09-01 08:40:05 +02:00
|
|
|
e = User(self.update_id, '', False)
|
2017-05-14 23:29:31 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert a == b
|
|
|
|
assert hash(a) == hash(b)
|
|
|
|
assert a is not b
|
2017-05-14 23:29:31 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert a == c
|
|
|
|
assert hash(a) == hash(c)
|
2017-05-14 23:29:31 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert a != d
|
|
|
|
assert hash(a) != hash(d)
|
2016-04-24 04:11:25 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert a != e
|
|
|
|
assert hash(a) != hash(e)
|