python-telegram-bot/tests/test_contact.py

105 lines
3.5 KiB
Python
Raw Normal View History

2016-05-15 02:39:11 +02:00
#!/usr/bin/env python
2015-09-07 18:42:51 +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
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
2015-09-07 18:42:51 +02:00
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser Public License as published by
2015-09-07 18:42:51 +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
# GNU Lesser Public License for more details.
2015-09-07 18:42:51 +02:00
#
# You should have received a copy of the GNU Lesser Public License
2015-09-07 18:42:51 +02:00
# along with this program. If not, see [http://www.gnu.org/licenses/].
import pytest
from telegram import Contact, Voice
2015-09-07 18:42:51 +02:00
@pytest.fixture(scope='class')
def contact():
return Contact(
TestContact.phone_number,
TestContact.first_name,
TestContact.last_name,
TestContact.user_id,
)
2015-09-07 18:42:51 +02:00
class TestContact:
phone_number = '+11234567890'
first_name = 'Leandro'
last_name = 'Toledo'
user_id = 23
2015-09-07 18:42:51 +02:00
def test_de_json_required(self, bot):
json_dict = {'phone_number': self.phone_number, 'first_name': self.first_name}
contact = Contact.de_json(json_dict, bot)
2015-09-07 18:42:51 +02:00
assert contact.phone_number == self.phone_number
assert contact.first_name == self.first_name
2015-09-07 18:42:51 +02:00
def test_de_json_all(self, bot):
json_dict = {
'phone_number': self.phone_number,
'first_name': self.first_name,
'last_name': self.last_name,
'user_id': self.user_id,
}
contact = Contact.de_json(json_dict, bot)
2015-09-07 18:42:51 +02:00
assert contact.phone_number == self.phone_number
assert contact.first_name == self.first_name
assert contact.last_name == self.last_name
assert contact.user_id == self.user_id
2015-09-07 18:42:51 +02:00
def test_send_with_contact(self, monkeypatch, bot, chat_id, contact):
def test(url, data, **kwargs):
phone = data['phone_number'] == contact.phone_number
first = data['first_name'] == contact.first_name
last = data['last_name'] == contact.last_name
return phone and first and last
monkeypatch.setattr(bot.request, 'post', test)
message = bot.send_contact(contact=contact, chat_id=chat_id)
assert message
def test_send_contact_without_required(self, bot, chat_id):
with pytest.raises(ValueError, match='Either contact or phone_number and first_name'):
bot.send_contact(chat_id=chat_id)
2015-09-07 18:42:51 +02:00
def test_to_dict(self, contact):
contact_dict = contact.to_dict()
2015-09-07 18:42:51 +02:00
assert isinstance(contact_dict, dict)
assert contact_dict['phone_number'] == contact.phone_number
assert contact_dict['first_name'] == contact.first_name
assert contact_dict['last_name'] == contact.last_name
assert contact_dict['user_id'] == contact.user_id
2015-09-07 18:42:51 +02:00
def test_equality(self):
a = Contact(self.phone_number, self.first_name)
b = Contact(self.phone_number, self.first_name)
c = Contact(self.phone_number, '')
d = Contact('', self.first_name)
API 4.5 (#1508) * Allow for nested MessageEntities in Message._parse_markdown/html, adjust tests * remove testing relict * Use MessageEntitys new equality check (#1465) * Remove unused variable * Update to custom_title feature and slow_mode_delay option Changes: - custom_title for ChatMember - new method setChatAdministratorCustomTitle for Bot - new slow_mode_delay for Chat Update due to new API future `custom_title` from API 4.5 (https://core.telegram.org/bots/api#december-31-2019) * Minor typo fix * Comply with Flake8 * Add new MessageEntities and MarkdownV2 * Added file_unique_id attrs from API 4.5 and updated tests for it * Fixed test and checked using flake8 * Fixed ChatPhoto documentation * Fix Flake8 * Add setChatAdminCstmTitle to Bot * Rename MDV2 methods * Change files id attrs to unique id * correct id_attrs for chat_photo * Revert "temporarily skip tests failing b/c missing api 4.5 (#1738)" This reverts commit 7cde6ca268754f029484e6cbded4363117df2e9b. * Fix text_markdown_v2 for monospace and text_links * closing remarks from pieter * Minor fix in escape_markdown, improve tests for it * Fix offset bug in Message._parse_* * Add test_chatphoto.py * remove debug print from test_message.py * try making codecov happy * Update readme * all hail codecov * Improve Link handling for MarkdownV1 and adjust tests. Closes #1654 * Dont use beginning of pre-entity as language in _parse_markdown * Remove debug print * Dummy commit to try fix codecov Co-authored-by: Hoi Dmytro <dmytro.hoi@gmail.com> Co-authored-by: Dmytro Hoi <code@dmytrohoi.com> Co-authored-by: poolitzer <25934244+poolitzer@users.noreply.github.com>
2020-03-28 16:37:26 +01:00
e = Voice('', 'unique_id', 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
2016-09-24 14:25:31 +02:00
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
2016-05-15 02:39:11 +02:00
assert a != e
assert hash(a) != hash(e)