gy9vin
143 lines
5.2 KiB
Python
143 lines
5.2 KiB
Python
"""Тесты для Telegram Stars-сценариев внутри PaymentService."""
|
|
|
|
from pathlib import Path
|
|
from typing import Any, Dict, Optional
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
ROOT_DIR = Path(__file__).resolve().parents[2]
|
|
if str(ROOT_DIR) not in sys.path:
|
|
sys.path.insert(0, str(ROOT_DIR))
|
|
|
|
from app.services.payment_service import PaymentService # noqa: E402
|
|
from app.config import settings # noqa: E402
|
|
|
|
|
|
@pytest.fixture
|
|
def anyio_backend() -> str:
|
|
"""Ограничиваем anyio тесты только бэкендом asyncio."""
|
|
return "asyncio"
|
|
|
|
|
|
class DummyBot:
|
|
"""Минимальная заглушка aiogram.Bot для тестов."""
|
|
|
|
def __init__(self) -> None:
|
|
self.calls: list[Dict[str, Any]] = []
|
|
|
|
async def create_invoice_link(self, **kwargs: Any) -> str:
|
|
"""Эмулируем создание платежной ссылки и сохраняем параметры вызова."""
|
|
self.calls.append(kwargs)
|
|
return "https://t.me/invoice/stars"
|
|
|
|
|
|
def _make_service(bot: Optional[DummyBot]) -> PaymentService:
|
|
"""Создаёт экземпляр PaymentService без выполнения полного конструктора."""
|
|
service = PaymentService.__new__(PaymentService) # type: ignore[call-arg]
|
|
service.bot = bot
|
|
# Stars-сервис достаточно обозначить любым truthy-значением.
|
|
service.stars_service = object() if bot else None
|
|
return service
|
|
|
|
|
|
@pytest.mark.anyio("asyncio")
|
|
async def test_create_stars_invoice_calculates_stars(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""Количество звёзд должно рассчитываться по курсу с округлением вниз и нижним порогом 1."""
|
|
bot = DummyBot()
|
|
service = _make_service(bot)
|
|
|
|
monkeypatch.setattr(
|
|
type(settings),
|
|
"get_stars_rate",
|
|
lambda self: 70,
|
|
raising=False,
|
|
)
|
|
monkeypatch.setattr(
|
|
type(settings),
|
|
"format_price",
|
|
lambda self, amount: f"{amount / 100:.0f}₽",
|
|
raising=False,
|
|
)
|
|
|
|
result = await service.create_stars_invoice(
|
|
amount_kopeks=14000,
|
|
description="Пополнение",
|
|
payload="custom_payload",
|
|
)
|
|
|
|
assert result == "https://t.me/invoice/stars"
|
|
assert len(bot.calls) == 1
|
|
call = bot.calls[0]
|
|
assert call["title"] == "Пополнение баланса VPN"
|
|
assert call["payload"] == "custom_payload"
|
|
prices = call["prices"]
|
|
assert len(prices) == 1
|
|
assert prices[0].amount == 2 # 14000 коп. → 140 ₽ → 2 звезды при курсе 70
|
|
assert "≈2 ⭐" in call["description"]
|
|
|
|
|
|
@pytest.mark.anyio("asyncio")
|
|
async def test_create_stars_invoice_enforces_minimum_star(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""При слишком маленькой сумме минимум должен составлять 1 звезду."""
|
|
bot = DummyBot()
|
|
service = _make_service(bot)
|
|
|
|
monkeypatch.setattr(type(settings), "get_stars_rate", lambda self: 500, raising=False)
|
|
monkeypatch.setattr(type(settings), "format_price", lambda self, amount: amount, raising=False)
|
|
|
|
await service.create_stars_invoice(
|
|
amount_kopeks=50, # 0.5 ₽ при курсе 500 => <1 звезды
|
|
description="Микроплатёж",
|
|
)
|
|
|
|
prices = bot.calls[0]["prices"]
|
|
assert prices[0].amount == 1
|
|
|
|
|
|
@pytest.mark.anyio("asyncio")
|
|
async def test_create_stars_invoice_uses_explicit_stars(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""Если передано значение stars_amount, функция должна использовать его напрямую."""
|
|
bot = DummyBot()
|
|
service = _make_service(bot)
|
|
|
|
# При явном указании звёзд курс не запрашивается.
|
|
monkeypatch.setattr(type(settings), "format_price", lambda self, amount: amount, raising=False)
|
|
|
|
await service.create_stars_invoice(
|
|
amount_kopeks=1000,
|
|
description="Оплата подписки",
|
|
stars_amount=5,
|
|
)
|
|
|
|
prices = bot.calls[0]["prices"]
|
|
assert prices[0].amount == 5
|
|
assert "≈5 ⭐" in bot.calls[0]["description"]
|
|
|
|
|
|
@pytest.mark.anyio("asyncio")
|
|
async def test_create_stars_invoice_rejects_invalid_rate(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""Отрицательный или нулевой курс должен приводить к исключению."""
|
|
bot = DummyBot()
|
|
service = _make_service(bot)
|
|
|
|
monkeypatch.setattr(type(settings), "get_stars_rate", lambda self: 0, raising=False)
|
|
|
|
with pytest.raises(ValueError, match="Stars rate must be positive"):
|
|
await service.create_stars_invoice(
|
|
amount_kopeks=1000,
|
|
description="Пополнение",
|
|
)
|
|
|
|
|
|
@pytest.mark.anyio("asyncio")
|
|
async def test_create_stars_invoice_requires_bot() -> None:
|
|
"""Без экземпляра бота и stars_service функция должна отказывать."""
|
|
service = _make_service(bot=None)
|
|
|
|
with pytest.raises(ValueError, match="Bot instance required"):
|
|
await service.create_stars_invoice(
|
|
amount_kopeks=1000,
|
|
description="Пополнение",
|
|
)
|