71bcd66794
Задачи Sprint 1 (Critical Fixes): - [1.1] Fallback в пустом embed (!morning) — добавлена проверка has_real_data и fallback сообщение при отключении внешних API - [1.2] Обработка ошибок токена/сети в bot.run() — raise_exception=True + логирование LoginFailure, HTTPException, Exception - [1.3] Рефакторинг парсинга погоды — вынесено в format_weather_data_for_console(), убран дублирующий код из 2 файлов Изменения: • utils/morning_runner.py — добавлена проверка has_real_data после формирования description_lines • bot.py — обработчики исключений для запуска бота с детализированным логированием и пользовательскими сообщениями • utils/pogoda.py — новая функция format_weather_data_for_console() для центрального форматирования погодных данных • console_commands/pogoda.py — замена 15 строк дублирующейся логики на вызов единой функции (24→8 строк) • console_commands/morning.py — аналогичные изменения для команды morning (19→13 строк с погодой) • tests/test_morning_runner.py — +2 теста для fallback сценариев empty embed и only weather data • tests/test_bot.py — новый файл с 2 тестами на проверку кода обработки ошибок • tests/test_pogoda.py — +6 тестов для format_weather_data_for_console() Статистика тестирования: • Общее количество тестов: 200 (было 190) • Новые тесты: 10 • Все тесты проходят успешно Примечания: - Сообщения без восклицательных знаков согласно preferencem
221 lines
9.3 KiB
Python
221 lines
9.3 KiB
Python
"""Тесты для utils/morning_runner.py — Scheduler и run_morning."""
|
||
|
||
import asyncio
|
||
from datetime import datetime
|
||
from unittest.mock import AsyncMock, MagicMock, patch, PropertyMock
|
||
|
||
import discord
|
||
import pytest
|
||
|
||
from utils.morning_runner import Scheduler, run_morning
|
||
|
||
|
||
class TestSchedulerInit:
|
||
"""Тесты инициализации Scheduler."""
|
||
|
||
def test_init_sets_morning_time(self):
|
||
"""Инициализация должна устанавливать время."""
|
||
bot = AsyncMock()
|
||
mock_loop = MagicMock()
|
||
with patch("utils.morning_runner.tasks.loop", return_value=mock_loop):
|
||
scheduler = Scheduler(bot, "08:30")
|
||
assert scheduler.morning_time == "08:30"
|
||
|
||
def test_init_default_morning_time(self):
|
||
"""Инициализация с дефолтным временем."""
|
||
bot = AsyncMock()
|
||
mock_loop = MagicMock()
|
||
with patch("utils.morning_runner.tasks.loop", return_value=mock_loop):
|
||
scheduler = Scheduler(bot)
|
||
assert scheduler.morning_time == "07:00"
|
||
|
||
def test_init_creates_loop(self):
|
||
"""Инициализация должна создавать loop."""
|
||
bot = AsyncMock()
|
||
mock_loop = MagicMock()
|
||
with patch("utils.morning_runner.tasks.loop", return_value=mock_loop):
|
||
scheduler = Scheduler(bot)
|
||
assert scheduler.morning_loop is not None
|
||
|
||
|
||
class TestSchedulerCalculateNextRun:
|
||
"""Тесты расчёта следующего запуска."""
|
||
|
||
@pytest.fixture(autouse=True)
|
||
def _mock_loop(self):
|
||
"""Замокать tasks.loop, чтобы не создавать реальный coroutine."""
|
||
with patch("utils.morning_runner.tasks.loop", return_value=MagicMock()):
|
||
yield
|
||
|
||
def test_next_run_today_before_time(self):
|
||
"""Если сейчас раньше времени — вернуть сегодня."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot, "14:00")
|
||
|
||
with patch("utils.morning_runner.datetime") as mock_dt:
|
||
mock_dt.now.return_value = datetime(2026, 5, 29, 10, 0, 0)
|
||
next_run = scheduler._calculate_next_run()
|
||
assert next_run == datetime(2026, 5, 29, 14, 0, 0)
|
||
|
||
def test_next_run_tomorning_after_time(self):
|
||
"""Если сейчас позже времени — вернуть завтра."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot, "14:00")
|
||
|
||
with patch("utils.morning_runner.datetime") as mock_dt:
|
||
mock_dt.now.return_value = datetime(2026, 5, 29, 15, 0, 0)
|
||
next_run = scheduler._calculate_next_run()
|
||
assert next_run == datetime(2026, 5, 30, 14, 0, 0)
|
||
|
||
def test_next_run_exact_time(self):
|
||
"""Если сейчас ровно время — вернуть завтра."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot, "14:00")
|
||
|
||
with patch("utils.morning_runner.datetime") as mock_dt:
|
||
mock_dt.now.return_value = datetime(2026, 5, 29, 14, 0, 0)
|
||
next_run = scheduler._calculate_next_run()
|
||
assert next_run == datetime(2026, 5, 30, 14, 0, 0)
|
||
|
||
|
||
class TestSchedulerStartStop:
|
||
"""Тесты запуска/остановки планировщика."""
|
||
|
||
@pytest.fixture(autouse=True)
|
||
def _mock_loop(self):
|
||
"""Замокать tasks.loop, чтобы не создавать реальный coroutine."""
|
||
with patch("utils.morning_runner.tasks.loop", return_value=MagicMock()):
|
||
yield
|
||
|
||
def test_start_starts_loop(self):
|
||
"""start() должен вызывать start() на loop."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot)
|
||
loop_mock = MagicMock()
|
||
scheduler.morning_loop = loop_mock
|
||
|
||
scheduler.start()
|
||
loop_mock.start.assert_called_once()
|
||
|
||
def test_stop_stops_loop(self):
|
||
"""stop() должен вызывать stop() на loop."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot)
|
||
loop_mock = MagicMock()
|
||
scheduler.morning_loop = loop_mock
|
||
|
||
scheduler.stop()
|
||
loop_mock.stop.assert_called_once()
|
||
|
||
|
||
class TestSchedulerCheckAndRun:
|
||
"""Тесты проверки и запуска morning."""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_check_and_run_same_day_no_duplicate(self):
|
||
"""Не должен запускать дважды в один день."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot, "07:00")
|
||
scheduler._last_run_date = datetime.now().day
|
||
|
||
with patch("utils.morning_runner.datetime") as mock_dt:
|
||
mock_dt.now.return_value = datetime(2026, 5, 29, 7, 0, 0)
|
||
await scheduler._check_and_run_morning()
|
||
|
||
# run_morning не должен вызываться
|
||
assert scheduler._last_run_date == datetime.now().day
|
||
|
||
|
||
class TestRunMorning:
|
||
"""Тесты run_morning."""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_run_morning_sends_embed(self):
|
||
"""run_morning должен отправлять embed в канал."""
|
||
bot = AsyncMock()
|
||
channel = AsyncMock()
|
||
channel.name = "test-channel"
|
||
channel.guild.me = MagicMock()
|
||
channel.permissions_for.return_value.send_messages = True
|
||
|
||
weather_data = {
|
||
"current_condition": [
|
||
{"temp_C": "20", "FeelsLikeC": "22", "weatherDesc": [{"value": "Clear"}], "humidity": "50", "windspeedKmph": "10", "pressure": "1013"}
|
||
]
|
||
}
|
||
articles = [{"title": "Test", "link": "http://test.com", "pub_date": "Mon, 01 Jan 2026 00:00:00 GMT", "creator": "", "tags": []}]
|
||
posts = [{"title": "Test", "link": "http://test.com", "pub_date": "Mon, 01 Jan 2026 00:00:00 GMT", "creator": "", "tags": []}]
|
||
|
||
with patch("utils.morning_runner.fetch_weather", new=AsyncMock(return_value=weather_data)), \
|
||
patch("utils.morning_runner.fetch_rss", new=AsyncMock(side_effect=[articles, posts])), \
|
||
patch("utils.morning_runner.fetch_cat", new=AsyncMock(return_value="http://cat.jpg")), \
|
||
patch("utils.morning_runner.discord.Embed") as mock_embed:
|
||
await run_morning(bot, channel)
|
||
|
||
channel.send.assert_called_once()
|
||
call_args = channel.send.call_args[1]
|
||
assert "embed" in call_args
|
||
assert call_args["embed"] is not None
|
||
|
||
|
||
class TestRunMorningWithFallback:
|
||
"""Тесты fallback в пустом embed."""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_run_morning_empty_embed_fallback(self):
|
||
"""run_morning должен добавлять fallback сообщение при пустых данных."""
|
||
bot = AsyncMock()
|
||
channel = AsyncMock()
|
||
channel.name = "test-channel"
|
||
channel.guild.me = MagicMock()
|
||
channel.permissions_for.return_value.send_messages = True
|
||
|
||
# Все API возвращают None/пусто
|
||
with patch("utils.morning_runner.fetch_weather", new=AsyncMock(return_value=None)), \
|
||
patch("utils.morning_runner.fetch_rss", new=AsyncMock(return_value=None)), \
|
||
patch("utils.morning_runner.fetch_cat", new=AsyncMock(return_value=None)), \
|
||
patch("utils.morning_runner.discord.Embed") as mock_embed_class:
|
||
|
||
embed_mock = AsyncMock()
|
||
mock_embed_class.return_value = embed_mock
|
||
|
||
await run_morning(bot, channel)
|
||
|
||
# Убедимся, что send был вызван
|
||
channel.send.assert_called_once()
|
||
call_args = channel.send.call_args[1]
|
||
assert "embed" in call_args
|
||
|
||
# Проверяем, что description содержит fallback сообщение
|
||
embed_description = call_args["embed"].description
|
||
assert "Не удалось получить данные из внешних источников" in embed_description
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_run_morning_only_weather_data(self):
|
||
"""run_morning должен корректно обрабатывать только погоду без новостей."""
|
||
bot = AsyncMock()
|
||
channel = AsyncMock()
|
||
channel.name = "test-channel"
|
||
channel.guild.me = MagicMock()
|
||
channel.permissions_for.return_value.send_messages = True
|
||
|
||
weather_data = {
|
||
"current_condition": [
|
||
{"temp_C": "20", "FeelsLikeC": "22", "weatherDesc": [{"value": "Clear"}], "humidity": "50", "windspeedKmph": "10", "pressure": "1013"}
|
||
]
|
||
}
|
||
|
||
with patch("utils.morning_runner.fetch_weather", new=AsyncMock(return_value=weather_data)), \
|
||
patch("utils.morning_runner.fetch_rss", new=AsyncMock(side_effect=[None, None])), \
|
||
patch("utils.morning_runner.fetch_cat", new=AsyncMock(return_value=None)):
|
||
await run_morning(bot, channel)
|
||
|
||
channel.send.assert_called_once()
|
||
call_args = channel.send.call_args[1]
|
||
assert "embed" in call_args
|
||
|
||
# Проверяем, что в embed есть только погода и нет fallback сообщения
|
||
embed_description = call_args["embed"].description
|
||
assert "Погода в Магнитогорске" in embed_description
|
||
assert "Не удалось получить данные из внешних источников" not in embed_description
|