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.
89 lines
3.2 KiB
Python
89 lines
3.2 KiB
Python
"""Unit tests for search_usernames — Fragment marketplace username search."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from pyfragment import FragmentClient
|
|
from pyfragment.types import UsernamesResult
|
|
|
|
FAKE_HTML = """
|
|
<tr class="tm-row-selectable">
|
|
<td><a href="/username/coolname" class="table-cell">
|
|
<div class="table-cell-value tm-value">@coolname</div>
|
|
<div class="table-cell-status-thin thin-only tm-status-avail">On auction</div>
|
|
</a></td>
|
|
<td><div class="table-cell-value tm-value icon-before icon-ton">5</div>
|
|
<time datetime="2026-06-01T12:00:00+00:00" data-relative="text">2 days</time>
|
|
</td>
|
|
<td><div class="table-cell-value tm-value tm-status-avail">On auction</div></td>
|
|
</tr>
|
|
"""
|
|
|
|
|
|
# search_usernames result parsing tests
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_usernames_basic(client: FragmentClient) -> None:
|
|
with patch.object(client, "call", AsyncMock(return_value={"ok": True, "html": FAKE_HTML})):
|
|
result = await client.search_usernames("coolname")
|
|
|
|
assert isinstance(result, UsernamesResult)
|
|
assert len(result.items) == 1
|
|
assert result.items[0]["slug"] == "username/coolname"
|
|
assert result.items[0]["name"] == "@coolname"
|
|
assert result.items[0]["date"] == "2026-06-01T12:00:00+00:00"
|
|
assert result.next_offset_id is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_usernames_empty_html(client: FragmentClient) -> None:
|
|
with patch.object(client, "call", AsyncMock(return_value={"ok": True})):
|
|
result = await client.search_usernames("zzz_no_results")
|
|
|
|
assert isinstance(result, UsernamesResult)
|
|
assert result.items == []
|
|
assert result.next_offset_id is None
|
|
|
|
|
|
# search_usernames parameter forwarding tests
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_usernames_with_sort_and_filter(client: FragmentClient) -> None:
|
|
mock_call = AsyncMock(return_value={"ok": True, "html": FAKE_HTML})
|
|
with patch.object(client, "call", mock_call):
|
|
result = await client.search_usernames("durov", sort="price_desc", filter="auction")
|
|
|
|
assert isinstance(result, UsernamesResult)
|
|
call_data = mock_call.call_args[0][1]
|
|
assert call_data["type"] == "usernames"
|
|
assert call_data["sort"] == "price_desc"
|
|
assert call_data["filter"] == "auction"
|
|
assert call_data["query"] == "durov"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_usernames_with_offset_id(client: FragmentClient) -> None:
|
|
mock_call = AsyncMock(return_value={"ok": True, "html": FAKE_HTML, "next_offset_id": "offset_99"})
|
|
with patch.object(client, "call", mock_call):
|
|
result = await client.search_usernames("durov", offset_id="offset_10")
|
|
|
|
assert isinstance(result, UsernamesResult)
|
|
assert result.next_offset_id == "offset_99"
|
|
call_data = mock_call.call_args[0][1]
|
|
assert call_data["offset_id"] == "offset_10"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_usernames_default_query(client: FragmentClient) -> None:
|
|
mock_call = AsyncMock(return_value={"ok": True, "html": FAKE_HTML})
|
|
with patch.object(client, "call", mock_call):
|
|
result = await client.search_usernames()
|
|
|
|
assert isinstance(result, UsernamesResult)
|
|
call_data = mock_call.call_args[0][1]
|
|
assert call_data["query"] == ""
|
|
assert call_data["type"] == "usernames"
|