mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-01-20 11:50:27 +00:00
427011fe41
2) У промогрупп появится приоритет 3) У пользователя может быть несколько промогрупп, но влиять будет только с наивысшим приоритетом 4) К промокодам можно будет добавить промогруппу. Все активировавшие промокод получат её 5) При выводе пользователей с промогруппой будет также выводиться ссылка на каждого. Можно будет отследить сливы промокодов "для своих". Я в целом это добавлю во все места, где пользователь выводится в админке 6) Исправить баг исчезновения триалки при пополнении 7) Исправить падающие тесты и добавить новых 8) Трафик: 0 ГБ в тестовой подписке исправить на Трафик: Безлимит
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
from pathlib import Path
|
|
import sys
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock
|
|
|
|
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 import referral_service # noqa: E402
|
|
|
|
|
|
async def test_commission_accrues_before_minimum_first_topup(monkeypatch):
|
|
user = SimpleNamespace(
|
|
id=1,
|
|
telegram_id=101,
|
|
full_name="Test User",
|
|
referred_by_id=2,
|
|
has_made_first_topup=False,
|
|
)
|
|
referrer = SimpleNamespace(
|
|
id=2,
|
|
telegram_id=202,
|
|
full_name="Referrer",
|
|
)
|
|
|
|
db = SimpleNamespace(
|
|
commit=AsyncMock(),
|
|
execute=AsyncMock(),
|
|
)
|
|
|
|
get_user_mock = AsyncMock(side_effect=[user, referrer])
|
|
monkeypatch.setattr(referral_service, "get_user_by_id", get_user_mock)
|
|
add_user_balance_mock = AsyncMock()
|
|
monkeypatch.setattr(referral_service, "add_user_balance", add_user_balance_mock)
|
|
create_referral_earning_mock = AsyncMock()
|
|
monkeypatch.setattr(referral_service, "create_referral_earning", create_referral_earning_mock)
|
|
|
|
monkeypatch.setattr(referral_service.settings, "REFERRAL_MINIMUM_TOPUP_KOPEKS", 20000)
|
|
monkeypatch.setattr(referral_service.settings, "REFERRAL_FIRST_TOPUP_BONUS_KOPEKS", 5000)
|
|
monkeypatch.setattr(referral_service.settings, "REFERRAL_INVITER_BONUS_KOPEKS", 10000)
|
|
monkeypatch.setattr(referral_service.settings, "REFERRAL_COMMISSION_PERCENT", 25)
|
|
|
|
topup_amount = 15000
|
|
|
|
result = await referral_service.process_referral_topup(db, user.id, topup_amount)
|
|
|
|
assert result is True
|
|
assert user.has_made_first_topup is False
|
|
|
|
add_user_balance_mock.assert_awaited_once()
|
|
add_call = add_user_balance_mock.await_args
|
|
assert add_call.args[1] is referrer
|
|
assert add_call.args[2] == 3750
|
|
assert "Комиссия" in add_call.args[3]
|
|
assert add_call.kwargs.get("bot") is None
|
|
|
|
create_referral_earning_mock.assert_awaited_once()
|
|
earning_call = create_referral_earning_mock.await_args
|
|
assert earning_call.kwargs["amount_kopeks"] == 3750
|
|
assert earning_call.kwargs["reason"] == "referral_commission_topup"
|
|
|
|
db.commit.assert_not_awaited()
|
|
db.execute.assert_not_awaited()
|