94 lines
4.4 KiB
Python
94 lines
4.4 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
Тест функциональности с чистыми данными
|
||
"""
|
||
import asyncio
|
||
import sys
|
||
import os
|
||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from src.core.database import async_session_maker, engine
|
||
from src.core.services import UserService, LotteryService
|
||
from src.utils.account_utils import validate_account_number, format_account_number, mask_account_number
|
||
from src.display.winner_display import format_winner_display
|
||
import random
|
||
|
||
async def test_features():
|
||
print("🧪 Тестирование с чистыми данными")
|
||
print("="*50)
|
||
|
||
async with async_session_maker() as session:
|
||
# Генерируем уникальные тестовые данные
|
||
unique_id = random.randint(1000000, 9999999)
|
||
test_account = f"{random.randint(10, 99)}-{random.randint(10, 99)}-{random.randint(10, 99)}-{random.randint(10, 99)}-{random.randint(10, 99)}-{random.randint(10, 99)}-{random.randint(10, 99)}-{random.randint(10, 99)}"
|
||
|
||
print(f"🆔 Используем уникальный ID: {unique_id}")
|
||
print(f"🔢 Используем номер счёта: {test_account}")
|
||
|
||
# 1. Создание пользователя
|
||
print(f"\n1. 👤 Создание пользователя:")
|
||
user = await UserService.get_or_create_user(
|
||
session,
|
||
telegram_id=unique_id,
|
||
username=f'test_user_{unique_id}',
|
||
first_name='Тестовый',
|
||
last_name='Пользователь'
|
||
)
|
||
print(f"✅ Пользователь создан: {user.first_name} {user.last_name}")
|
||
|
||
# 2. Установка номера счёта
|
||
print(f"\n2. 🔢 Установка номера счёта:")
|
||
success = await UserService.set_account_number(
|
||
session, user.telegram_id, test_account
|
||
)
|
||
|
||
if success:
|
||
print(f"✅ Номер счёта {test_account} установлен")
|
||
# Обновляем данные пользователя
|
||
await session.refresh(user)
|
||
print(f"✅ Подтверждено: {user.account_number}")
|
||
else:
|
||
print(f"❌ Не удалось установить номер счёта")
|
||
return
|
||
|
||
# 3. Поиск пользователя по номеру
|
||
print(f"\n3. 🔍 Поиск по номеру счёта:")
|
||
found_user = await UserService.get_user_by_account(session, test_account)
|
||
if found_user:
|
||
print(f"✅ Пользователь найден: {found_user.first_name} {found_user.last_name}")
|
||
else:
|
||
print("❌ Пользователь не найден")
|
||
|
||
# 4. Тестирование маскирования
|
||
print(f"\n4. 🎭 Тестирование маскирования:")
|
||
masked = mask_account_number(test_account)
|
||
print(f"Полный номер: {test_account}")
|
||
print(f"Маскированный: {masked}")
|
||
|
||
# 5. Создание розыгрыша с типами отображения
|
||
print(f"\n5. 🎨 Тестирование отображения победителей:")
|
||
lottery = await LotteryService.create_lottery(
|
||
session,
|
||
title="Тест отображения",
|
||
description="Тестовый розыгрыш",
|
||
prizes=["Приз 1"],
|
||
creator_id=1
|
||
)
|
||
|
||
# Тестируем все типы отображения
|
||
for display_type in ["username", "chat_id", "account_number"]:
|
||
await LotteryService.set_winner_display_type(session, lottery.id, display_type)
|
||
lottery.winner_display_type = display_type # Обновляем локально
|
||
|
||
public_display = format_winner_display(user, lottery, show_sensitive_data=False)
|
||
admin_display = format_winner_display(user, lottery, show_sensitive_data=True)
|
||
|
||
print(f"\n📺 Тип: {display_type}")
|
||
print(f" 👥 Публично: {public_display}")
|
||
print(f" 🔧 Админ: {admin_display}")
|
||
|
||
print(f"\n🎉 Все тесты завершены успешно!")
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(test_features())
|