mirror of
https://github.com/bohd4nx/FragmentAPI.git
synced 2026-07-25 06:14:29 +00:00
bohd4nx
ebd45d2991
- Implement tests for Stars methods: purchase_stars and giveaway_stars, including validation and success scenarios. - Add tests for Premium methods: purchase_premium and giveaway_premium, covering validation and success cases. - Create tests for topup_ton functionality, validating input and mocking successful operations. - Introduce wallet tests to verify get_wallet() functionality, ensuring correct wallet info retrieval. - Add tests for FragmentClient.call() method, validating API response handling and data merging. - Implement tests for anonymous number methods, including login codes and session management. - Create recharge_ads tests for self-service Telegram Ads recharge, validating input and success scenarios. - Add username search tests for the Fragment marketplace, ensuring correct result parsing and parameter forwarding. - Implement number search tests for the Fragment marketplace, validating result parsing and parameter handling. - Create gift search tests for the Fragment gifts marketplace, ensuring correct result parsing and parameter forwarding. - Add cookie extraction tests for get_cookies_from_browser(), validating successful extraction and error handling.
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
"""Unit tests for get_wallet() — wallet address and TON balance lookup."""
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from pyfragment import FragmentClient, WalletInfo
|
|
from tests.shared import FAKE_ADDRESS, FAKE_BALANCE_NANOTON
|
|
|
|
# Wallet mocked tests
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_wallet_returns_wallet_info(client: FragmentClient) -> None:
|
|
mock_wallet = MagicMock()
|
|
mock_wallet.refresh = AsyncMock()
|
|
mock_wallet.balance = FAKE_BALANCE_NANOTON
|
|
mock_wallet.state = MagicMock(value="active")
|
|
mock_wallet.address.to_str.return_value = FAKE_ADDRESS
|
|
|
|
with (
|
|
patch("pyfragment.utils.wallet.TonapiClient") as mock_tonapi,
|
|
patch("pyfragment.utils.wallet.WALLET_CLASSES") as mock_classes,
|
|
):
|
|
mock_tonapi.return_value.__aenter__ = AsyncMock(return_value=MagicMock())
|
|
mock_tonapi.return_value.__aexit__ = AsyncMock(return_value=False)
|
|
mock_classes["V5R1"].from_mnemonic.return_value = (mock_wallet, MagicMock(), None, None)
|
|
|
|
result = await client.get_wallet()
|
|
|
|
assert isinstance(result, WalletInfo)
|
|
assert result.address == FAKE_ADDRESS
|
|
assert result.state == "active"
|
|
assert result.balance == round(FAKE_BALANCE_NANOTON / 1_000_000_000, 4)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_wallet_balance_is_zero(client: FragmentClient) -> None:
|
|
mock_wallet = MagicMock()
|
|
mock_wallet.refresh = AsyncMock()
|
|
mock_wallet.balance = 0
|
|
mock_wallet.state = MagicMock(value="uninit")
|
|
mock_wallet.address.to_str.return_value = FAKE_ADDRESS
|
|
|
|
with (
|
|
patch("pyfragment.utils.wallet.TonapiClient") as mock_tonapi,
|
|
patch("pyfragment.utils.wallet.WALLET_CLASSES") as mock_classes,
|
|
):
|
|
mock_tonapi.return_value.__aenter__ = AsyncMock(return_value=MagicMock())
|
|
mock_tonapi.return_value.__aexit__ = AsyncMock(return_value=False)
|
|
mock_classes["V5R1"].from_mnemonic.return_value = (mock_wallet, MagicMock(), None, None)
|
|
|
|
result = await client.get_wallet()
|
|
|
|
assert result.balance == 0.0
|
|
assert result.state == "uninit"
|