diff --git a/tests/test_api/test_types/test_shipping_query.py b/tests/test_api/test_types/test_shipping_query.py index 0d1eda8a..822e6b83 100644 --- a/tests/test_api/test_types/test_shipping_query.py +++ b/tests/test_api/test_types/test_shipping_query.py @@ -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