mirror of
https://github.com/bohd4nx/FragmentAPI.git
synced 2026-07-25 06:14:29 +00:00
bohd4nx
8f865f4911
- Updated imports and references from TON to GRAM across multiple modules. - Renamed functions and variables to reflect the change from TON to GRAM. - Adjusted validation checks for top-up and recharge amounts to use GRAM limits. - Modified error messages and exceptions to indicate GRAM instead of TON. - Updated tests to ensure they reflect the new GRAM terminology and functionality. - Introduced new constants for GRAM-related configurations.
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
"""Cover Telegram Ads recharge flow, including request preparation and KYC handling."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
import pyfragment.domains.ads.recharge as _recharge_ads_mod
|
|
from pyfragment import AdsRechargeResult, ConfigurationError, FragmentClient
|
|
from pyfragment.core.constants.limits import GRAM_TOPUP_MAX, GRAM_TOPUP_MIN
|
|
from tests.shared import FAKE_ACCOUNT, FAKE_ADS_ACCOUNT, FAKE_REQ_ID, FAKE_TRANSACTION, FAKE_TX_HASH
|
|
|
|
# recharge_ads validation tests
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_recharge_ads_amount_zero(client: FragmentClient) -> None:
|
|
with pytest.raises(ConfigurationError):
|
|
await client.recharge_ads(FAKE_ADS_ACCOUNT, amount=GRAM_TOPUP_MIN - 1)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_recharge_ads_amount_too_high(client: FragmentClient) -> None:
|
|
with pytest.raises(ConfigurationError):
|
|
await client.recharge_ads(FAKE_ADS_ACCOUNT, amount=GRAM_TOPUP_MAX + 1)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_recharge_ads_float_amount(client: FragmentClient) -> None:
|
|
with pytest.raises(ConfigurationError):
|
|
await client.recharge_ads(FAKE_ADS_ACCOUNT, amount=5.5) # type: ignore[arg-type]
|
|
|
|
|
|
# recharge_ads mocked tests
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_recharge_ads_success(client: FragmentClient) -> None:
|
|
with (
|
|
patch.object(
|
|
client,
|
|
"call",
|
|
AsyncMock(
|
|
side_effect=[
|
|
{}, # updateAdsState
|
|
{"req_id": FAKE_REQ_ID}, # initAdsRechargeRequest
|
|
FAKE_TRANSACTION, # getAdsRechargeLink
|
|
]
|
|
),
|
|
),
|
|
patch.object(_recharge_ads_mod, "get_account_info", AsyncMock(return_value=FAKE_ACCOUNT)),
|
|
patch.object(_recharge_ads_mod, "process_transaction", AsyncMock(return_value=FAKE_TX_HASH)),
|
|
):
|
|
result = await client.recharge_ads(FAKE_ADS_ACCOUNT, amount=10)
|
|
|
|
assert isinstance(result, AdsRechargeResult)
|
|
assert result.transaction_id == FAKE_TX_HASH
|
|
assert result.amount == 10
|