503 lines
16 KiB
Python
503 lines
16 KiB
Python
import asyncio
|
|
import importlib
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
|
|
from bot.keyboards.inline.user_keyboards import get_payment_method_keyboard
|
|
from bot.payment_providers import (
|
|
build_provider_configs,
|
|
get_provider_spec,
|
|
iter_provider_specs,
|
|
iter_service_keys,
|
|
pending_statuses,
|
|
provider_admin_only_pairs,
|
|
provider_emoji_map,
|
|
provider_label_map,
|
|
provider_telegram_button_text,
|
|
resolve_provider_presentation,
|
|
)
|
|
from bot.payment_providers.shared import (
|
|
PaymentCallbackParts,
|
|
format_number_for_payload,
|
|
payment_record_amounts,
|
|
payment_units_for_activation,
|
|
quote_hwid_callback_parts,
|
|
sale_mode_base,
|
|
sale_mode_is_hwid_devices,
|
|
sale_mode_is_traffic,
|
|
sale_mode_tariff_key,
|
|
)
|
|
from bot.payment_providers.yookassa import _resolve_yookassa_activation_amounts
|
|
from config.settings import Settings
|
|
|
|
_LEGACY_PROVIDER_FILES = [
|
|
"backend/bot/services/yookassa_service.py",
|
|
"backend/bot/services/freekassa_service.py",
|
|
"backend/bot/services/platega_service.py",
|
|
"backend/bot/services/severpay_service.py",
|
|
"backend/bot/services/crypto_pay_service.py",
|
|
"backend/bot/services/stars_service.py",
|
|
"backend/bot/services/wata_service.py",
|
|
"backend/bot/handlers/user/payment.py",
|
|
"backend/bot/handlers/user/subscription/payment_methods.py",
|
|
"backend/bot/handlers/user/subscription/payments_yookassa.py",
|
|
"backend/bot/handlers/user/subscription/payments_freekassa.py",
|
|
"backend/bot/handlers/user/subscription/payments_platega.py",
|
|
"backend/bot/handlers/user/subscription/payments_severpay.py",
|
|
"backend/bot/handlers/user/subscription/payments_crypto.py",
|
|
"backend/bot/handlers/user/subscription/payments_stars.py",
|
|
"backend/bot/handlers/user/subscription/payments_wata.py",
|
|
]
|
|
|
|
_PROVIDER_MODULES = {
|
|
"yookassa": "YooKassaService",
|
|
"freekassa": "FreeKassaService",
|
|
"platega": "PlategaService",
|
|
"severpay": "SeverPayService",
|
|
"cryptopay": "CryptoPayService",
|
|
"stars": "StarsService",
|
|
"wata": "WataService",
|
|
"heleket": "HeleketService",
|
|
}
|
|
|
|
|
|
def test_legacy_provider_integration_files_are_removed():
|
|
repo_root = Path(__file__).resolve().parents[1]
|
|
|
|
for relative_path in _LEGACY_PROVIDER_FILES:
|
|
assert not (repo_root / relative_path).exists()
|
|
|
|
|
|
def test_every_provider_module_owns_its_service_and_spec():
|
|
for module_name, service_class_name in _PROVIDER_MODULES.items():
|
|
module = importlib.import_module(f"bot.payment_providers.{module_name}")
|
|
|
|
assert hasattr(module, service_class_name)
|
|
assert hasattr(module, "SPEC") or hasattr(module, "SPECS")
|
|
|
|
|
|
def test_wata_is_registered_as_single_provider_module():
|
|
spec = get_provider_spec("wata")
|
|
|
|
assert spec is not None
|
|
assert spec.service_key == "wata_service"
|
|
assert spec.pending_status == "pending_wata"
|
|
assert spec.callback_prefix == "pay_wata"
|
|
assert spec.router is not None
|
|
assert spec.create_service is not None
|
|
assert spec.webhook_route is not None
|
|
assert spec.create_webapp_payment is not None
|
|
|
|
|
|
def test_yookassa_provider_keeps_autorenew_entrypoints_local():
|
|
yookassa = importlib.import_module("bot.payment_providers.yookassa")
|
|
spec = get_provider_spec("yookassa")
|
|
|
|
assert spec is not None
|
|
assert spec.service_key == "yookassa_service"
|
|
assert spec.create_service is not None
|
|
assert spec.webhook_route is yookassa.yookassa_webhook_route
|
|
assert yookassa.payment_processing_lock is not None
|
|
assert callable(yookassa.process_successful_payment)
|
|
assert callable(yookassa.process_cancelled_payment)
|
|
assert callable(yookassa.yookassa_webhook_route)
|
|
|
|
|
|
def test_every_payment_method_has_registry_driven_webapp_creator():
|
|
missing = [spec.id for spec in iter_provider_specs() if spec.create_webapp_payment is None]
|
|
|
|
assert missing == []
|
|
|
|
|
|
def test_service_keys_and_statuses_come_from_provider_specs():
|
|
assert set(iter_service_keys()) == {
|
|
"yookassa_service",
|
|
"freekassa_service",
|
|
"platega_service",
|
|
"severpay_service",
|
|
"wata_service",
|
|
"stars_service",
|
|
"cryptopay_service",
|
|
"heleket_service",
|
|
}
|
|
assert set(pending_statuses()) >= {
|
|
"pending",
|
|
"pending_yookassa",
|
|
"pending_freekassa",
|
|
"pending_platega",
|
|
"pending_severpay",
|
|
"pending_wata",
|
|
"pending_cryptopay",
|
|
"pending_stars",
|
|
"pending_heleket",
|
|
}
|
|
|
|
|
|
def test_provider_labels_and_emojis_include_storage_keys_and_method_aliases():
|
|
labels = provider_label_map()
|
|
emojis = provider_emoji_map()
|
|
|
|
assert labels["wata"] == "Wata"
|
|
assert labels["telegram_stars"] == "Telegram Stars"
|
|
assert labels["stars"] == "Telegram Stars"
|
|
assert labels["platega"] == "Platega"
|
|
assert labels["platega_sbp"] == "Platega"
|
|
assert labels["platega_crypto"] == "Platega"
|
|
assert emojis["stars"] == get_provider_spec("stars").default_telegram_emoji
|
|
assert emojis["telegram_stars"] == get_provider_spec("stars").default_telegram_emoji
|
|
assert emojis["cryptopay"] == get_provider_spec("cryptopay").default_telegram_emoji
|
|
|
|
|
|
def test_provider_presentation_resolves_defaults_and_overrides():
|
|
spec = get_provider_spec("yookassa")
|
|
assert spec is not None
|
|
|
|
default = resolve_provider_presentation(spec)
|
|
assert default.webapp_label == spec.webapp_label
|
|
assert default.webapp_icon == "CreditCard"
|
|
assert default.telegram_label == spec.telegram_labels["ru"]
|
|
assert default.telegram_emoji == spec.default_telegram_emoji
|
|
assert not default.telegram_customized
|
|
|
|
settings = SimpleNamespace(
|
|
PAYMENT_YOOKASSA_WEBAPP_LABEL_EN="Card in app",
|
|
PAYMENT_YOOKASSA_WEBAPP_ICON="WalletCards",
|
|
PAYMENT_YOOKASSA_TELEGRAM_LABEL_EN="Card in bot",
|
|
PAYMENT_YOOKASSA_TELEGRAM_EMOJI="💸",
|
|
)
|
|
custom = resolve_provider_presentation(spec, settings, language="en")
|
|
assert custom.webapp_label == "Card in app"
|
|
assert custom.webapp_icon == "WalletCards"
|
|
assert custom.telegram_label == "Card in bot"
|
|
assert custom.telegram_emoji == "💸"
|
|
assert custom.telegram_customized
|
|
|
|
|
|
def test_provider_telegram_button_text_uses_provider_defaults_until_customized():
|
|
spec = get_provider_spec("wata")
|
|
assert spec is not None
|
|
|
|
assert (
|
|
provider_telegram_button_text(spec, SimpleNamespace(), language="en")
|
|
== f"{spec.default_telegram_emoji} Wata"
|
|
)
|
|
|
|
settings = SimpleNamespace(PAYMENT_WATA_TELEGRAM_LABEL_EN="Pay Wata")
|
|
assert (
|
|
provider_telegram_button_text(spec, settings, language="en")
|
|
== f"{spec.default_telegram_emoji} Pay Wata"
|
|
)
|
|
|
|
|
|
def test_provider_presentation_ignores_cross_language_override():
|
|
spec = get_provider_spec("yookassa")
|
|
assert spec is not None
|
|
|
|
settings = SimpleNamespace(PAYMENT_YOOKASSA_WEBAPP_LABEL_RU="Карта")
|
|
|
|
assert resolve_provider_presentation(spec, settings, language="en").webapp_label == "YooKassa"
|
|
|
|
|
|
def test_payment_method_keyboard_uses_custom_telegram_text_without_changing_callback(monkeypatch):
|
|
monkeypatch.setenv("WATA_ENABLED", "True")
|
|
monkeypatch.setenv("PAYMENT_WATA_TELEGRAM_LABEL_EN", "Wata custom")
|
|
monkeypatch.setenv("PAYMENT_WATA_TELEGRAM_EMOJI", "💸")
|
|
build_provider_configs(force=True)
|
|
|
|
settings = Settings(
|
|
_env_file=None,
|
|
BOT_TOKEN="token",
|
|
POSTGRES_USER="app_user",
|
|
POSTGRES_PASSWORD="app_password",
|
|
TARIFFS_CONFIG_PATH="missing-tariffs.json",
|
|
PAYMENT_METHODS_ORDER="wata",
|
|
)
|
|
i18n = SimpleNamespace(gettext=lambda _lang, key, **_kwargs: key)
|
|
|
|
markup = get_payment_method_keyboard(
|
|
months=1,
|
|
price=150,
|
|
stars_price=None,
|
|
currency_symbol_val="RUB",
|
|
lang="en",
|
|
i18n_instance=i18n,
|
|
settings=settings,
|
|
)
|
|
|
|
button = markup.inline_keyboard[0][0]
|
|
assert button.text == "💸 Wata custom"
|
|
assert button.callback_data == "pay_wata:1:150:subscription"
|
|
|
|
|
|
def test_payment_method_keyboard_filters_providers_by_payment_currency(monkeypatch):
|
|
monkeypatch.setenv("YOOKASSA_ENABLED", "True")
|
|
monkeypatch.setenv("WATA_ENABLED", "True")
|
|
build_provider_configs(force=True)
|
|
|
|
settings = Settings(
|
|
_env_file=None,
|
|
BOT_TOKEN="token",
|
|
POSTGRES_USER="app_user",
|
|
POSTGRES_PASSWORD="app_password",
|
|
TARIFFS_CONFIG_PATH="missing-tariffs.json",
|
|
PAYMENT_METHODS_ORDER="yookassa,wata",
|
|
)
|
|
i18n = SimpleNamespace(gettext=lambda _lang, key, **_kwargs: key)
|
|
|
|
markup = get_payment_method_keyboard(
|
|
months=1,
|
|
price=10,
|
|
stars_price=None,
|
|
currency_symbol_val="USD",
|
|
lang="en",
|
|
i18n_instance=i18n,
|
|
settings=settings,
|
|
)
|
|
|
|
callbacks = [
|
|
button.callback_data for row in markup.inline_keyboard for button in row if button.callback_data
|
|
]
|
|
assert "pay_wata:1:10:subscription" in callbacks
|
|
assert all(not callback.startswith("pay_yk:") for callback in callbacks)
|
|
|
|
|
|
def test_admin_only_provider_is_visible_only_to_admins(monkeypatch):
|
|
from bot.app.web.webapp.serializers import _serialize_payment_methods
|
|
|
|
monkeypatch.setenv("WATA_ENABLED", "False")
|
|
monkeypatch.setenv("WATA_ADMIN_ONLY_ENABLED", "True")
|
|
build_provider_configs(force=True)
|
|
|
|
settings = Settings(
|
|
_env_file=None,
|
|
BOT_TOKEN="token",
|
|
POSTGRES_USER="app_user",
|
|
POSTGRES_PASSWORD="app_password",
|
|
TARIFFS_CONFIG_PATH="missing-tariffs.json",
|
|
PAYMENT_METHODS_ORDER="wata",
|
|
ADMIN_IDS="42",
|
|
STARS_ENABLED=False,
|
|
)
|
|
i18n = SimpleNamespace(gettext=lambda _lang, key, **_kwargs: key)
|
|
app = {"wata_service": SimpleNamespace(configured=True)}
|
|
spec = get_provider_spec("wata")
|
|
|
|
regular_markup = get_payment_method_keyboard(
|
|
months=1,
|
|
price=150,
|
|
stars_price=None,
|
|
currency_symbol_val="RUB",
|
|
lang="en",
|
|
i18n_instance=i18n,
|
|
settings=settings,
|
|
user_id=7,
|
|
)
|
|
admin_markup = get_payment_method_keyboard(
|
|
months=1,
|
|
price=150,
|
|
stars_price=None,
|
|
currency_symbol_val="RUB",
|
|
lang="en",
|
|
i18n_instance=i18n,
|
|
settings=settings,
|
|
user_id=42,
|
|
)
|
|
|
|
assert spec is not None
|
|
assert not spec.is_visible(settings, app)
|
|
assert not spec.is_visible_for_user(settings, app, is_admin=False)
|
|
assert spec.is_visible_for_user(settings, app, is_admin=True)
|
|
assert all(
|
|
not button.callback_data.startswith("pay_wata:")
|
|
for row in regular_markup.inline_keyboard
|
|
for button in row
|
|
)
|
|
assert admin_markup.inline_keyboard[0][0].callback_data == "pay_wata:1:150:subscription"
|
|
assert _serialize_payment_methods(settings, app, "en", is_admin=False) == []
|
|
assert _serialize_payment_methods(settings, app, "en", is_admin=True)[0]["id"] == "wata"
|
|
|
|
|
|
def test_webapp_payment_methods_filter_by_default_currency(monkeypatch):
|
|
from bot.app.web.webapp.serializers import _serialize_payment_methods
|
|
|
|
monkeypatch.setenv("YOOKASSA_ENABLED", "True")
|
|
monkeypatch.setenv("WATA_ENABLED", "True")
|
|
build_provider_configs(force=True)
|
|
|
|
settings = Settings(
|
|
_env_file=None,
|
|
BOT_TOKEN="token",
|
|
POSTGRES_USER="app_user",
|
|
POSTGRES_PASSWORD="app_password",
|
|
TARIFFS_CONFIG_PATH="missing-tariffs.json",
|
|
PAYMENT_METHODS_ORDER="yookassa,wata",
|
|
DEFAULT_CURRENCY_SYMBOL="USD",
|
|
STARS_ENABLED=False,
|
|
)
|
|
app = {
|
|
"yookassa_service": SimpleNamespace(configured=True),
|
|
"wata_service": SimpleNamespace(configured=True),
|
|
}
|
|
|
|
methods = _serialize_payment_methods(settings, app, "en", is_admin=False)
|
|
|
|
assert [method["id"] for method in methods] == ["wata"]
|
|
|
|
|
|
|
|
def test_admin_only_provider_toggle_pairs_are_declared():
|
|
pairs = set(provider_admin_only_pairs())
|
|
|
|
assert ("WATA_ENABLED", "WATA_ADMIN_ONLY_ENABLED") in pairs
|
|
assert ("PLATEGA_SBP_ENABLED", "PLATEGA_SBP_ADMIN_ONLY_ENABLED") in pairs
|
|
assert ("PLATEGA_CRYPTO_ENABLED", "PLATEGA_CRYPTO_ADMIN_ONLY_ENABLED") in pairs
|
|
assert ("STARS_ENABLED", "STARS_ADMIN_ONLY_ENABLED") in pairs
|
|
|
|
|
|
def test_admin_only_provider_toggle_normalization_disables_public_pair():
|
|
from bot.services.settings_override_service import _normalize_exclusive_provider_toggles
|
|
|
|
updates, deletes = _normalize_exclusive_provider_toggles(
|
|
{"WATA_ADMIN_ONLY_ENABLED": True},
|
|
["WATA_ENABLED"],
|
|
)
|
|
|
|
assert updates["WATA_ADMIN_ONLY_ENABLED"] is True
|
|
assert updates["WATA_ENABLED"] is False
|
|
assert deletes == []
|
|
|
|
|
|
def test_provider_callbacks_are_built_from_specs():
|
|
wata = get_provider_spec("wata")
|
|
stars = get_provider_spec("stars")
|
|
|
|
assert wata is not None
|
|
assert stars is not None
|
|
assert (
|
|
wata.callback_data(
|
|
value="1",
|
|
rub_price=150,
|
|
stars_price=None,
|
|
sale_mode="subscription",
|
|
)
|
|
== "pay_wata:1:150:subscription"
|
|
)
|
|
assert (
|
|
stars.callback_data(
|
|
value="1",
|
|
rub_price=150,
|
|
stars_price=42,
|
|
sale_mode="subscription",
|
|
)
|
|
== "pay_stars:1:42:subscription"
|
|
)
|
|
assert (
|
|
stars.callback_data(
|
|
value="1",
|
|
rub_price=150,
|
|
stars_price=None,
|
|
sale_mode="subscription",
|
|
)
|
|
is None
|
|
)
|
|
|
|
|
|
def test_provider_visibility_uses_service_configuration(monkeypatch):
|
|
monkeypatch.setenv("WATA_ENABLED", "True")
|
|
monkeypatch.delenv("WATA_ADMIN_ONLY_ENABLED", raising=False)
|
|
build_provider_configs(force=True)
|
|
|
|
spec = get_provider_spec("wata")
|
|
assert spec is not None
|
|
|
|
settings = SimpleNamespace(WATA_ENABLED=True)
|
|
assert spec.is_visible(settings, {"wata_service": SimpleNamespace(configured=True)})
|
|
assert not spec.is_visible(settings, {"wata_service": SimpleNamespace(configured=False)})
|
|
assert not spec.is_visible(SimpleNamespace(WATA_ENABLED=False), {})
|
|
|
|
|
|
def test_common_sale_mode_helpers_cover_provider_payment_records():
|
|
assert sale_mode_base("traffic_package@premium|anything") == "traffic_package"
|
|
assert sale_mode_is_traffic("premium_topup@vip")
|
|
assert sale_mode_is_hwid_devices("hwid_devices@vip")
|
|
assert sale_mode_tariff_key("subscription@vip") == "vip"
|
|
assert sale_mode_tariff_key("subscription@vip|bot") == "vip"
|
|
assert format_number_for_payload(10.0) == "10"
|
|
assert format_number_for_payload(10.5) == "10.5"
|
|
|
|
traffic = payment_record_amounts(months=20, traffic_gb=20.5, sale_mode="topup@vip")
|
|
assert traffic.months == 20
|
|
assert traffic.purchased_gb == 20.5
|
|
assert traffic.purchased_hwid_devices is None
|
|
assert traffic.tariff_key == "vip"
|
|
assert traffic.traffic_sale
|
|
|
|
hwid = payment_record_amounts(months=3, sale_mode="hwid_devices@vip")
|
|
assert hwid.months == 3
|
|
assert hwid.purchased_gb is None
|
|
assert hwid.purchased_hwid_devices == 3
|
|
assert hwid.tariff_key == "vip"
|
|
assert hwid.hwid_devices_sale
|
|
|
|
payment = SimpleNamespace(
|
|
purchased_gb=None,
|
|
purchased_hwid_devices=3,
|
|
subscription_duration_months=None,
|
|
)
|
|
assert payment_units_for_activation(payment, "hwid_devices@vip") == 3
|
|
|
|
|
|
def test_yookassa_hwid_webapp_metadata_uses_device_count_for_activation():
|
|
(
|
|
subscription_months,
|
|
traffic_amount_gb,
|
|
hwid_devices_count,
|
|
months_for_activation,
|
|
traffic_gb_for_activation,
|
|
) = _resolve_yookassa_activation_amounts(
|
|
sale_mode_base="hwid_devices",
|
|
subscription_months_raw="0",
|
|
traffic_gb_raw=None,
|
|
hwid_devices_raw="3",
|
|
)
|
|
|
|
assert subscription_months == 0
|
|
assert traffic_amount_gb == 0
|
|
assert hwid_devices_count == 3
|
|
assert months_for_activation == 3
|
|
assert traffic_gb_for_activation is None
|
|
|
|
|
|
def test_hwid_callback_quote_rejects_fractional_device_count():
|
|
subscription_service = SimpleNamespace(quote_hwid_device_topup=AsyncMock())
|
|
|
|
quoted_parts, quote = asyncio.run(
|
|
quote_hwid_callback_parts(
|
|
session=AsyncMock(),
|
|
user_id=42,
|
|
parts=PaymentCallbackParts(
|
|
months=1.9,
|
|
price=50,
|
|
sale_mode="hwid_devices@vip",
|
|
),
|
|
subscription_service=subscription_service,
|
|
)
|
|
)
|
|
|
|
assert quoted_parts is None
|
|
assert quote is None
|
|
subscription_service.quote_hwid_device_topup.assert_not_awaited()
|
|
|
|
|
|
def test_yookassa_hwid_metadata_rejects_fractional_device_count():
|
|
with pytest.raises(ValueError):
|
|
_resolve_yookassa_activation_amounts(
|
|
sale_mode_base="hwid_devices",
|
|
subscription_months_raw="0",
|
|
traffic_gb_raw=None,
|
|
hwid_devices_raw="1.9",
|
|
)
|