66d188c1cd
- utils/morning_runner.py: Scheduler + run_morning() - bot.py: автоматический запуск планировщика при старте - commands/morning.py: использовать run_morning() вместо дублирования - .env.example: добавить MORNING_TIME=07:00 - AGENTS.md: обновить документацию - tests/test_morning_runner.py: 10 тестов для Scheduler
135 lines
5.2 KiB
Python
135 lines
5.2 KiB
Python
"""Тесты для utils/morning_runner.py — Scheduler и run_morning."""
|
||
|
||
import asyncio
|
||
from datetime import datetime
|
||
from unittest.mock import AsyncMock, MagicMock, patch
|
||
|
||
import discord
|
||
import pytest
|
||
|
||
from utils.morning_runner import Scheduler, run_morning
|
||
|
||
|
||
class TestSchedulerInit:
|
||
"""Тесты инициализации Scheduler."""
|
||
|
||
def test_init_sets_morning_time(self):
|
||
"""Инициализация должна устанавливать время."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot, "08:30")
|
||
assert scheduler.morning_time == "08:30"
|
||
|
||
def test_init_default_morning_time(self):
|
||
"""Инициализация с дефолтным временем."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot)
|
||
assert scheduler.morning_time == "07:00"
|
||
|
||
def test_init_creates_loop(self):
|
||
"""Инициализация должна создавать loop."""
|
||
bot = AsyncMock()
|
||
scheduler = Scheduler(bot)
|
||
assert scheduler.morning_loop is not None
|
||
|
||
|
||
class TestSchedulerCalculateNextRun:
|
||
"""Тесты расчёта следующего запуска."""
|
||
|
||
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_tomorrow_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:
|
||
"""Тесты запуска/остановки планировщика."""
|
||
|
||
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
|
||
|
||
with patch("utils.morning_runner.asyncio.gather", new=AsyncMock(return_value=(
|
||
{"current_condition": [{"temp_C": "20", "FeelsLikeC": "22", "weatherDesc": [{"value": "Clear"}], "humidity": "50", "windspeedKmph": "10", "pressure": "1013"}]},
|
||
[{"title": "Test", "link": "http://test.com", "pub_date": "Mon, 01 Jan 2026 00:00:00 GMT", "creator": "", "tags": []}],
|
||
[{"title": "Test", "link": "http://test.com", "pub_date": "Mon, 01 Jan 2026 00:00:00 GMT", "creator": "", "tags": []}],
|
||
"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
|