Fringg
eb18994b7d
- Migrate 660+ datetime.utcnow() across 153 files to datetime.now(UTC) - Migrate 30+ datetime.now() without UTC to datetime.now(UTC) - Convert all 170 DateTime columns to DateTime(timezone=True) - Add migrate_datetime_to_timestamptz() in universal_migration with SET LOCAL timezone='UTC' safety - Remove 70+ .replace(tzinfo=None) workarounds - Fix utcfromtimestamp → fromtimestamp(..., tz=UTC) - Fix fromtimestamp() without tz= (system_logs, backup_service, referral_diagnostics) - Fix fromisoformat/isoparse to ensure aware output (platega, yookassa, wata, miniapp, nalogo) - Fix strptime() to add .replace(tzinfo=UTC) (backup_service, referral_diagnostics) - Fix datetime.combine() to include tzinfo=UTC (remnawave_sync, traffic_monitoring) - Fix datetime.max/datetime.min sentinels with .replace(tzinfo=UTC) - Rename panel_datetime_to_naive_utc → panel_datetime_to_utc - Remove DTZ003 from ruff ignore list
154 lines
5.9 KiB
Python
154 lines
5.9 KiB
Python
"""
|
|
Тесты для сервиса диагностики реферальной системы.
|
|
"""
|
|
|
|
import tempfile
|
|
from datetime import UTC, datetime, timedelta
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from app.services.referral_diagnostics_service import ReferralDiagnosticsService
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_log_file():
|
|
"""Создаёт временный лог-файл для тестов."""
|
|
with tempfile.NamedTemporaryFile(mode='w', suffix='.log', delete=False) as f:
|
|
yield Path(f.name)
|
|
# Cleanup
|
|
Path(f.name).unlink(missing_ok=True)
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_log_content():
|
|
"""Пример содержимого лог-файла с реферальными событиями."""
|
|
today = datetime.now(UTC).strftime('%Y-%m-%d')
|
|
return f"""
|
|
{today} 10:00:00,123 - app.handlers.start - INFO - 🔎 Найден реферальный код: <ABC123>
|
|
{today} 10:00:05,456 - app.handlers.start - INFO - ✅ Реферальный код ABC123 применен для пользователя 123456789
|
|
{today} 10:00:10,789 - app.services.referral_service - INFO - ✅ Реферальная регистрация обработана для 123456789
|
|
{today} 10:00:15,012 - app.services.referral_service - INFO - 💰 Реферал 123456789 получил бонус
|
|
|
|
{today} 11:00:00,345 - app.handlers.start - INFO - 🔎 Найден реферальный код: <XYZ999>
|
|
{today} 11:00:05,678 - app.handlers.start - INFO - ✅ Реферальный код XYZ999 применен для пользователя 987654321
|
|
|
|
{today} 12:00:00,901 - app.handlers.start - INFO - 🔎 Найден реферальный код: <TEST777>
|
|
|
|
{today} 13:00:00,234 - unrelated module - INFO - Some other log message
|
|
"""
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_parse_logs_basic(temp_log_file, sample_log_content):
|
|
"""Тест базового парсинга логов."""
|
|
# Записываем тестовые данные в файл
|
|
temp_log_file.write_text(sample_log_content)
|
|
|
|
service = ReferralDiagnosticsService(log_path=str(temp_log_file))
|
|
|
|
today = datetime.now(UTC).replace(hour=0, minute=0, second=0, microsecond=0)
|
|
tomorrow = today + timedelta(days=1)
|
|
|
|
events = await service._parse_logs(today, tomorrow)
|
|
|
|
# Проверяем что нашлись все события
|
|
assert len(events) >= 6, f'Expected at least 6 events, found {len(events)}'
|
|
|
|
# Проверяем типы событий
|
|
event_types = [e.event_type for e in events]
|
|
assert 'code_found' in event_types
|
|
assert 'code_applied' in event_types
|
|
assert 'registration_processed' in event_types
|
|
assert 'bonus_given' in event_types
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_analyze_period_with_issues(temp_log_file, sample_log_content):
|
|
"""Тест анализа с проблемными случаями."""
|
|
temp_log_file.write_text(sample_log_content)
|
|
|
|
service = ReferralDiagnosticsService(log_path=str(temp_log_file))
|
|
|
|
today = datetime.now(UTC).replace(hour=0, minute=0, second=0, microsecond=0)
|
|
tomorrow = today + timedelta(days=1)
|
|
|
|
# Используем None вместо db для базового теста парсинга
|
|
from unittest.mock import AsyncMock
|
|
|
|
mock_db = AsyncMock()
|
|
mock_db.execute.return_value.scalar_one_or_none.return_value = None
|
|
|
|
report = await service.analyze_period(mock_db, today, tomorrow)
|
|
|
|
# Проверяем статистику
|
|
# Примечание: code_found не имеет telegram_id, поэтому total_link_clicks будет 0
|
|
# Это нормально - мы считаем только события с telegram_id
|
|
assert report.total_codes_applied >= 1, 'Should have applied codes'
|
|
|
|
# Проверяем что нашлись проблемные случаи
|
|
# (987654321 применил код, но не завершил регистрацию)
|
|
assert 987654321 in report.users_applied_no_registration, (
|
|
f'Expected 987654321 in problems, got: {report.users_applied_no_registration}'
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_empty_log_file(temp_log_file):
|
|
"""Тест работы с пустым лог-файлом."""
|
|
temp_log_file.write_text('')
|
|
|
|
service = ReferralDiagnosticsService(log_path=str(temp_log_file))
|
|
|
|
today = datetime.now(UTC).replace(hour=0, minute=0, second=0, microsecond=0)
|
|
tomorrow = today + timedelta(days=1)
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
mock_db = AsyncMock()
|
|
|
|
report = await service.analyze_period(mock_db, today, tomorrow)
|
|
|
|
# Проверяем что отчёт пустой
|
|
assert report.total_link_clicks == 0
|
|
assert report.total_codes_applied == 0
|
|
assert report.total_registrations == 0
|
|
assert len(report.events) == 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_nonexistent_log_file():
|
|
"""Тест работы с несуществующим лог-файлом."""
|
|
service = ReferralDiagnosticsService(log_path='/nonexistent/path/to/log.log')
|
|
|
|
today = datetime.now(UTC).replace(hour=0, minute=0, second=0, microsecond=0)
|
|
tomorrow = today + timedelta(days=1)
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
mock_db = AsyncMock()
|
|
|
|
# Не должно быть исключений
|
|
report = await service.analyze_period(mock_db, today, tomorrow)
|
|
|
|
assert report.total_link_clicks == 0
|
|
assert len(report.events) == 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_analyze_today(temp_log_file, sample_log_content):
|
|
"""Тест метода analyze_today."""
|
|
temp_log_file.write_text(sample_log_content)
|
|
|
|
service = ReferralDiagnosticsService(log_path=str(temp_log_file))
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
mock_db = AsyncMock()
|
|
|
|
report = await service.analyze_today(mock_db)
|
|
|
|
# Проверяем что период установлен корректно
|
|
today = datetime.now(UTC).replace(hour=0, minute=0, second=0, microsecond=0)
|
|
assert report.analysis_period_start.date() == today.date()
|