mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
ec7da0f678
* Migrate from Black to Ruff and reformat code with enabling additional linter checks * Add changelog for migration to Ruff as formatter and linter * Add type ignores for specific attributes and replace tuple with set for chat type check * Remove file from another changes
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from datetime import datetime
|
|
|
|
from aiogram.enums import TransactionPartnerUserTransactionTypeEnum
|
|
from aiogram.methods import GetStarTransactions
|
|
from aiogram.types import (
|
|
StarTransaction,
|
|
StarTransactions,
|
|
TransactionPartnerUser,
|
|
User,
|
|
)
|
|
from tests.mocked_bot import MockedBot
|
|
|
|
|
|
class TestGetStarTransactions:
|
|
async def test_bot_method(self, bot: MockedBot):
|
|
user = User(id=42, is_bot=False, first_name="Test")
|
|
prepare_result = bot.add_result_for(
|
|
GetStarTransactions,
|
|
ok=True,
|
|
result=StarTransactions(
|
|
transactions=[
|
|
StarTransaction(
|
|
id="test1",
|
|
user=user,
|
|
amount=1,
|
|
date=datetime.now(),
|
|
source=TransactionPartnerUser(
|
|
user=user,
|
|
transaction_type=TransactionPartnerUserTransactionTypeEnum.GIFT_PURCHASE,
|
|
),
|
|
),
|
|
StarTransaction(
|
|
id="test2",
|
|
user=user,
|
|
amount=1,
|
|
date=datetime.now(),
|
|
receiver=TransactionPartnerUser(
|
|
user=user,
|
|
transaction_type=TransactionPartnerUserTransactionTypeEnum.GIFT_PURCHASE,
|
|
),
|
|
),
|
|
]
|
|
),
|
|
)
|
|
|
|
response: StarTransactions = await bot.get_star_transactions(limit=10, offset=0)
|
|
bot.get_request()
|
|
assert response == prepare_result.result
|