refator: use MessageFactory instead of Message, just a part

This commit is contained in:
Egor 2020-07-03 17:08:37 +05:00
parent e883366dc6
commit 0e005f661c
15 changed files with 54 additions and 143 deletions

View file

@ -1,9 +1,8 @@
import datetime
import pytest
from aiogram.api.methods import Request, SendInvoice
from aiogram.api.types import Chat, Invoice, LabeledPrice, Message
from tests.factories.message import MessageFactory
from tests.mocked_bot import MockedBot
@ -13,9 +12,7 @@ class TestSendInvoice:
prepare_result = bot.add_result_for(
SendInvoice,
ok=True,
result=Message(
message_id=42,
date=datetime.datetime.now(),
result=MessageFactory(
invoice=Invoice(
title="test",
description="test",
@ -23,7 +20,6 @@ class TestSendInvoice:
currency="BTC",
total_amount=1,
),
chat=private_chat,
),
)
@ -46,9 +42,7 @@ class TestSendInvoice:
prepare_result = bot.add_result_for(
SendInvoice,
ok=True,
result=Message(
message_id=42,
date=datetime.datetime.now(),
result=MessageFactory(
invoice=Invoice(
title="test",
description="test",
@ -56,7 +50,6 @@ class TestSendInvoice:
currency="BTC",
total_amount=1,
),
chat=private_chat,
),
)