mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-11-22 15:17:00 +01:00
307 lines
13 KiB
Python
307 lines
13 KiB
Python
#!/usr/bin/env python
|
|
#
|
|
# A library that provides a Python interface to the Telegram Bot API
|
|
# Copyright (C) 2015-2023
|
|
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
|
|
#
|
|
# 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
|
|
# 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.
|
|
#
|
|
# You should have received a copy of the GNU Lesser Public License
|
|
# along with this program. If not, see [http://www.gnu.org/licenses/].
|
|
|
|
import pytest
|
|
|
|
from telegram import InputInvoiceMessageContent, InputTextMessageContent, LabeledPrice
|
|
from tests.auxil.slots import mro_slots
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def input_invoice_message_content():
|
|
return InputInvoiceMessageContent(
|
|
title=TestInputInvoiceMessageContentBase.title,
|
|
description=TestInputInvoiceMessageContentBase.description,
|
|
payload=TestInputInvoiceMessageContentBase.payload,
|
|
provider_token=TestInputInvoiceMessageContentBase.provider_token,
|
|
currency=TestInputInvoiceMessageContentBase.currency,
|
|
prices=TestInputInvoiceMessageContentBase.prices,
|
|
max_tip_amount=TestInputInvoiceMessageContentBase.max_tip_amount,
|
|
suggested_tip_amounts=TestInputInvoiceMessageContentBase.suggested_tip_amounts,
|
|
provider_data=TestInputInvoiceMessageContentBase.provider_data,
|
|
photo_url=TestInputInvoiceMessageContentBase.photo_url,
|
|
photo_size=TestInputInvoiceMessageContentBase.photo_size,
|
|
photo_width=TestInputInvoiceMessageContentBase.photo_width,
|
|
photo_height=TestInputInvoiceMessageContentBase.photo_height,
|
|
need_name=TestInputInvoiceMessageContentBase.need_name,
|
|
need_phone_number=TestInputInvoiceMessageContentBase.need_phone_number,
|
|
need_email=TestInputInvoiceMessageContentBase.need_email,
|
|
need_shipping_address=TestInputInvoiceMessageContentBase.need_shipping_address,
|
|
send_phone_number_to_provider=TestInputInvoiceMessageContentBase.send_phone_number_to_provider, # noqa: E501
|
|
send_email_to_provider=TestInputInvoiceMessageContentBase.send_email_to_provider,
|
|
is_flexible=TestInputInvoiceMessageContentBase.is_flexible,
|
|
)
|
|
|
|
|
|
class TestInputInvoiceMessageContentBase:
|
|
title = "invoice title"
|
|
description = "invoice description"
|
|
payload = "invoice payload"
|
|
provider_token = "provider token"
|
|
currency = "PTBCoin"
|
|
prices = [LabeledPrice("label1", 42), LabeledPrice("label2", 314)]
|
|
max_tip_amount = 420
|
|
suggested_tip_amounts = [314, 256]
|
|
provider_data = "provider data"
|
|
photo_url = "photo_url"
|
|
photo_size = 314
|
|
photo_width = 420
|
|
photo_height = 256
|
|
need_name = True
|
|
need_phone_number = True
|
|
need_email = True
|
|
need_shipping_address = True
|
|
send_phone_number_to_provider = True
|
|
send_email_to_provider = True
|
|
is_flexible = True
|
|
|
|
|
|
class TestInputInvoiceMessageContentWithoutRequest(TestInputInvoiceMessageContentBase):
|
|
def test_slot_behaviour(self, input_invoice_message_content):
|
|
inst = input_invoice_message_content
|
|
for attr in inst.__slots__:
|
|
assert getattr(inst, attr, "err") != "err", f"got extra slot '{attr}'"
|
|
assert len(mro_slots(inst)) == len(set(mro_slots(inst))), "duplicate slot"
|
|
|
|
def test_expected_values(self, input_invoice_message_content):
|
|
assert input_invoice_message_content.title == self.title
|
|
assert input_invoice_message_content.description == self.description
|
|
assert input_invoice_message_content.payload == self.payload
|
|
assert input_invoice_message_content.provider_token == self.provider_token
|
|
assert input_invoice_message_content.currency == self.currency
|
|
assert input_invoice_message_content.prices == tuple(self.prices)
|
|
assert input_invoice_message_content.max_tip_amount == self.max_tip_amount
|
|
assert input_invoice_message_content.suggested_tip_amounts == tuple(
|
|
int(amount) for amount in self.suggested_tip_amounts
|
|
)
|
|
assert input_invoice_message_content.provider_data == self.provider_data
|
|
assert input_invoice_message_content.photo_url == self.photo_url
|
|
assert input_invoice_message_content.photo_size == int(self.photo_size)
|
|
assert input_invoice_message_content.photo_width == int(self.photo_width)
|
|
assert input_invoice_message_content.photo_height == int(self.photo_height)
|
|
assert input_invoice_message_content.need_name == self.need_name
|
|
assert input_invoice_message_content.need_phone_number == self.need_phone_number
|
|
assert input_invoice_message_content.need_email == self.need_email
|
|
assert input_invoice_message_content.need_shipping_address == self.need_shipping_address
|
|
assert (
|
|
input_invoice_message_content.send_phone_number_to_provider
|
|
== self.send_phone_number_to_provider
|
|
)
|
|
assert input_invoice_message_content.send_email_to_provider == self.send_email_to_provider
|
|
assert input_invoice_message_content.is_flexible == self.is_flexible
|
|
|
|
def test_suggested_tip_amonuts_always_tuple(self, input_invoice_message_content):
|
|
assert isinstance(input_invoice_message_content.suggested_tip_amounts, tuple)
|
|
assert input_invoice_message_content.suggested_tip_amounts == tuple(
|
|
int(amount) for amount in self.suggested_tip_amounts
|
|
)
|
|
|
|
input_invoice_message_content = InputInvoiceMessageContent(
|
|
title=self.title,
|
|
description=self.description,
|
|
payload=self.payload,
|
|
provider_token=self.provider_token,
|
|
currency=self.currency,
|
|
prices=self.prices,
|
|
)
|
|
assert input_invoice_message_content.suggested_tip_amounts == tuple()
|
|
|
|
def test_to_dict(self, input_invoice_message_content):
|
|
input_invoice_message_content_dict = input_invoice_message_content.to_dict()
|
|
|
|
assert isinstance(input_invoice_message_content_dict, dict)
|
|
assert input_invoice_message_content_dict["title"] == input_invoice_message_content.title
|
|
assert (
|
|
input_invoice_message_content_dict["description"]
|
|
== input_invoice_message_content.description
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["payload"] == input_invoice_message_content.payload
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["provider_token"]
|
|
== input_invoice_message_content.provider_token
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["currency"]
|
|
== input_invoice_message_content.currency
|
|
)
|
|
assert input_invoice_message_content_dict["prices"] == [
|
|
price.to_dict() for price in input_invoice_message_content.prices
|
|
]
|
|
assert (
|
|
input_invoice_message_content_dict["max_tip_amount"]
|
|
== input_invoice_message_content.max_tip_amount
|
|
)
|
|
assert input_invoice_message_content_dict["suggested_tip_amounts"] == list(
|
|
input_invoice_message_content.suggested_tip_amounts
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["provider_data"]
|
|
== input_invoice_message_content.provider_data
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["photo_url"]
|
|
== input_invoice_message_content.photo_url
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["photo_size"]
|
|
== input_invoice_message_content.photo_size
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["photo_width"]
|
|
== input_invoice_message_content.photo_width
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["photo_height"]
|
|
== input_invoice_message_content.photo_height
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["need_name"]
|
|
== input_invoice_message_content.need_name
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["need_phone_number"]
|
|
== input_invoice_message_content.need_phone_number
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["need_email"]
|
|
== input_invoice_message_content.need_email
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["need_shipping_address"]
|
|
== input_invoice_message_content.need_shipping_address
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["send_phone_number_to_provider"]
|
|
== input_invoice_message_content.send_phone_number_to_provider
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["send_email_to_provider"]
|
|
== input_invoice_message_content.send_email_to_provider
|
|
)
|
|
assert (
|
|
input_invoice_message_content_dict["is_flexible"]
|
|
== input_invoice_message_content.is_flexible
|
|
)
|
|
|
|
def test_de_json(self, bot):
|
|
assert InputInvoiceMessageContent.de_json({}, bot=bot) is None
|
|
|
|
json_dict = {
|
|
"title": self.title,
|
|
"description": self.description,
|
|
"payload": self.payload,
|
|
"provider_token": self.provider_token,
|
|
"currency": self.currency,
|
|
"prices": [price.to_dict() for price in self.prices],
|
|
"max_tip_amount": self.max_tip_amount,
|
|
"suggested_tip_amounts": self.suggested_tip_amounts,
|
|
"provider_data": self.provider_data,
|
|
"photo_url": self.photo_url,
|
|
"photo_size": self.photo_size,
|
|
"photo_width": self.photo_width,
|
|
"photo_height": self.photo_height,
|
|
"need_name": self.need_name,
|
|
"need_phone_number": self.need_phone_number,
|
|
"need_email": self.need_email,
|
|
"need_shipping_address": self.need_shipping_address,
|
|
"send_phone_number_to_provider": self.send_phone_number_to_provider,
|
|
"send_email_to_provider": self.send_email_to_provider,
|
|
"is_flexible": self.is_flexible,
|
|
}
|
|
|
|
input_invoice_message_content = InputInvoiceMessageContent.de_json(json_dict, bot=bot)
|
|
assert input_invoice_message_content.api_kwargs == {}
|
|
|
|
assert input_invoice_message_content.title == self.title
|
|
assert input_invoice_message_content.description == self.description
|
|
assert input_invoice_message_content.payload == self.payload
|
|
assert input_invoice_message_content.provider_token == self.provider_token
|
|
assert input_invoice_message_content.currency == self.currency
|
|
assert input_invoice_message_content.prices == tuple(self.prices)
|
|
assert input_invoice_message_content.max_tip_amount == self.max_tip_amount
|
|
assert input_invoice_message_content.suggested_tip_amounts == tuple(
|
|
int(amount) for amount in self.suggested_tip_amounts
|
|
)
|
|
assert input_invoice_message_content.provider_data == self.provider_data
|
|
assert input_invoice_message_content.photo_url == self.photo_url
|
|
assert input_invoice_message_content.photo_size == int(self.photo_size)
|
|
assert input_invoice_message_content.photo_width == int(self.photo_width)
|
|
assert input_invoice_message_content.photo_height == int(self.photo_height)
|
|
assert input_invoice_message_content.need_name == self.need_name
|
|
assert input_invoice_message_content.need_phone_number == self.need_phone_number
|
|
assert input_invoice_message_content.need_email == self.need_email
|
|
assert input_invoice_message_content.need_shipping_address == self.need_shipping_address
|
|
assert (
|
|
input_invoice_message_content.send_phone_number_to_provider
|
|
== self.send_phone_number_to_provider
|
|
)
|
|
assert input_invoice_message_content.send_email_to_provider == self.send_email_to_provider
|
|
assert input_invoice_message_content.is_flexible == self.is_flexible
|
|
|
|
def test_equality(self):
|
|
a = InputInvoiceMessageContent(
|
|
self.title,
|
|
self.description,
|
|
self.payload,
|
|
self.provider_token,
|
|
self.currency,
|
|
self.prices,
|
|
)
|
|
b = InputInvoiceMessageContent(
|
|
self.title,
|
|
self.description,
|
|
self.payload,
|
|
self.provider_token,
|
|
self.currency,
|
|
self.prices,
|
|
max_tip_amount=100,
|
|
provider_data="foobar",
|
|
)
|
|
c = InputInvoiceMessageContent(
|
|
self.title,
|
|
self.description,
|
|
self.payload,
|
|
self.provider_token,
|
|
self.currency,
|
|
# the first prices amount & the second lebal changed
|
|
[LabeledPrice("label1", 24), LabeledPrice("label22", 314)],
|
|
)
|
|
d = InputInvoiceMessageContent(
|
|
self.title,
|
|
self.description,
|
|
"different_payload",
|
|
self.provider_token,
|
|
self.currency,
|
|
self.prices,
|
|
)
|
|
e = InputTextMessageContent("text")
|
|
|
|
assert a == b
|
|
assert hash(a) == hash(b)
|
|
|
|
assert a != c
|
|
assert hash(a) != hash(c)
|
|
|
|
assert a != d
|
|
assert hash(a) != hash(d)
|
|
|
|
assert a != e
|
|
assert hash(a) != hash(e)
|