Improving test_shipping_query

This commit is contained in:
Gabben 2020-05-17 16:58:15 +05:00
parent 83730276bc
commit c5e0742337

View file

@ -1,5 +1,5 @@
from aiogram.api.methods import AnswerShippingQuery
from aiogram.api.types import ShippingAddress, ShippingQuery, User
from aiogram.api.types import ShippingAddress, ShippingQuery, User, ShippingOption, LabeledPrice
class TestInlineQuery:
@ -18,7 +18,14 @@ class TestInlineQuery:
),
)
api_method = shipping_query.answer(True)
shipping_options = [
ShippingOption(id="id", title="foo", prices=[LabeledPrice(label="foo", amount=123)])]
kwargs = dict(ok=True, shipping_options=shipping_options, error_message="foo")
api_method = shipping_query.answer(**kwargs)
assert isinstance(api_method, AnswerShippingQuery)
assert api_method.shipping_query_id == shipping_query.id
for key, value in kwargs.items():
assert getattr(api_method, key) == value