add
This commit is contained in:
@@ -1,19 +1,11 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import datetime, timedelta
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import select
|
||||
|
||||
from app.config import Settings
|
||||
from app.db.base import utcnow
|
||||
from app.db.models import (
|
||||
PromoCode,
|
||||
PromoCodeApplication,
|
||||
ReferralCode,
|
||||
ReferralInvite,
|
||||
RemnawaveUser,
|
||||
TelegramUser,
|
||||
)
|
||||
from app.db.models import ReferralCode, ReferralInvite, RemnawaveUser, TelegramUser
|
||||
from app.schemas.remnawave import PaginatedUsers
|
||||
from app.services.sync_service import SyncService
|
||||
from tests.helpers import make_remote_user
|
||||
@@ -221,56 +213,6 @@ async def test_get_admin_telegram_user_profile_returns_referral_and_access_data(
|
||||
assert profile.accesses[0].record.username == "oreol-profile"
|
||||
|
||||
|
||||
async def test_promo_code_can_be_applied_through_referral_input(session_factory) -> None:
|
||||
service = SyncService(
|
||||
settings=Settings.model_construct(
|
||||
bot_admin_ids_raw="",
|
||||
sync_subscription_history=False,
|
||||
),
|
||||
session_factory=session_factory,
|
||||
remnawave_client=_StubRemnawaveClient([]),
|
||||
)
|
||||
|
||||
promo = await service.create_promo_code(
|
||||
code="white50",
|
||||
plan_code="white",
|
||||
discount_percent=50,
|
||||
expires_at=utcnow() + timedelta(days=7),
|
||||
created_by_telegram_id=1001,
|
||||
)
|
||||
applied_code = await service.apply_referral_code(
|
||||
telegram_id=2001,
|
||||
username="buyer",
|
||||
first_name="Buyer",
|
||||
last_name=None,
|
||||
language_code="ru",
|
||||
referral_code="white50",
|
||||
)
|
||||
summary = await service.get_referral_summary(2001)
|
||||
active_promo = await service.get_active_promo_for_user(2001)
|
||||
|
||||
assert promo.code == "WHITE50"
|
||||
assert applied_code == "WHITE50"
|
||||
assert summary.applied_referral_code == ""
|
||||
assert summary.applied_promo_code == "WHITE50"
|
||||
assert summary.applied_promo_discount_percent == 50
|
||||
assert summary.applied_promo_plan_code == "white"
|
||||
assert active_promo is not None
|
||||
assert active_promo.code == "WHITE50"
|
||||
|
||||
async with session_factory() as session:
|
||||
promo_record = await session.scalar(
|
||||
select(PromoCode).where(PromoCode.code == "WHITE50")
|
||||
)
|
||||
application = await session.scalar(
|
||||
select(PromoCodeApplication).where(PromoCodeApplication.telegram_id == 2001)
|
||||
)
|
||||
|
||||
assert promo_record is not None
|
||||
assert application is not None
|
||||
assert application.promo_code_id == promo_record.id
|
||||
|
||||
|
||||
async def test_sync_all_users_does_not_autoflush_half_built_user(
|
||||
session_factory,
|
||||
) -> None:
|
||||
|
||||
Reference in New Issue
Block a user