Compare commits
No commits in common. "dda5753f8c04a193affbb94450107f907f5e4503" and "5a11fb2b5d4ac1039578644297b4e9b99892e98a" have entirely different histories.
dda5753f8c
...
5a11fb2b5d
@ -1,7 +0,0 @@
|
||||
repos:
|
||||
- repo: https://github.com/astral-sh/ruff-pre-commit
|
||||
rev: v0.8.6
|
||||
hooks:
|
||||
- id: ruff
|
||||
args: [--fix]
|
||||
- id: ruff-format
|
||||
32
ISSUES.md
32
ISSUES.md
@ -1,28 +1,18 @@
|
||||
# ISSUES — Задачи и баг-трекер
|
||||
|
||||
## Исправлено
|
||||
|
||||
- [x] ~~Эмодзи в embed-сообщениях~~ — удалены `🌅`, `✅`, `❌` из `utils/morning_runner.py`
|
||||
- [x] ~~f-string в logger~~ — замена на `%`-формат в `bot.py`
|
||||
- [x] ~~`import time` внутри `__init__`~~ — перенесён на уровень модуля в `bot.py`
|
||||
- [x] ~~Проверка наличия `.env`~~ — `.env` в `.gitignore`, есть `.env.example`
|
||||
- [x] ~~Команда `!msg` в BotRunner~~ — удалена из проекта
|
||||
- [x] ~~Отсутствует `.gitignore`~~ — файл существует
|
||||
- [x] ~~Отсутствует Dockerfile~~ — файл существует
|
||||
- [x] ~~Отсутствует `.env.example`~~ — файл существует
|
||||
|
||||
## Средний приоритет
|
||||
|
||||
- [x] **Type hints в production-коде** — добавлены аннотации ко всем 20 функциям
|
||||
- [x] **`asyncio.iscoroutinefunction` deprecated** — monkey-patch в `bot.py` + `conftest.py`
|
||||
- [x] **Утечка корутины Scheduler в тестах** — mock `_start_scheduler` вместо `asyncio.create_task`
|
||||
- [x] **Graceful shutdown (SIGTERM)** — уже реализован через `on_shutdown` listener + `async with self.bot`
|
||||
- [x] **Пустой `__init__` в Morning** — удалён
|
||||
- [x] **Global RateLimiter на модульном уровне** — добавлены factory-функции `make_*_limiter()`
|
||||
- [ ] **Проверка наличия `.env`** — есть проверка DISCORD_TOKEN в `_validate_config()`, но нет проверки существования самого файла `.env` (без файла бот падает с непонятной ошибкой)
|
||||
- [ ] **Graceful shutdown (SIGTERM/SIGINT)** — нет обработчиков сигналов, бот завершается резко без закрытия handler-ов и cleanup
|
||||
- [ ] **`asyncio.iscoroutinefunction` deprecated** — 35 warnings, функция удалена в Python 3.16; заменить на `inspect.iscoroutinefunction` (решение: обновление discord.py или monkey-patch в `bot.py`)
|
||||
- [ ] **Утечка корутины в тестах** — `RuntimeWarning: coroutine 'trigger_morning.<locals>._run' was never awaited` в `test_trigger_morning.py`
|
||||
|
||||
## Низкий приоритет
|
||||
|
||||
- [x] **Type hints в тестах** — добавлены `-> None` ко всем 140 test-функциям
|
||||
- [x] **`import` внутри функций в тестах** — вынесены наверх модулей (35 вхождений)
|
||||
- [x] **Добавить pre-commit хуки** — `.pre-commit-config.yaml` (ruff + ruff-format), `requirements-dev.txt`
|
||||
- [x] **Интеграционные тесты** — 9 тестов загрузок когов, команд и утилит
|
||||
- [ ] **Добавить pre-commit хуки** — lint (flake8/ruff), форматирование (black)
|
||||
- [ ] **Добавить type hints** повсеместно — не все функции имеют аннотации (отсутствуют в `utils/pogoda.py`, `utils/news.py`, присутствуют в `utils/cat.py`, `utils/morning_runner.py`)
|
||||
- [ ] **Интеграционные тесты** — сейчас только unit-тесты с моками
|
||||
|
||||
## Замечания
|
||||
|
||||
- [ ] В тестовых файлах используется `asyncio.run()` внутри синхронных тестов (50 вхождений: `test_fetch_cat.py` — 10, `test_fetch_rss.py` — 21, `test_fetch_weather.py` — 19) — конфликт с `pytest-asyncio` event loop, рекомендуется `async def` тесты
|
||||
|
||||
@ -39,13 +39,14 @@ python bot.py
|
||||
| `!hp` | Список всех команд бота с описанием (автогенерация из `bot.commands`) |
|
||||
| `!morning` | Погода + топ-5 статей + топ-5 новостей + котик (утренний дайджест) |
|
||||
| `!cat` | Случайный котик |
|
||||
| `!msg <текст>` | Повторить текст в чате |
|
||||
| `!status` | Статус бота: пинг к Discord gateway, uptime |
|
||||
| `!stats` | Количество серверов, каналов, пользователей |
|
||||
|
||||
## Архитектура
|
||||
|
||||
```
|
||||
bot.py # Точка входа, инициализация бота
|
||||
bot.py # Точка входа, инициализация бота, команда !msg (в BotRunner._setup_events)
|
||||
commands/ # Discord команды (cogs)
|
||||
__init__.py # ALL_COMMANDS — явные импорты
|
||||
pg.py # !pg — погода (обёртка над utils.pogoda)
|
||||
|
||||
25
bot.py
25
bot.py
@ -1,16 +1,10 @@
|
||||
import asyncio
|
||||
import inspect
|
||||
import logging
|
||||
import os
|
||||
import sys
|
||||
import threading
|
||||
import time
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
# Python 3.14+: asyncio.iscoroutinefunction deprecated, removed in 3.16
|
||||
# discord.py 2.7.1 ещё не обновлена — применяем monkey-patch до импорта
|
||||
asyncio.iscoroutinefunction = inspect.iscoroutinefunction # type: ignore[assignment]
|
||||
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
from discord.ext.commands import CommandNotFound
|
||||
@ -34,6 +28,8 @@ class BotRunner:
|
||||
"""Управляет жизненным циклом бота."""
|
||||
|
||||
def __init__(self) -> None:
|
||||
import time
|
||||
|
||||
self.bot = commands.Bot(command_prefix="!", intents=intents)
|
||||
self.bot._start_time = time.time()
|
||||
self.stop_event = threading.Event()
|
||||
@ -84,7 +80,7 @@ class BotRunner:
|
||||
# Терминал — детали для разработчика
|
||||
cmd_name = ctx.command.name if ctx and ctx.command else "?"
|
||||
logger.error(
|
||||
"Ошибка команды %s: %s", cmd_name, error,
|
||||
f"Ошибка команды {cmd_name}: {error}",
|
||||
exc_info=True,
|
||||
)
|
||||
|
||||
@ -104,6 +100,11 @@ class BotRunner:
|
||||
except (discord.NotFound, discord.Forbidden):
|
||||
pass # Бот не может писать в канал — игнорируем
|
||||
|
||||
@self.bot.command(name="msg")
|
||||
async def msg(ctx: commands.Context, *, text: str) -> None:
|
||||
"""Повторяет текст после !msg"""
|
||||
await ctx.send(text)
|
||||
|
||||
def _on_shutdown(self) -> None:
|
||||
"""Остановить планировщик при завершении работы бота."""
|
||||
if self.scheduler:
|
||||
@ -115,19 +116,13 @@ class BotRunner:
|
||||
self._on_shutdown()
|
||||
|
||||
def run(self, token: str) -> None:
|
||||
"""Запустить бота с graceful shutdown.
|
||||
|
||||
Graceful shutdown обеспечивается:
|
||||
- discord.py on_shutdown событие для остановки планировщика
|
||||
- async with self.bot (context manager) для graceful disconnect
|
||||
- KeyboardInterrupt (Ctrl+C) на POSIX
|
||||
- SIGTERM обрабатывается через entrypoint-скрипт (Docker/K8s)
|
||||
"""
|
||||
"""Запустить бота с graceful shutdown."""
|
||||
logger.info("Запуск бота...")
|
||||
|
||||
async def main() -> None:
|
||||
try:
|
||||
async with self.bot:
|
||||
# Регистрируем on_shutdown внутри контекста бота
|
||||
self.bot.add_listener(self._on_shutdown_async, "on_shutdown")
|
||||
await self.bot.start(token, reconnect=True)
|
||||
except discord.LoginFailure as e:
|
||||
|
||||
@ -11,7 +11,7 @@ class Cat(commands.Cog):
|
||||
"""Команда !cat — случайный котик"""
|
||||
|
||||
@commands.command(name="cat")
|
||||
async def cat(self, ctx: commands.Context) -> None:
|
||||
async def cat(self, ctx):
|
||||
"""Получить случайного котика"""
|
||||
url = await fetch_cat()
|
||||
if url is None:
|
||||
|
||||
@ -6,7 +6,7 @@ class Help(commands.Cog):
|
||||
"""Команда !hp — список всех команд бота"""
|
||||
|
||||
@commands.command(name="hp")
|
||||
async def hp(self, ctx: commands.Context) -> None:
|
||||
async def hp(self, ctx):
|
||||
"""Показать список доступных команд"""
|
||||
await self._show_help(ctx)
|
||||
|
||||
|
||||
@ -10,8 +10,11 @@ logger = logging.getLogger(__name__)
|
||||
class Morning(commands.Cog):
|
||||
"""Команда !morning — погода и новости утром"""
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
@commands.command(name="morning")
|
||||
async def morning(self, ctx: commands.Context) -> None:
|
||||
async def morning(self, ctx):
|
||||
"""Погода, лучшие статьи за сутки и котик"""
|
||||
logger.info("%s: !morning запущен", ctx.author)
|
||||
await run_morning(ctx.bot, ctx.channel)
|
||||
|
||||
@ -16,7 +16,7 @@ class News(commands.Cog):
|
||||
"""Команда !news — свежие статьи и новости по AI с Habr"""
|
||||
|
||||
@commands.command(name="nw")
|
||||
async def nw(self, ctx: commands.Context) -> None:
|
||||
async def nw(self, ctx):
|
||||
"""Топ-5 свежих статей и новостей по AI с Habr"""
|
||||
articles = await fetch_rss(RSS_URL_ARTICLES)
|
||||
if articles is None:
|
||||
|
||||
@ -12,7 +12,7 @@ class Pg(commands.Cog):
|
||||
self.api_url = API_URL_WEATHER
|
||||
|
||||
@commands.command(name="pg")
|
||||
async def pg(self, ctx: commands.Context) -> None:
|
||||
async def pg(self, ctx):
|
||||
"""Прогноз погоды в Магнитогорске"""
|
||||
data = await fetch_weather(self.api_url)
|
||||
if data is None:
|
||||
|
||||
@ -9,7 +9,7 @@ class Stats(commands.Cog):
|
||||
"""Команда !stats — статистика серверов"""
|
||||
|
||||
@commands.command(name="stats")
|
||||
async def stats(self, ctx: commands.Context) -> None:
|
||||
async def stats(self, ctx):
|
||||
"""Количество серверов, каналов и пользователей"""
|
||||
guilds = ctx.bot.guilds
|
||||
total_guilds = len(guilds)
|
||||
|
||||
@ -11,7 +11,7 @@ class Status(commands.Cog):
|
||||
"""Команда !status — статус бота, пинг, uptime"""
|
||||
|
||||
@commands.command(name="status")
|
||||
async def status(self, ctx: commands.Context) -> None:
|
||||
async def status(self, ctx):
|
||||
"""Статус бота: пинг к Discord gateway и время работы"""
|
||||
latency_ms = round(ctx.bot.latency * 1000, 1)
|
||||
start_time = getattr(ctx.bot, "_start_time", time.time())
|
||||
|
||||
@ -1,8 +0,0 @@
|
||||
"""Pytest configuration — применяется до импорта тестов."""
|
||||
|
||||
import asyncio
|
||||
import inspect
|
||||
|
||||
# Python 3.14+: asyncio.iscoroutinefunction deprecated, removed in 3.16
|
||||
# discord.py 2.7.1 ещё не обновлена — применяем monkey-patch до импорта
|
||||
asyncio.iscoroutinefunction = inspect.iscoroutinefunction # type: ignore[assignment]
|
||||
@ -1,4 +1,2 @@
|
||||
pre-commit>=3.5.0
|
||||
pytest>=7.4.0
|
||||
pytest-asyncio>=0.21.0
|
||||
ruff>=0.8.0
|
||||
|
||||
@ -11,8 +11,6 @@ import sys
|
||||
from pathlib import Path
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
import discord
|
||||
|
||||
# Добавляем корень проекта в путь импорта
|
||||
ROOT_DIR = Path(__file__).resolve().parent.parent
|
||||
sys.path.insert(0, str(ROOT_DIR))
|
||||
@ -21,7 +19,7 @@ sys.path.insert(0, str(ROOT_DIR))
|
||||
class TestBotInit:
|
||||
"""Тесты для инициализации бота."""
|
||||
|
||||
def test_bot_created_with_default_prefix(self) -> None:
|
||||
def test_bot_created_with_default_prefix(self):
|
||||
"""Проверка, что бот создан с правильным префиксом команд."""
|
||||
import bot
|
||||
|
||||
@ -37,9 +35,10 @@ class TestBotInit:
|
||||
class TestBotErrorHandling:
|
||||
"""Тесты для проверки обработки ошибок запуска бота."""
|
||||
|
||||
def test_bot_handles_login_failure(self) -> None:
|
||||
def test_bot_handles_login_failure(self):
|
||||
"""BotRunner.run() обрабатывает discord.LoginFailure."""
|
||||
import bot
|
||||
import discord
|
||||
|
||||
runner = bot.BotRunner()
|
||||
with patch.object(runner.bot, "start", side_effect=discord.LoginFailure("bad token")):
|
||||
@ -49,9 +48,10 @@ class TestBotErrorHandling:
|
||||
runner.run("fake_token")
|
||||
mock_exit.assert_called_once_with(1)
|
||||
|
||||
def test_bot_handles_http_exception(self) -> None:
|
||||
def test_bot_handles_http_exception(self):
|
||||
"""BotRunner.run() обрабатывает discord.HTTPException."""
|
||||
import bot
|
||||
import discord
|
||||
|
||||
runner = bot.BotRunner()
|
||||
mock_response = MagicMock(status=502)
|
||||
@ -62,7 +62,7 @@ class TestBotErrorHandling:
|
||||
runner.run("fake_token")
|
||||
mock_exit.assert_called_once_with(1)
|
||||
|
||||
def test_shutdown_uses_on_shutdown_listener(self) -> None:
|
||||
def test_shutdown_uses_on_shutdown_listener(self):
|
||||
"""BotRunner.run() регистрирует on_shutdown вместо signal handlers.
|
||||
|
||||
Signal handlers с asyncio.new_event_loop() создают race condition
|
||||
@ -81,7 +81,7 @@ class TestBotErrorHandling:
|
||||
content = f.read()
|
||||
assert "signal.signal" not in content, "Не должно быть signal.signal — используется on_shutdown"
|
||||
|
||||
def test_code_uses_async_bot_pattern(self) -> None:
|
||||
def test_code_uses_async_bot_pattern(self):
|
||||
"""Проверка, что bot.py использует async with / asyncio.run."""
|
||||
with open(ROOT_DIR / "bot.py", encoding="utf-8") as f:
|
||||
content = f.read()
|
||||
|
||||
@ -7,7 +7,7 @@ from commands.pg import Pg
|
||||
class TestPgInit:
|
||||
"""Тесты инициализации Cog Pg."""
|
||||
|
||||
def test_init_sets_api_url(self) -> None:
|
||||
def test_init_sets_api_url(self):
|
||||
"""__init__ должен устанавливать api_url."""
|
||||
cog = Pg()
|
||||
assert cog.api_url == "https://wttr.in/Magnitogorsk?format=j1&lang=ru"
|
||||
@ -42,7 +42,7 @@ class TestPgCommand:
|
||||
return defaults
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_success(self) -> None:
|
||||
async def test_pg_success(self):
|
||||
"""Успешный запрос погоды должен отправить embed с данными."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -61,7 +61,7 @@ class TestPgCommand:
|
||||
assert "Давление: 759.8 мм рт. ст." in args
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_fetch_returns_none(self) -> None:
|
||||
async def test_pg_fetch_returns_none(self):
|
||||
"""fetch_weather вернул None — бот должен сообщить об ошибке."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -72,7 +72,7 @@ class TestPgCommand:
|
||||
ctx.send.assert_called_once_with("Не удалось получить данные о погоде.")
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_empty_current_condition(self) -> None:
|
||||
async def test_pg_empty_current_condition(self):
|
||||
"""current_condition пустой список — graceful fallback."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -84,7 +84,7 @@ class TestPgCommand:
|
||||
assert "Не удалось получить данные о погоде" in ctx.send.call_args[0][0]
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_current_condition_none(self) -> None:
|
||||
async def test_pg_current_condition_none(self):
|
||||
"""current_condition — пустой dict — бот должен сообщить об ошибке."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -96,7 +96,7 @@ class TestPgCommand:
|
||||
ctx.send.assert_called_once_with("Не удалось получить данные о погоде.")
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_wind_non_numeric(self) -> None:
|
||||
async def test_pg_wind_non_numeric(self):
|
||||
"""windspeedKmph — не число — wind должен быть '—'."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -109,7 +109,7 @@ class TestPgCommand:
|
||||
assert "Ветер: — м/с" in args
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_wind_none(self) -> None:
|
||||
async def test_pg_wind_none(self):
|
||||
"""windspeedKmph отсутствует — wind должен быть '—'."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -122,7 +122,7 @@ class TestPgCommand:
|
||||
assert "Ветер: — м/с" in args
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_zero_wind(self) -> None:
|
||||
async def test_pg_zero_wind(self):
|
||||
"""windspeedKmph = 0 — wind должен быть 0.0."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -135,7 +135,7 @@ class TestPgCommand:
|
||||
assert "Ветер: 0.0 м/с" in args
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_default_values(self) -> None:
|
||||
async def test_pg_default_values(self):
|
||||
"""Поля с отсутствующими значениями должны давать '—'."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -159,7 +159,7 @@ class TestPgCommand:
|
||||
assert "Давление: — мм рт. ст." in args
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_translate_unknown_weather(self) -> None:
|
||||
async def test_pg_translate_unknown_weather(self):
|
||||
"""Неизвестное описание погоды должно возвращать оригинал."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -172,7 +172,7 @@ class TestPgCommand:
|
||||
assert "Описание: UnknownXYZ" in args
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_russian_weather_description(self) -> None:
|
||||
async def test_pg_russian_weather_description(self):
|
||||
"""Описание погоды на русском должно корректно переводиться."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -185,7 +185,7 @@ class TestPgCommand:
|
||||
assert "Описание: Переменная облачность" in args
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_negative_pressure(self) -> None:
|
||||
async def test_pg_negative_pressure(self):
|
||||
"""Отрицательное давление должно конвертироваться."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
@ -198,7 +198,7 @@ class TestPgCommand:
|
||||
assert "Давление: -37.5 мм рт. ст." in args
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_pg_high_wind(self) -> None:
|
||||
async def test_pg_high_wind(self):
|
||||
"""Большая скорость ветра должна корректно округляться."""
|
||||
cog = self._make_cog()
|
||||
ctx = self._make_ctx()
|
||||
|
||||
@ -2,7 +2,6 @@
|
||||
import pytest
|
||||
from unittest.mock import AsyncMock, MagicMock
|
||||
|
||||
from commands.stats import Stats
|
||||
|
||||
class TestStatsCommand:
|
||||
"""Тесты Discord-команды stats."""
|
||||
@ -14,8 +13,10 @@ class TestStatsCommand:
|
||||
guild.member_count = member_count
|
||||
return guild
|
||||
|
||||
async def test_stats_sends_embed(self) -> None:
|
||||
async def test_stats_sends_embed(self):
|
||||
"""Команда stats отправляет embed-сообщение."""
|
||||
from commands.stats import Stats
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
mock_ctx.bot.latency = 0.035
|
||||
mock_ctx.bot.guilds = []
|
||||
@ -29,8 +30,10 @@ class TestStatsCommand:
|
||||
embed = call_args[1]["embed"] if call_args[1] else call_args[0][0]
|
||||
assert embed.title == "Статистика серверов"
|
||||
|
||||
async def test_stats_correct_values(self) -> None:
|
||||
async def test_stats_correct_values(self):
|
||||
"""Значения серверов, каналов и пользователей считаются верно."""
|
||||
from commands.stats import Stats
|
||||
|
||||
guild1 = self._make_mock_guild(channel_count=10, member_count=200)
|
||||
guild2 = self._make_mock_guild(channel_count=5, member_count=50)
|
||||
|
||||
@ -51,8 +54,10 @@ class TestStatsCommand:
|
||||
assert fields["Пользователей"] == "250"
|
||||
assert "35.0 мс" in fields["Пинг"]
|
||||
|
||||
async def test_stats_empty_guilds(self) -> None:
|
||||
async def test_stats_empty_guilds(self):
|
||||
"""Пустой список серверов не вызывает ошибок."""
|
||||
from commands.stats import Stats
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
mock_ctx.bot.latency = 0.050
|
||||
mock_ctx.bot.guilds = []
|
||||
@ -68,8 +73,10 @@ class TestStatsCommand:
|
||||
assert fields["Каналов"] == "0"
|
||||
assert fields["Пользователей"] == "0"
|
||||
|
||||
async def test_stats_none_member_count(self) -> None:
|
||||
async def test_stats_none_member_count(self):
|
||||
"""member_count=None не вызывает ошибок."""
|
||||
from commands.stats import Stats
|
||||
|
||||
guild = self._make_mock_guild(channel_count=3, member_count=None)
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
@ -85,9 +92,10 @@ class TestStatsCommand:
|
||||
fields = {f.name: f.value for f in embed.fields}
|
||||
assert fields["Пользователей"] == "0"
|
||||
|
||||
async def test_stats_excludes_categories(self) -> None:
|
||||
async def test_stats_excludes_categories(self):
|
||||
"""Категории не входят в счётчик каналов."""
|
||||
import discord
|
||||
from commands.stats import Stats
|
||||
|
||||
guild = MagicMock()
|
||||
text_ch = MagicMock()
|
||||
|
||||
@ -4,14 +4,14 @@ import time
|
||||
import pytest
|
||||
from unittest.mock import AsyncMock, MagicMock, patch
|
||||
|
||||
from commands.status import Status
|
||||
|
||||
|
||||
class TestStatusCommand:
|
||||
"""Тесты Discord-команды status."""
|
||||
|
||||
async def test_status_sends_embed(self) -> None:
|
||||
async def test_status_sends_embed(self):
|
||||
"""Команда status отправляет embed-сообщение."""
|
||||
from commands.status import Status
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
mock_ctx.bot.latency = 0.042
|
||||
mock_ctx.bot._start_time = time.time()
|
||||
@ -26,8 +26,10 @@ class TestStatusCommand:
|
||||
assert embed.title == "Статус бота"
|
||||
assert "42.0 мс" in embed.fields[0].value
|
||||
|
||||
async def test_status_uptime_format(self) -> None:
|
||||
async def test_status_uptime_format(self):
|
||||
"""Uptime форматируется корректно."""
|
||||
from commands.status import Status
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
mock_ctx.bot.latency = 0.050
|
||||
mock_ctx.bot._start_time = time.time() - 90061 # 1д 1ч 1м 1с
|
||||
@ -48,22 +50,30 @@ class TestStatusCommand:
|
||||
class TestFormatUptime:
|
||||
"""Тесты форматирования uptime."""
|
||||
|
||||
def test_zero_seconds(self) -> None:
|
||||
def test_zero_seconds(self):
|
||||
from commands.status import Status
|
||||
|
||||
result = Status._format_uptime(0)
|
||||
assert result == "0с"
|
||||
|
||||
def test_minutes_and_seconds(self) -> None:
|
||||
def test_minutes_and_seconds(self):
|
||||
from commands.status import Status
|
||||
|
||||
result = Status._format_uptime(125) # 2м 5с
|
||||
assert "2м" in result
|
||||
assert "5с" in result
|
||||
|
||||
def test_hours_minutes_seconds(self) -> None:
|
||||
def test_hours_minutes_seconds(self):
|
||||
from commands.status import Status
|
||||
|
||||
result = Status._format_uptime(3661) # 1ч 1м 1с
|
||||
assert "1ч" in result
|
||||
assert "1м" in result
|
||||
assert "1с" in result
|
||||
|
||||
def test_full_day(self) -> None:
|
||||
def test_full_day(self):
|
||||
from commands.status import Status
|
||||
|
||||
result = Status._format_uptime(90061) # 1д 1ч 1м 1с
|
||||
assert "1д" in result
|
||||
assert "1ч" in result
|
||||
|
||||
@ -1,7 +1,5 @@
|
||||
import json
|
||||
import pytest
|
||||
import requests
|
||||
from requests.exceptions import ConnectionError, Timeout, SSLError
|
||||
from unittest.mock import patch, MagicMock
|
||||
from utils.cat import fetch_cat
|
||||
|
||||
@ -10,7 +8,7 @@ class TestFetchCat:
|
||||
"""Тесты функции fetch_cat() — получение URL случайного котика."""
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_success(self, mock_get) -> None:
|
||||
async def test_fetch_cat_success(self, mock_get):
|
||||
"""Успешный ответ с URL должен вернуть строку."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = [{"url": "https://example.com/cat.jpg"}]
|
||||
@ -20,7 +18,7 @@ class TestFetchCat:
|
||||
assert result == "https://example.com/cat.jpg"
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_empty_array(self, mock_get) -> None:
|
||||
async def test_fetch_cat_empty_array(self, mock_get):
|
||||
"""Пустой массив должен вернуть None."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = []
|
||||
@ -30,8 +28,9 @@ class TestFetchCat:
|
||||
assert result is None
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_http_error(self, mock_get) -> None:
|
||||
async def test_fetch_cat_http_error(self, mock_get):
|
||||
"""HTTP-ошибка (raise_for_status) должна вернуть None."""
|
||||
import requests
|
||||
mock_response = MagicMock()
|
||||
mock_response.raise_for_status.side_effect = requests.HTTPError("404 Not Found")
|
||||
mock_get.return_value = mock_response
|
||||
@ -39,29 +38,33 @@ class TestFetchCat:
|
||||
assert result is None
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_connection_error(self, mock_get) -> None:
|
||||
async def test_fetch_cat_connection_error(self, mock_get):
|
||||
"""ConnectionError должна вернуть None."""
|
||||
from requests.exceptions import ConnectionError
|
||||
mock_get.side_effect = ConnectionError("No connection")
|
||||
result = await fetch_cat()
|
||||
assert result is None
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_timeout(self, mock_get) -> None:
|
||||
async def test_fetch_cat_timeout(self, mock_get):
|
||||
"""Timeout должна вернуть None."""
|
||||
from requests.exceptions import Timeout
|
||||
mock_get.side_effect = Timeout("Request timed out")
|
||||
result = await fetch_cat()
|
||||
assert result is None
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_ssl_error(self, mock_get) -> None:
|
||||
async def test_fetch_cat_ssl_error(self, mock_get):
|
||||
"""SSLError должна вернуть None."""
|
||||
from requests.exceptions import SSLError
|
||||
mock_get.side_effect = SSLError("SSL handshake failed")
|
||||
result = await fetch_cat()
|
||||
assert result is None
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_json_parse_error(self, mock_get) -> None:
|
||||
async def test_fetch_cat_json_parse_error(self, mock_get):
|
||||
"""Ошибка парсинга JSON должна вернуть None."""
|
||||
import requests
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.side_effect = requests.JSONDecodeError("Expecting value", "", 0)
|
||||
mock_response.raise_for_status = MagicMock()
|
||||
@ -70,7 +73,7 @@ class TestFetchCat:
|
||||
assert result is None
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_missing_url_key(self, mock_get) -> None:
|
||||
async def test_fetch_cat_missing_url_key(self, mock_get):
|
||||
"""Отсутствие ключа 'url' в ответе должно вернуть None."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = [{"error": "no image"}]
|
||||
@ -80,14 +83,15 @@ class TestFetchCat:
|
||||
assert result is None
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_request_exception(self, mock_get) -> None:
|
||||
async def test_fetch_cat_request_exception(self, mock_get):
|
||||
"""Общий RequestException должен вернуть None."""
|
||||
import requests
|
||||
mock_get.side_effect = requests.RequestException("Generic error")
|
||||
result = await fetch_cat()
|
||||
assert result is None
|
||||
|
||||
@patch("utils.cat._session.get")
|
||||
async def test_fetch_cat_url_with_special_chars(self, mock_get) -> None:
|
||||
async def test_fetch_cat_url_with_special_chars(self, mock_get):
|
||||
"""URL со спецсимволами должен вернуться как есть."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = [{"url": "https://example.com/cat?w=100&h=200"}]
|
||||
|
||||
@ -1,6 +1,5 @@
|
||||
|
||||
import pytest
|
||||
import requests
|
||||
from unittest.mock import patch, MagicMock
|
||||
from utils.news import fetch_rss
|
||||
|
||||
@ -9,7 +8,7 @@ class TestFetchRss:
|
||||
"""Тесты функции fetch_rss() — получение и парсинг RSS-ленты."""
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_success_rss20(self, mock_get) -> None:
|
||||
async def test_fetch_rss_success_rss20(self, mock_get):
|
||||
"""Успешный ответ RSS 2.0 должен вернуть список статей."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -48,7 +47,7 @@ class TestFetchRss:
|
||||
assert result[1]["tags"] == []
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_success_atom(self, mock_get) -> None:
|
||||
async def test_fetch_rss_success_atom(self, mock_get):
|
||||
"""Успешный ответ Atom должен вернуть список статей."""
|
||||
atom_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<feed xmlns="http://www.w3.org/2005/Atom">
|
||||
@ -74,7 +73,7 @@ class TestFetchRss:
|
||||
assert result[0]["tags"] == ["AI"]
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_empty_items(self, mock_get) -> None:
|
||||
async def test_fetch_rss_empty_items(self, mock_get):
|
||||
"""RSS без items должен вернуть пустой список."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -89,7 +88,7 @@ class TestFetchRss:
|
||||
assert result == []
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_no_matching_format(self, mock_get) -> None:
|
||||
async def test_fetch_rss_no_matching_format(self, mock_get):
|
||||
"""Неизвестный формат XML должен вернуть пустой список."""
|
||||
xml_content = """<?xml version="1.0"?>
|
||||
<unknown></unknown>""".encode()
|
||||
@ -101,7 +100,7 @@ class TestFetchRss:
|
||||
assert result == []
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_missing_title(self, mock_get) -> None:
|
||||
async def test_fetch_rss_missing_title(self, mock_get):
|
||||
"""Статья без title должна получить 'Без названия'."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -126,7 +125,7 @@ class TestFetchRss:
|
||||
assert result[0]["tags"] == []
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_missing_guid(self, mock_get) -> None:
|
||||
async def test_fetch_rss_missing_guid(self, mock_get):
|
||||
"""Статья без guid isPermaLink должна иметь пустую ссылку."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -148,7 +147,7 @@ class TestFetchRss:
|
||||
assert result[0]["link"] == ""
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_limit_to_10(self, mock_get) -> None:
|
||||
async def test_fetch_rss_limit_to_10(self, mock_get):
|
||||
"""Больше 10 items должно быть обрезано до 10."""
|
||||
items = "\n".join(
|
||||
f""" <item>
|
||||
@ -175,35 +174,39 @@ class TestFetchRss:
|
||||
assert result[9]["title"] == "Статья 9"
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_http_error(self, mock_get) -> None:
|
||||
async def test_fetch_rss_http_error(self, mock_get):
|
||||
"""HTTP-ошибка должна вернуть None."""
|
||||
import requests
|
||||
mock_get.side_effect = requests.exceptions.HTTPError("404 Not Found")
|
||||
result = await fetch_rss("https://example.com/rss")
|
||||
assert result is None
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_connection_error(self, mock_get) -> None:
|
||||
async def test_fetch_rss_connection_error(self, mock_get):
|
||||
"""Ошибка соединения должна вернуть None."""
|
||||
import requests
|
||||
mock_get.side_effect = requests.exceptions.ConnectionError("No connection")
|
||||
result = await fetch_rss("https://example.com/rss")
|
||||
assert result is None
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_timeout(self, mock_get) -> None:
|
||||
async def test_fetch_rss_timeout(self, mock_get):
|
||||
"""Таймаут должен вернуть None."""
|
||||
import requests
|
||||
mock_get.side_effect = requests.exceptions.Timeout("Request timed out")
|
||||
result = await fetch_rss("https://example.com/rss")
|
||||
assert result is None
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_ssl_error(self, mock_get) -> None:
|
||||
async def test_fetch_rss_ssl_error(self, mock_get):
|
||||
"""SSLError должен вернуть None."""
|
||||
import requests
|
||||
mock_get.side_effect = requests.exceptions.SSLError("SSL handshake failed")
|
||||
result = await fetch_rss("https://example.com/rss")
|
||||
assert result is None
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_empty_tags(self, mock_get) -> None:
|
||||
async def test_fetch_rss_empty_tags(self, mock_get):
|
||||
"""Статья с пустыми тегами должна иметь пустые строки."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -228,7 +231,7 @@ class TestFetchRss:
|
||||
assert result[0]["tags"] == []
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_category_without_text(self, mock_get) -> None:
|
||||
async def test_fetch_rss_category_without_text(self, mock_get):
|
||||
"""Категория без текста должна быть пропущена."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -250,7 +253,7 @@ class TestFetchRss:
|
||||
assert result[0]["tags"] == ["AI"]
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_atom_missing_author(self, mock_get) -> None:
|
||||
async def test_fetch_rss_atom_missing_author(self, mock_get):
|
||||
"""Atom feed без автора должен иметь пустого creator."""
|
||||
atom_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<feed xmlns="http://www.w3.org/2005/Atom">
|
||||
@ -270,7 +273,7 @@ class TestFetchRss:
|
||||
assert result[0]["creator"] == ""
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_atom_missing_link(self, mock_get) -> None:
|
||||
async def test_fetch_rss_atom_missing_link(self, mock_get):
|
||||
"""Atom feed без link должен иметь пустую ссылку."""
|
||||
atom_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<feed xmlns="http://www.w3.org/2005/Atom">
|
||||
@ -289,14 +292,15 @@ class TestFetchRss:
|
||||
assert result[0]["link"] == ""
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_request_exception(self, mock_get) -> None:
|
||||
async def test_fetch_rss_request_exception(self, mock_get):
|
||||
"""Общий RequestException должен вернуть None."""
|
||||
import requests
|
||||
mock_get.side_effect = requests.RequestException("Generic error")
|
||||
result = await fetch_rss("https://example.com/rss")
|
||||
assert result is None
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_guid_fallback_to_link(self, mock_get) -> None:
|
||||
async def test_fetch_rss_guid_fallback_to_link(self, mock_get):
|
||||
"""Если нет guid isPermaLink, ссылка должна быть пустой."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -317,7 +321,7 @@ class TestFetchRss:
|
||||
assert result[0]["link"] == ""
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_single_item(self, mock_get) -> None:
|
||||
async def test_fetch_rss_single_item(self, mock_get):
|
||||
"""Один item должен быть распарсен корректно."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -344,7 +348,7 @@ class TestFetchRss:
|
||||
assert result[0]["tags"] == ["ML"]
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_special_characters_in_title(self, mock_get) -> None:
|
||||
async def test_fetch_rss_special_characters_in_title(self, mock_get):
|
||||
"""Заголовки со спецсимволами должны парситься корректно."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -366,7 +370,7 @@ class TestFetchRss:
|
||||
assert "ML" in result[0]["title"]
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_date_with_gmt(self, mock_get) -> None:
|
||||
async def test_fetch_rss_date_with_gmt(self, mock_get):
|
||||
"""Дата с GMT должна парситься корректно."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
@ -387,7 +391,7 @@ class TestFetchRss:
|
||||
assert result[0]["pub_date"] == "Mon, 28 May 2026 10:00:00 GMT"
|
||||
|
||||
@patch("utils.news._session.get")
|
||||
async def test_fetch_rss_many_categories(self, mock_get) -> None:
|
||||
async def test_fetch_rss_many_categories(self, mock_get):
|
||||
"""Множество категорий должны быть собраны."""
|
||||
rss_content = """<?xml version="1.0" encoding="UTF-8"?>
|
||||
<rss version="2.0">
|
||||
|
||||
@ -1,7 +1,5 @@
|
||||
|
||||
import pytest
|
||||
import requests
|
||||
from requests.exceptions import ConnectionError, Timeout, SSLError
|
||||
from unittest.mock import patch, MagicMock
|
||||
from utils.pogoda import fetch_weather, fetch_open_meteo
|
||||
|
||||
@ -10,7 +8,7 @@ class TestFetchWeather:
|
||||
"""Тесты функции fetch_weather() — получение погоды с retry-логикой."""
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_weather_success(self, mock_get) -> None:
|
||||
async def test_fetch_weather_success(self, mock_get):
|
||||
"""Успешный ответ должен вернуть JSON-данные."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = {"current_condition": [{"temp_C": 20}]}
|
||||
@ -20,8 +18,9 @@ class TestFetchWeather:
|
||||
assert result == {"current_condition": [{"temp_C": 20}]}
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_weather_fallback_on_ssl_error(self, mock_get) -> None:
|
||||
async def test_fetch_weather_fallback_on_ssl_error(self, mock_get):
|
||||
"""SSLError на первой попытке → fallback на Open-Meteo."""
|
||||
from requests.exceptions import SSLError
|
||||
mock_get.side_effect = [
|
||||
SSLError("SSL Error"),
|
||||
MagicMock(json=MagicMock(return_value={"result": "fallback"})),
|
||||
@ -32,8 +31,9 @@ class TestFetchWeather:
|
||||
assert result == {"result": "fallback"}
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_weather_fallback_on_connection_error(self, mock_get) -> None:
|
||||
async def test_fetch_weather_fallback_on_connection_error(self, mock_get):
|
||||
"""ConnectionError → fallback на Open-Meteo."""
|
||||
from requests.exceptions import ConnectionError
|
||||
mock_get.side_effect = ConnectionError("No connection")
|
||||
with patch("utils.pogoda.fetch_open_meteo") as mock_fallback:
|
||||
mock_fallback.return_value = {"result": "fallback"}
|
||||
@ -41,8 +41,9 @@ class TestFetchWeather:
|
||||
assert result == {"result": "fallback"}
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_weather_fallback_on_timeout(self, mock_get) -> None:
|
||||
async def test_fetch_weather_fallback_on_timeout(self, mock_get):
|
||||
"""Timeout → fallback на Open-Meteo."""
|
||||
from requests.exceptions import Timeout
|
||||
mock_get.side_effect = Timeout("Timed out")
|
||||
with patch("utils.pogoda.fetch_open_meteo") as mock_fallback:
|
||||
mock_fallback.return_value = {"result": "fallback"}
|
||||
@ -50,8 +51,9 @@ class TestFetchWeather:
|
||||
assert result == {"result": "fallback"}
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_weather_all_retries_fail(self, mock_get) -> None:
|
||||
async def test_fetch_weather_all_retries_fail(self, mock_get):
|
||||
"""Все попытки не удались → fallback на Open-Meteo."""
|
||||
from requests.exceptions import ConnectionError
|
||||
mock_get.side_effect = ConnectionError("No connection")
|
||||
with patch("utils.pogoda.fetch_open_meteo") as mock_fallback:
|
||||
mock_fallback.return_value = None
|
||||
@ -59,8 +61,9 @@ class TestFetchWeather:
|
||||
assert result is None
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_weather_request_exception(self, mock_get) -> None:
|
||||
async def test_fetch_weather_request_exception(self, mock_get):
|
||||
"""Общий RequestException → fallback на Open-Meteo."""
|
||||
import requests
|
||||
mock_get.side_effect = requests.RequestException("Generic error")
|
||||
with patch("utils.pogoda.fetch_open_meteo") as mock_fallback:
|
||||
mock_fallback.return_value = {"result": "fallback"}
|
||||
@ -68,7 +71,7 @@ class TestFetchWeather:
|
||||
assert result == {"result": "fallback"}
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_weather_http_error_no_fallback(self, mock_get) -> None:
|
||||
async def test_fetch_weather_http_error_no_fallback(self, mock_get):
|
||||
"""HTTP-ошибка (raise_for_status) не ловится, падает."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.raise_for_status.side_effect = Exception("HTTP 500")
|
||||
@ -81,7 +84,7 @@ class TestFetchOpenMeteo:
|
||||
"""Тесты функции fetch_open_meteo() — fallback на Open-Meteo API."""
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_success(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_success(self, mock_get):
|
||||
"""Успешный ответ должен вернуть данные в формате current_condition."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = {
|
||||
@ -105,7 +108,7 @@ class TestFetchOpenMeteo:
|
||||
assert result["current_condition"][0]["pressure"] == 1013
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_custom_coords(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_custom_coords(self, mock_get):
|
||||
"""Кастомные координаты должны быть в URL."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = {"current": {"temperature": 25, "apparent_temperature": 22, "weather_code": 0, "wind_speed_10m": 3, "relative_humidity_2m": 50, "pressure_msl": 1020}}
|
||||
@ -119,7 +122,7 @@ class TestFetchOpenMeteo:
|
||||
assert "37.6173" in call_url
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_missing_weather_code(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_missing_weather_code(self, mock_get):
|
||||
"""Отсутствующий weather_code → 'Неизвестно'."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = {"current": {"temperature": 10}}
|
||||
@ -130,8 +133,9 @@ class TestFetchOpenMeteo:
|
||||
assert result["current_condition"][0]["weatherDesc"] == [{"value": "Неизвестно"}]
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_ssl_error(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_ssl_error(self, mock_get):
|
||||
"""SSLError → вернуть None."""
|
||||
from requests.exceptions import SSLError
|
||||
mock_get.side_effect = SSLError("SSL Error")
|
||||
with patch("utils.pogoda.fetch_open_meteo") as mock_fallback:
|
||||
# Внутренний fallback тоже падает, проверяем что возвращается None
|
||||
@ -140,29 +144,33 @@ class TestFetchOpenMeteo:
|
||||
assert result is None
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_connection_error(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_connection_error(self, mock_get):
|
||||
"""ConnectionError → вернуть None."""
|
||||
from requests.exceptions import ConnectionError
|
||||
mock_get.side_effect = ConnectionError("No connection")
|
||||
result = await fetch_open_meteo()
|
||||
assert result is None
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_timeout(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_timeout(self, mock_get):
|
||||
"""Timeout → вернуть None."""
|
||||
from requests.exceptions import Timeout
|
||||
mock_get.side_effect = Timeout("Timed out")
|
||||
result = await fetch_open_meteo()
|
||||
assert result is None
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_request_exception(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_request_exception(self, mock_get):
|
||||
"""Общий RequestException → вернуть None."""
|
||||
import requests
|
||||
mock_get.side_effect = requests.RequestException("Error")
|
||||
result = await fetch_open_meteo()
|
||||
assert result is None
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_json_parse_error(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_json_parse_error(self, mock_get):
|
||||
"""Ошибка парсинга JSON → вернуть None."""
|
||||
import requests
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.side_effect = requests.RequestException("JSON Error")
|
||||
mock_response.raise_for_status = MagicMock()
|
||||
@ -171,8 +179,9 @@ class TestFetchOpenMeteo:
|
||||
assert result is None
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_retry_on_error(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_retry_on_error(self, mock_get):
|
||||
"""Retry: первая попытка падает, вторая успешна."""
|
||||
from requests.exceptions import ConnectionError
|
||||
success_response = MagicMock()
|
||||
success_response.json.return_value = {"current": {"temperature": 20, "apparent_temperature": 18, "weather_code": 1, "wind_speed_10m": 4, "relative_humidity_2m": 60, "pressure_msl": 1015}}
|
||||
success_response.raise_for_status = MagicMock()
|
||||
@ -182,16 +191,18 @@ class TestFetchOpenMeteo:
|
||||
assert mock_get.call_count == 2
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_all_retries_fail(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_all_retries_fail(self, mock_get):
|
||||
"""Все попытки неудачны → None."""
|
||||
from requests.exceptions import ConnectionError
|
||||
mock_get.side_effect = [ConnectionError("fail"), ConnectionError("fail"), ConnectionError("fail")]
|
||||
result = await fetch_open_meteo(max_retries=3)
|
||||
assert result is None
|
||||
assert mock_get.call_count == 3
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_http_error(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_http_error(self, mock_get):
|
||||
"""HTTP 404 → raise_for_status бросит исключение → None."""
|
||||
import requests
|
||||
mock_response = MagicMock()
|
||||
mock_response.raise_for_status.side_effect = requests.HTTPError("HTTP 404")
|
||||
mock_get.return_value = mock_response
|
||||
@ -199,7 +210,7 @@ class TestFetchOpenMeteo:
|
||||
assert result is None
|
||||
|
||||
@patch("utils.pogoda._session.get")
|
||||
async def test_fetch_open_meteo_wind_speed_0(self, mock_get) -> None:
|
||||
async def test_fetch_open_meteo_wind_speed_0(self, mock_get):
|
||||
"""Нулевая скорость ветра должна корректно обрабатываться."""
|
||||
mock_response = MagicMock()
|
||||
mock_response.json.return_value = {
|
||||
|
||||
@ -17,11 +17,11 @@ class TestTruncateTitle:
|
||||
("A" * 50, 100, "A" * 50),
|
||||
],
|
||||
)
|
||||
def test_truncate(self, title, max_len, expected) -> None:
|
||||
def test_truncate(self, title, max_len, expected):
|
||||
"""Проверка обрезки заголовка."""
|
||||
assert truncate_title(title, max_len) == expected
|
||||
|
||||
def test_truncate_default_max_len(self) -> None:
|
||||
def test_truncate_default_max_len(self):
|
||||
"""По умолчанию max_len=60."""
|
||||
long_title = "A" * 61
|
||||
result = truncate_title(long_title)
|
||||
@ -42,7 +42,7 @@ class TestParseDate:
|
||||
("2026-01-01T00:00:00Z", "2026.01.01"),
|
||||
],
|
||||
)
|
||||
def test_parse_date_known(self, pub_date, expected) -> None:
|
||||
def test_parse_date_known(self, pub_date, expected):
|
||||
"""Известные форматы даты должны парситься корректно."""
|
||||
assert _parse_date(pub_date) == expected
|
||||
|
||||
@ -53,11 +53,11 @@ class TestParseDate:
|
||||
(None, ""),
|
||||
],
|
||||
)
|
||||
def test_parse_date_empty(self, pub_date, expected) -> None:
|
||||
def test_parse_date_empty(self, pub_date, expected):
|
||||
"""Пустая или None дата должна вернуть пустую строку."""
|
||||
assert _parse_date(pub_date) == expected
|
||||
|
||||
def test_parse_date_invalid(self) -> None:
|
||||
def test_parse_date_invalid(self):
|
||||
"""Невалидная дата должна вернуть первые 10 символов."""
|
||||
result = _parse_date("invalid-date-string")
|
||||
assert result == "invalid.da" # первые 10 символов: 'invalid-da' → 'invalid.da' (replace('-','.'))
|
||||
@ -66,7 +66,7 @@ class TestParseDate:
|
||||
class TestFormatArticles:
|
||||
"""Тесты функции format_articles() — формирование строк для вывода."""
|
||||
|
||||
def test_format_articles_normal(self) -> None:
|
||||
def test_format_articles_normal(self):
|
||||
"""Нормальный список статей должен вернуть заголовок + 5 статей."""
|
||||
articles = [
|
||||
{
|
||||
@ -90,7 +90,7 @@ class TestFormatArticles:
|
||||
assert result[1] == "Статья 1\n28.05.2026 <https://habr.com/1>"
|
||||
assert result[2] == "Статья 2\n29.05.2026 <https://habr.com/2>"
|
||||
|
||||
def test_format_articles_limit_to_5(self) -> None:
|
||||
def test_format_articles_limit_to_5(self):
|
||||
"""Больше 5 статей должно быть обрезано до 5."""
|
||||
articles = [
|
||||
{"title": f"Статья {i}", "link": f"https://habr.com/{i}", "pub_date": "Mon, 28 May 2026 10:00:00 +0000", "creator": "", "tags": []}
|
||||
@ -100,18 +100,18 @@ class TestFormatArticles:
|
||||
assert len(result) == 6 # заголовок + 5 статей
|
||||
assert result[-1] == "Статья 4\n28.05.2026 <https://habr.com/4>"
|
||||
|
||||
def test_format_articles_empty_list(self) -> None:
|
||||
def test_format_articles_empty_list(self):
|
||||
"""Пустой список должен вернуть только заголовок."""
|
||||
result = format_articles([], "Заголовок", "https://habr.com/feed")
|
||||
assert result == ["**Заголовок**\n<https://habr.com/feed>"]
|
||||
assert len(result) == 1
|
||||
|
||||
def test_format_articles_none(self) -> None:
|
||||
def test_format_articles_none(self):
|
||||
"""None должен вызвать TypeError (articles[:5] на None)."""
|
||||
with pytest.raises(TypeError):
|
||||
format_articles(None, "Заголовок", "https://habr.com/feed")
|
||||
|
||||
def test_format_articles_single_article(self) -> None:
|
||||
def test_format_articles_single_article(self):
|
||||
"""Одна статья должна быть корректно отформатирована."""
|
||||
articles = [
|
||||
{
|
||||
@ -127,7 +127,7 @@ class TestFormatArticles:
|
||||
assert result[0] == "**Новости AI**\n<https://habr.com/ai>"
|
||||
assert result[1] == "Единственная статья\n28.05.2026 <https://habr.com/1>"
|
||||
|
||||
def test_format_articles_long_title_truncated(self) -> None:
|
||||
def test_format_articles_long_title_truncated(self):
|
||||
"""Длинный заголовок должен быть обрезан до 60 символов с '...'."""
|
||||
long_title = "A" * 100
|
||||
articles = [
|
||||
@ -137,7 +137,7 @@ class TestFormatArticles:
|
||||
assert len(result[1].split("\n")[0]) == 63 # 60 + "..."
|
||||
assert result[1].split("\n")[0].endswith("...")
|
||||
|
||||
def test_format_articles_short_title_unchanged(self) -> None:
|
||||
def test_format_articles_short_title_unchanged(self):
|
||||
"""Короткий заголовок должен остаться без изменений."""
|
||||
short_title = "Кот"
|
||||
articles = [
|
||||
@ -146,7 +146,7 @@ class TestFormatArticles:
|
||||
result = format_articles(articles, "Заголовок", "https://habr.com/feed")
|
||||
assert result[1].split("\n")[0] == "Кот"
|
||||
|
||||
def test_format_articles_exact_60_chars(self) -> None:
|
||||
def test_format_articles_exact_60_chars(self):
|
||||
"""Заголовок ровно 60 символов не должен обрезаться."""
|
||||
exact_title = "A" * 60
|
||||
articles = [
|
||||
@ -156,7 +156,7 @@ class TestFormatArticles:
|
||||
assert result[1].split("\n")[0] == exact_title
|
||||
assert "..." not in result[1]
|
||||
|
||||
def test_format_articles_iso_date(self) -> None:
|
||||
def test_format_articles_iso_date(self):
|
||||
"""Дата в формате ISO должна парситься корректно."""
|
||||
articles = [
|
||||
{
|
||||
@ -170,7 +170,7 @@ class TestFormatArticles:
|
||||
result = format_articles(articles, "Заголовок", "https://habr.com/feed")
|
||||
assert result[1] == "Статья\n2026.05.28 <https://habr.com/1>"
|
||||
|
||||
def test_format_articles_empty_date(self) -> None:
|
||||
def test_format_articles_empty_date(self):
|
||||
"""Пустая дата должна быть пустой строкой."""
|
||||
articles = [
|
||||
{"title": "Статья", "link": "https://habr.com/1", "pub_date": "", "creator": "", "tags": []}
|
||||
@ -178,7 +178,7 @@ class TestFormatArticles:
|
||||
result = format_articles(articles, "Заголовок", "https://habr.com/feed")
|
||||
assert result[1] == "Статья\n <https://habr.com/1>"
|
||||
|
||||
def test_format_articles_none_date(self) -> None:
|
||||
def test_format_articles_none_date(self):
|
||||
"""None дата должна быть пустой строкой."""
|
||||
articles = [
|
||||
{"title": "Статья", "link": "https://habr.com/1", "pub_date": None, "creator": "", "tags": []}
|
||||
@ -186,7 +186,7 @@ class TestFormatArticles:
|
||||
result = format_articles(articles, "Заголовок", "https://habr.com/feed")
|
||||
assert result[1] == "Статья\n <https://habr.com/1>"
|
||||
|
||||
def test_format_articles_empty_link(self) -> None:
|
||||
def test_format_articles_empty_link(self):
|
||||
"""Пустая ссылка должна быть пустой строкой в угловых скобках."""
|
||||
articles = [
|
||||
{"title": "Статья", "link": "", "pub_date": "Mon, 28 May 2026 10:00:00 +0000", "creator": "", "tags": []}
|
||||
@ -194,7 +194,7 @@ class TestFormatArticles:
|
||||
result = format_articles(articles, "Заголовок", "https://habr.com/feed")
|
||||
assert result[1].endswith(" <>")
|
||||
|
||||
def test_format_articles_russian_title(self) -> None:
|
||||
def test_format_articles_russian_title(self):
|
||||
"""Русские заголовки должны корректно отображаться."""
|
||||
articles = [
|
||||
{
|
||||
@ -208,7 +208,7 @@ class TestFormatArticles:
|
||||
result = format_articles(articles, "Новости AI", "https://habr.com/ai")
|
||||
assert "Искусственный интеллект в медицине" in result[1]
|
||||
|
||||
def test_format_articles_exact_5_articles(self) -> None:
|
||||
def test_format_articles_exact_5_articles(self):
|
||||
"""Ровно 5 статей должно быть включено."""
|
||||
articles = [
|
||||
{"title": f"Статья {i}", "link": f"https://habr.com/{i}", "pub_date": "Mon, 28 May 2026 10:00:00 +0000", "creator": "", "tags": []}
|
||||
@ -218,7 +218,7 @@ class TestFormatArticles:
|
||||
assert len(result) == 6 # заголовок + 5 статей
|
||||
assert result[-1] == "Статья 4\n28.05.2026 <https://habr.com/4>"
|
||||
|
||||
def test_format_articles_6th_article_excluded(self) -> None:
|
||||
def test_format_articles_6th_article_excluded(self):
|
||||
"""6-я статья должна быть исключена."""
|
||||
articles = [
|
||||
{"title": f"Статья {i}", "link": f"https://habr.com/{i}", "pub_date": "Mon, 28 May 2026 10:00:00 +0000", "creator": "", "tags": []}
|
||||
|
||||
@ -3,8 +3,6 @@
|
||||
import pytest
|
||||
from unittest.mock import AsyncMock, MagicMock
|
||||
|
||||
from commands.help import Help
|
||||
|
||||
|
||||
class TestHelpCommandDiscord:
|
||||
"""Тесты для команды help на Discord."""
|
||||
@ -17,8 +15,10 @@ class TestHelpCommandDiscord:
|
||||
cmd.__doc__ = doc
|
||||
return cmd
|
||||
|
||||
async def test_show_help_sends_simple_text(self) -> None:
|
||||
async def test_show_help_sends_simple_text(self):
|
||||
"""Проверка, что команда отправляет простое текстовое сообщение."""
|
||||
from commands.help import Help
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
mock_ctx.bot.commands = []
|
||||
mock_ctx.send = AsyncMock(return_value=None)
|
||||
@ -28,8 +28,10 @@ class TestHelpCommandDiscord:
|
||||
|
||||
mock_ctx.send.assert_awaited_once()
|
||||
|
||||
async def test_show_help_message_content(self) -> None:
|
||||
async def test_show_help_message_content(self):
|
||||
"""Проверка содержания отправленного сообщения."""
|
||||
from commands.help import Help
|
||||
|
||||
message_calls = []
|
||||
|
||||
def send_side_effect(text: str, *args, **kwargs):
|
||||
@ -42,6 +44,7 @@ class TestHelpCommandDiscord:
|
||||
self._make_mock_command("nw", "Топ-5 статей и топ-5 новостей AI с Habr"),
|
||||
self._make_mock_command("morning", "Утренний дайджест: погода + новости + котик"),
|
||||
self._make_mock_command("cat", "Случайный котик"),
|
||||
self._make_mock_command("msg", "Повторить текст в чате"),
|
||||
]
|
||||
mock_ctx.send = AsyncMock(side_effect=send_side_effect)
|
||||
|
||||
@ -56,13 +59,13 @@ class TestHelpCommandDiscord:
|
||||
assert "=" * 40 in message
|
||||
|
||||
# Проверяем наличие всех команд без кавычек
|
||||
commands = ["!pg", "!nw", "!morning", "!cat"]
|
||||
commands = ["!pg", "!nw", "!morning", "!cat", "!msg"]
|
||||
for cmd in commands:
|
||||
assert cmd in message, f"Команда {cmd} не найдена"
|
||||
|
||||
# Проверяем разделение тире между командой и описанием
|
||||
lines = [l.strip() for l in message.split("\n") if "—" in l]
|
||||
assert len(lines) >= 4
|
||||
assert len(lines) >= 5
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
@ -1,154 +0,0 @@
|
||||
"""Интеграционные тесты — проверка взаимодействия компонентов без внешних API.
|
||||
|
||||
Тестируют реальные объекты (без моков на уровне команд),
|
||||
но мокают только сетевые вызовы.
|
||||
"""
|
||||
|
||||
import pytest
|
||||
import requests
|
||||
from unittest.mock import AsyncMock, MagicMock, patch
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
async def loaded_bot():
|
||||
"""Загрузить все ког-модули в бота."""
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
|
||||
intents = discord.Intents.default()
|
||||
intents.message_content = True
|
||||
bot = commands.Bot(command_prefix="!", intents=intents)
|
||||
|
||||
from commands import ALL_COMMANDS
|
||||
for cog_class in ALL_COMMANDS:
|
||||
await bot.add_cog(cog_class())
|
||||
return bot
|
||||
|
||||
|
||||
class TestCogLoading:
|
||||
"""Проверка загрузки ког-модулей."""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_all_cogs_load(self, loaded_bot) -> None:
|
||||
"""Все ког-модули должны загружаться без ошибок."""
|
||||
from commands import ALL_COMMANDS
|
||||
assert len(loaded_bot.cogs) == len(ALL_COMMANDS)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_commands_registered(self, loaded_bot) -> None:
|
||||
"""Все команды должны быть зарегистрированы."""
|
||||
command_names = {cmd.name for cmd in loaded_bot.commands if cmd.cog is not None}
|
||||
expected = {"pg", "nw", "morning", "cat", "hp", "stats", "status"}
|
||||
assert command_names == expected
|
||||
|
||||
|
||||
class TestCommandFlow:
|
||||
"""Проверка полного потока команд (без сетевых вызовов)."""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_cat_command_success(self, loaded_bot) -> None:
|
||||
"""Команда !cat должна отправить embed с котиком при успешном ответе API."""
|
||||
with patch("commands.cat.fetch_cat", new_callable=AsyncMock) as mock_fetch:
|
||||
mock_fetch.return_value = "https://example.com/cat.jpg"
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
mock_ctx.author.name = "TestUser"
|
||||
mock_ctx.send = AsyncMock(return_value=None)
|
||||
|
||||
cat_cmd = loaded_bot.get_command("cat")
|
||||
assert cat_cmd is not None
|
||||
await cat_cmd.callback(cat_cmd.cog, mock_ctx)
|
||||
|
||||
mock_ctx.send.assert_awaited_once()
|
||||
embed = mock_ctx.send.call_args[1]["embed"]
|
||||
assert embed.title == "Котик для тебя!"
|
||||
assert embed.image.url == "https://example.com/cat.jpg"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_cat_command_failure(self, loaded_bot) -> None:
|
||||
"""Команда !cat при ошибке API должна отправить fallback сообщение."""
|
||||
with patch("commands.cat.fetch_cat", new_callable=AsyncMock) as mock_fetch:
|
||||
mock_fetch.return_value = None
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
mock_ctx.author.name = "TestUser"
|
||||
mock_ctx.send = AsyncMock(return_value=None)
|
||||
|
||||
cat_cmd = loaded_bot.get_command("cat")
|
||||
await cat_cmd.callback(cat_cmd.cog, mock_ctx)
|
||||
|
||||
mock_ctx.send.assert_awaited_once()
|
||||
content = mock_ctx.send.call_args[0][0]
|
||||
assert "Не удалось получить котика" in content
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stats_command_flow(self) -> None:
|
||||
"""Команда !stats должна показать реальную статистику бота."""
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
|
||||
intents = discord.Intents.default()
|
||||
intents.message_content = True
|
||||
bot = commands.Bot(command_prefix="!", intents=intents)
|
||||
|
||||
from commands.stats import Stats
|
||||
await bot.add_cog(Stats())
|
||||
|
||||
mock_ctx = MagicMock()
|
||||
guild = MagicMock()
|
||||
guild.channels = [] # нет CategoryChannel
|
||||
guild.member_count = 100
|
||||
mock_ctx.bot = MagicMock()
|
||||
mock_ctx.bot.guilds = [guild]
|
||||
mock_ctx.bot.latency = 0.050
|
||||
mock_ctx.send = AsyncMock(return_value=None)
|
||||
|
||||
stats_cmd = bot.get_command("stats")
|
||||
await stats_cmd.callback(stats_cmd.cog, mock_ctx)
|
||||
|
||||
mock_ctx.send.assert_awaited_once()
|
||||
embed = mock_ctx.send.call_args[1]["embed"]
|
||||
fields = {f.name: f.value for f in embed.fields}
|
||||
assert fields["Серверов"] == "1"
|
||||
assert fields["Каналов"] == "0"
|
||||
assert fields["Пользователей"] == "100"
|
||||
assert fields["Пинг"] == "50.0 мс"
|
||||
|
||||
|
||||
class TestUtilityFunctions:
|
||||
"""Проверка утилит без моков."""
|
||||
|
||||
def test_wmo_codes_mapping(self) -> None:
|
||||
"""wmo_to_russian должен переводить известные коды."""
|
||||
from utils.pogoda import wmo_to_russian
|
||||
|
||||
assert wmo_to_russian(0) == "Ясно"
|
||||
assert wmo_to_russian(3) == "Пасмурно"
|
||||
assert wmo_to_russian(51) == "Лёгкая морось"
|
||||
assert wmo_to_russian(None) == "Неизвестно"
|
||||
|
||||
def test_pressure_conversion(self) -> None:
|
||||
"""pressure_to_mmhg должен конвертировать mb в мм рт. ст."""
|
||||
from utils.pogoda import pressure_to_mmhg
|
||||
|
||||
assert pressure_to_mmhg(1013) == 759.8 # 1013 * 0.750062 = 759.81
|
||||
assert pressure_to_mmhg("—") == "—"
|
||||
assert pressure_to_mmhg(None) == "—"
|
||||
|
||||
def test_title_truncation(self) -> None:
|
||||
"""truncate_title должен обрезать длинные заголовки."""
|
||||
from utils.news import truncate_title
|
||||
|
||||
short = "Короткий заголовок"
|
||||
assert truncate_title(short) == short
|
||||
long_title = "A" * 100
|
||||
assert len(truncate_title(long_title, 20)) == 23 # 17 + "..."
|
||||
|
||||
def test_embed_text_truncation(self) -> None:
|
||||
"""truncate_embed_text должен обрезать до 4096 символов."""
|
||||
from utils.news import truncate_embed_text, truncate_embed_field
|
||||
|
||||
long_text = "A" * 5000
|
||||
assert len(truncate_embed_text(long_text)) == 4096
|
||||
assert truncate_embed_text(long_text).endswith("...")
|
||||
assert len(truncate_embed_field(long_text)) == 1024
|
||||
@ -1,8 +1,6 @@
|
||||
"""Тесты для utils/logger.py — проверка настройки логирования."""
|
||||
|
||||
import io
|
||||
import logging
|
||||
import logging.handlers
|
||||
import os
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
@ -45,6 +43,8 @@ def test_invalid_level_defaults_to_info() -> None:
|
||||
|
||||
def test_file_handler_when_logs_dir_exists() -> None:
|
||||
"""RotatingFileHandler добавляется если директория logs существует."""
|
||||
import logging.handlers
|
||||
|
||||
orig_cwd = os.getcwd()
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
logs_dir = Path(tmpdir) / "logs"
|
||||
@ -68,6 +68,8 @@ def test_file_handler_when_logs_dir_exists() -> None:
|
||||
|
||||
def test_logs_dir_created_automatically() -> None:
|
||||
"""Директория logs создаётся автоматически, если её нет."""
|
||||
import logging.handlers
|
||||
|
||||
orig_cwd = os.getcwd()
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
os.chdir(tmpdir)
|
||||
@ -107,6 +109,9 @@ def test_discord_level_is_info() -> None:
|
||||
|
||||
def test_log_message_format() -> None:
|
||||
"""Формат сообщения: время, уровень, имя модуля, текст."""
|
||||
import io
|
||||
import sys
|
||||
|
||||
with _isolated_logger() as root:
|
||||
# Replace stdout with our buffer
|
||||
buffer = io.StringIO()
|
||||
|
||||
@ -13,54 +13,58 @@ from utils.morning_runner import Scheduler, run_morning
|
||||
class TestSchedulerInit:
|
||||
"""Тесты инициализации Scheduler."""
|
||||
|
||||
def test_init_sets_morning_time(self) -> None:
|
||||
def test_init_sets_morning_time(self):
|
||||
"""Инициализация должна устанавливать время."""
|
||||
bot = AsyncMock()
|
||||
with patch.object(Scheduler, "_start_scheduler"):
|
||||
with patch("asyncio.create_task") as mock_task:
|
||||
scheduler = Scheduler(bot, "08:30")
|
||||
assert scheduler.morning_time == "08:30"
|
||||
|
||||
def test_init_default_morning_time(self) -> None:
|
||||
def test_init_default_morning_time(self):
|
||||
"""Инициализация с дефолтным временем."""
|
||||
bot = AsyncMock()
|
||||
with patch.object(Scheduler, "_start_scheduler"):
|
||||
with patch("asyncio.create_task") as mock_task:
|
||||
scheduler = Scheduler(bot)
|
||||
assert scheduler.morning_time == "07:00"
|
||||
|
||||
def test_init_creates_task(self) -> None:
|
||||
"""Инициализация должна вызывать _start_scheduler."""
|
||||
def test_init_creates_task(self):
|
||||
"""Инициализация должна создавать asyncio.Task."""
|
||||
bot = AsyncMock()
|
||||
with patch.object(Scheduler, "_start_scheduler") as mock_start:
|
||||
with patch("asyncio.create_task") as mock_task:
|
||||
scheduler = Scheduler(bot)
|
||||
mock_start.assert_called_once()
|
||||
mock_task.assert_called_once()
|
||||
assert scheduler._task is not None
|
||||
|
||||
|
||||
class TestSchedulerCalculateNextRun:
|
||||
"""Тесты расчёта следующего запуска."""
|
||||
|
||||
def test_next_run_today_before_time(self) -> None:
|
||||
def test_next_run_today_before_time(self):
|
||||
"""Если сейчас раньше времени — вернуть сегодня."""
|
||||
bot = AsyncMock()
|
||||
with patch.object(Scheduler, "_start_scheduler"):
|
||||
with patch("asyncio.create_task"):
|
||||
scheduler = Scheduler(bot, "14:00")
|
||||
|
||||
now = datetime(2026, 5, 29, 10, 0, 0)
|
||||
next_run = scheduler._calculate_next_run(now)
|
||||
assert next_run == datetime(2026, 5, 29, 14, 0, 0)
|
||||
|
||||
def test_next_run_tomorrow_after_time(self) -> None:
|
||||
def test_next_run_tomorrow_after_time(self):
|
||||
"""Если сейчас позже времени — вернуть завтра."""
|
||||
bot = AsyncMock()
|
||||
with patch.object(Scheduler, "_start_scheduler"):
|
||||
with patch("asyncio.create_task"):
|
||||
scheduler = Scheduler(bot, "14:00")
|
||||
|
||||
now = datetime(2026, 5, 29, 15, 0, 0)
|
||||
next_run = scheduler._calculate_next_run(now)
|
||||
assert next_run == datetime(2026, 5, 30, 14, 0, 0)
|
||||
|
||||
def test_next_run_exact_time(self) -> None:
|
||||
def test_next_run_exact_time(self):
|
||||
"""Если сейчас ровно время — вернуть завтра."""
|
||||
bot = AsyncMock()
|
||||
with patch.object(Scheduler, "_start_scheduler"):
|
||||
with patch("asyncio.create_task"):
|
||||
scheduler = Scheduler(bot, "14:00")
|
||||
|
||||
now = datetime(2026, 5, 29, 14, 0, 0)
|
||||
next_run = scheduler._calculate_next_run(now)
|
||||
assert next_run == datetime(2026, 5, 30, 14, 0, 0)
|
||||
@ -69,16 +73,14 @@ class TestSchedulerCalculateNextRun:
|
||||
class TestSchedulerStartStop:
|
||||
"""Тесты запуска/остановки планировщика."""
|
||||
|
||||
def test_start_starts_task(self) -> None:
|
||||
"""start() должен вызывать _start_scheduler (1 в __init__ + 1 в start, но реальный task один)."""
|
||||
def test_start_starts_task(self):
|
||||
"""start() должен запустить task."""
|
||||
bot = AsyncMock()
|
||||
with patch.object(Scheduler, "_start_scheduler") as mock_start:
|
||||
with patch("asyncio.create_task"):
|
||||
scheduler = Scheduler(bot)
|
||||
scheduler.start()
|
||||
# __init__ вызывает _start_scheduler, start() тоже вызывает
|
||||
assert mock_start.call_count == 2
|
||||
assert scheduler._running is True
|
||||
|
||||
def test_stop_stops_task(self) -> None:
|
||||
def test_stop_stops_task(self):
|
||||
"""stop() должен остановить task."""
|
||||
bot = AsyncMock()
|
||||
with patch("asyncio.create_task"):
|
||||
@ -86,20 +88,21 @@ class TestSchedulerStartStop:
|
||||
scheduler.stop()
|
||||
assert scheduler._running is False
|
||||
|
||||
def test_double_start_no_duplicate(self) -> None:
|
||||
"""Повторный start должен вызывать _start_scheduler дважды (реальный task не дублируется благодаря флагам)."""
|
||||
def test_double_start_no_duplicate(self):
|
||||
"""Повторный start не должен создавать второй task."""
|
||||
bot = AsyncMock()
|
||||
with patch.object(Scheduler, "_start_scheduler") as mock_start:
|
||||
with patch("asyncio.create_task") as mock_task:
|
||||
scheduler = Scheduler(bot)
|
||||
scheduler.start() # второй вызов
|
||||
assert mock_start.call_count == 2
|
||||
# create_task вызван только при инициализации
|
||||
assert mock_task.call_count == 1
|
||||
|
||||
|
||||
class TestRunMorning:
|
||||
"""Тесты run_morning."""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_morning_sends_embed(self) -> None:
|
||||
async def test_run_morning_sends_embed(self):
|
||||
"""run_morning должен отправлять embed в канал."""
|
||||
bot = AsyncMock()
|
||||
channel = AsyncMock()
|
||||
@ -131,7 +134,7 @@ class TestRunMorningWithFallback:
|
||||
"""Тесты fallback в пустом embed."""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_morning_empty_embed_fallback(self) -> None:
|
||||
async def test_run_morning_empty_embed_fallback(self):
|
||||
"""run_morning должен добавлять fallback сообщение при пустых данных."""
|
||||
bot = AsyncMock()
|
||||
channel = AsyncMock()
|
||||
@ -160,7 +163,7 @@ class TestRunMorningWithFallback:
|
||||
assert "Не удалось получить данные из внешних источников" in embed_description
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_morning_only_weather_data(self) -> None:
|
||||
async def test_run_morning_only_weather_data(self):
|
||||
"""run_morning должен корректно обрабатывать только погоду без новостей."""
|
||||
bot = AsyncMock()
|
||||
channel = AsyncMock()
|
||||
|
||||
@ -5,7 +5,7 @@ from utils.pogoda import translate_weather, pressure_to_mmhg, wmo_to_russian, fo
|
||||
class TestFormatWeatherDataForConsole:
|
||||
"""Тесты функции format_weather_data_for_console()."""
|
||||
|
||||
def test_format_valid_data(self) -> None:
|
||||
def test_format_valid_data(self):
|
||||
"""Полные данные должны быть отформатированы корректно."""
|
||||
data = {
|
||||
"current_condition": [{
|
||||
@ -28,7 +28,7 @@ class TestFormatWeatherDataForConsole:
|
||||
assert "Ветер: 2.8 м/с" in result[3] # 10 / 3.6 = 2.777... ≈ 2.8
|
||||
assert "Давление: 759.8 мм рт. ст." in result[4]
|
||||
|
||||
def test_format_empty_data(self) -> None:
|
||||
def test_format_empty_data(self):
|
||||
"""Пустые данные должны возвращать None."""
|
||||
data = {
|
||||
"current_condition": [{}]
|
||||
@ -38,7 +38,7 @@ class TestFormatWeatherDataForConsole:
|
||||
|
||||
assert result is None, "Пустые данные должны возвращать None"
|
||||
|
||||
def test_format_missing_current_condition(self) -> None:
|
||||
def test_format_missing_current_condition(self):
|
||||
"""Отсутствие current_condition должно вернуть None."""
|
||||
data = {}
|
||||
|
||||
@ -46,7 +46,7 @@ class TestFormatWeatherDataForConsole:
|
||||
|
||||
assert result is None, "Отсутствие current_condition должно вернуть None"
|
||||
|
||||
def test_format_with_dashes(self) -> None:
|
||||
def test_format_with_dashes(self):
|
||||
"""Неизвестные значения должны отображаться как '—'."""
|
||||
data = {
|
||||
"current_condition": [{
|
||||
@ -68,7 +68,7 @@ class TestFormatWeatherDataForConsole:
|
||||
assert "Ветер: — м/с" in result[3]
|
||||
assert "Давление: — мм рт. ст." in result[4]
|
||||
|
||||
def test_format_wind_conversion(self) -> None:
|
||||
def test_format_wind_conversion(self):
|
||||
"""Проверка конвертации ветра из км/ч в м/с."""
|
||||
data = {
|
||||
"current_condition": [{
|
||||
@ -85,7 +85,7 @@ class TestFormatWeatherDataForConsole:
|
||||
# 36 / 3.6 = 10.0
|
||||
assert "Ветер: 10.0 м/с" in result[3]
|
||||
|
||||
def test_format_negative_temperature(self) -> None:
|
||||
def test_format_negative_temperature(self):
|
||||
"""Отрицательная температура должна отображаться корректно."""
|
||||
data = {
|
||||
"current_condition": [{
|
||||
@ -141,7 +141,7 @@ class TestTranslateWeather:
|
||||
("Moderate or heavy rain in area", "Дождь"),
|
||||
],
|
||||
)
|
||||
def test_translate_known(self, english, expected) -> None:
|
||||
def test_translate_known(self, english, expected):
|
||||
"""Известные переводы должны возвращать ожидаемый результат."""
|
||||
assert translate_weather(english) == expected
|
||||
|
||||
@ -153,34 +153,34 @@ class TestTranslateWeather:
|
||||
(" ", " "), # пробелы не считаются пустыми
|
||||
],
|
||||
)
|
||||
def test_translate_empty(self, input_value, expected) -> None:
|
||||
def test_translate_empty(self, input_value, expected):
|
||||
"""Пустой или None ввод должен возвращать '—'."""
|
||||
assert translate_weather(input_value) == expected
|
||||
|
||||
def test_translate_unknown_returns_original(self) -> None:
|
||||
def test_translate_unknown_returns_original(self):
|
||||
"""Неизвестный перевод должен возвращать оригинальный текст."""
|
||||
unknown_text = "Unknown weather condition XYZ"
|
||||
assert translate_weather(unknown_text) == unknown_text
|
||||
|
||||
def test_translate_partial_match(self) -> None:
|
||||
def test_translate_partial_match(self):
|
||||
"""Частичное совпадение ключа в тексте должно сработать."""
|
||||
# "Moderate or heavy rain in area" должно найтись в "Light Moderate or heavy rain in area"
|
||||
text_with_prefix = "Light Moderate or heavy rain in area"
|
||||
assert translate_weather(text_with_prefix) == "Дождь"
|
||||
|
||||
def test_translate_longer_key_priority(self) -> None:
|
||||
def test_translate_longer_key_priority(self):
|
||||
"""Длинные ключи проверяются первыми (_WEATHER_MAPPING отсортирован по убыванию длины).
|
||||
"Moderate or heavy rain at times" проверится до "Heavy rain"."""
|
||||
text = "Moderate or heavy rain at times"
|
||||
assert translate_weather(text) == "Дождь"
|
||||
|
||||
def test_translate_case_insensitive(self) -> None:
|
||||
def test_translate_case_insensitive(self):
|
||||
"""Перевод должен быть регистронезависимым."""
|
||||
assert translate_weather("CLEAR") == "Ясно"
|
||||
assert translate_weather("partly cloudy") == "Переменная облачность"
|
||||
assert translate_weather("HEAVY RAIN") == "Сильный дождь"
|
||||
|
||||
def test_translate_with_whitespace(self) -> None:
|
||||
def test_translate_with_whitespace(self):
|
||||
"""Текст с пробелами по краям должен корректно переводиться."""
|
||||
assert translate_weather(" Clear ") == "Ясно"
|
||||
|
||||
@ -198,7 +198,7 @@ class TestPressureToMMHG:
|
||||
# (0, "—"), # 0 — falsy, возвращается '—' (баг)
|
||||
],
|
||||
)
|
||||
def test_pressure_valid(self, mb, expected) -> None:
|
||||
def test_pressure_valid(self, mb, expected):
|
||||
"""Валидные числовые значения должны конвертироваться корректно."""
|
||||
assert pressure_to_mmhg(mb) == expected
|
||||
|
||||
@ -210,7 +210,7 @@ class TestPressureToMMHG:
|
||||
("980", 735.1),
|
||||
],
|
||||
)
|
||||
def test_pressure_string(self, mb, expected) -> None:
|
||||
def test_pressure_string(self, mb, expected):
|
||||
"""Строка-число должна конвертироваться корректно."""
|
||||
assert pressure_to_mmhg(mb) == expected
|
||||
|
||||
@ -222,27 +222,27 @@ class TestPressureToMMHG:
|
||||
("", "—"),
|
||||
],
|
||||
)
|
||||
def test_pressure_invalid(self, input_value, expected) -> None:
|
||||
def test_pressure_invalid(self, input_value, expected):
|
||||
"""Невалидные значения должны возвращать '—'."""
|
||||
assert pressure_to_mmhg(input_value) == expected
|
||||
|
||||
def test_pressure_non_numeric_string(self) -> None:
|
||||
def test_pressure_non_numeric_string(self):
|
||||
"""Невалидная строка должна возвращать '—'."""
|
||||
assert pressure_to_mmhg("abc") == "—"
|
||||
|
||||
def test_pressure_zero(self) -> None:
|
||||
def test_pressure_zero(self):
|
||||
"""Нулевое значение — корректно конвертируется в 0.0."""
|
||||
assert pressure_to_mmhg(0) == 0.0
|
||||
|
||||
def test_pressure_negative(self) -> None:
|
||||
def test_pressure_negative(self):
|
||||
"""Отрицательное значение должно конвертироваться."""
|
||||
assert pressure_to_mmhg(-100) == -75.0
|
||||
|
||||
def test_pressure_float_string(self) -> None:
|
||||
def test_pressure_float_string(self):
|
||||
"""Строка с десятичной точкой должна конвертироваться."""
|
||||
assert pressure_to_mmhg("1013.25") == 760.0
|
||||
|
||||
def test_pressure_very_large(self) -> None:
|
||||
def test_pressure_very_large(self):
|
||||
"""Очень большое значение должно работать."""
|
||||
assert pressure_to_mmhg(999999) == 750061.2
|
||||
|
||||
@ -283,26 +283,26 @@ class TestWmoToRussian:
|
||||
(99, "Сильная гроза с градом"),
|
||||
],
|
||||
)
|
||||
def test_wmo_known(self, code, expected) -> None:
|
||||
def test_wmo_known(self, code, expected):
|
||||
"""Известные WMO коды должны возвращать ожидаемый перевод."""
|
||||
assert wmo_to_russian(code) == expected
|
||||
|
||||
def test_wmo_unknown(self) -> None:
|
||||
def test_wmo_unknown(self):
|
||||
"""Неизвестный код должен возвращать 'Неизвестно'."""
|
||||
assert wmo_to_russian(999) == "Неизвестно"
|
||||
|
||||
def test_wmo_negative_code(self) -> None:
|
||||
def test_wmo_negative_code(self):
|
||||
"""Отрицательный код должен возвращать 'Неизвестно'."""
|
||||
assert wmo_to_russian(-1) == "Неизвестно"
|
||||
|
||||
def test_wmo_none(self) -> None:
|
||||
def test_wmo_none(self):
|
||||
"""None должен возвращать 'Неизвестно'."""
|
||||
assert wmo_to_russian(None) == "Неизвестно"
|
||||
|
||||
def test_wmo_large_code(self) -> None:
|
||||
def test_wmo_large_code(self):
|
||||
"""Очень большой код должен возвращать 'Неизвестно'."""
|
||||
assert wmo_to_russian(9999) == "Неизвестно"
|
||||
|
||||
def test_wmo_float_code(self) -> None:
|
||||
def test_wmo_float_code(self):
|
||||
"""Дробный код — не найдётся в mapping."""
|
||||
assert wmo_to_russian(1.5) == "Неизвестно"
|
||||
|
||||
@ -52,13 +52,13 @@ async def gather_morning() -> MorningData:
|
||||
)
|
||||
|
||||
|
||||
async def run_morning(bot: "commands.Bot", channel: discord.TextChannel) -> None:
|
||||
async def run_morning(bot: "commands.Bot", channel: discord.TextChannel):
|
||||
"""Выполнить утренний дайджест и отправить в канал."""
|
||||
try:
|
||||
data = await gather_morning()
|
||||
|
||||
# --- Формируем embed ---
|
||||
embed = discord.Embed(title="Утренний дайджест", color=0xF4A460)
|
||||
embed = discord.Embed(title="🌅 Утренний дайджест!", color=0xF4A460)
|
||||
|
||||
# Котик как thumbnail
|
||||
if data.cat_url:
|
||||
@ -115,12 +115,12 @@ async def run_morning(bot: "commands.Bot", channel: discord.TextChannel) -> None
|
||||
description = "\n".join(description_lines)
|
||||
embed.description = truncate_embed_text(description)
|
||||
await channel.send(embed=embed)
|
||||
logger.info("Утренний дайджест отправлен в #%s", channel.name)
|
||||
logger.info("✅ Утренний дайджест отправлен в #%s", channel.name)
|
||||
|
||||
except Exception as e:
|
||||
logger.error("Ошибка при выполнении утреннего дайджеста: %s", e, exc_info=True)
|
||||
try:
|
||||
await channel.send("Не удалось выполнить утренний дайджест.")
|
||||
await channel.send("❌ Не удалось выполнить утренний дайджест.")
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
@ -249,8 +249,8 @@ class Scheduler:
|
||||
if not sent:
|
||||
logger.error("Не удалось найти канал для отправки morning-дайджеста")
|
||||
|
||||
def start(self) -> None:
|
||||
def start(self):
|
||||
self._start_scheduler()
|
||||
|
||||
def stop(self) -> None:
|
||||
def stop(self):
|
||||
self._stop_scheduler()
|
||||
|
||||
@ -1,7 +1,6 @@
|
||||
import asyncio
|
||||
import logging
|
||||
from datetime import datetime
|
||||
from typing import Optional
|
||||
|
||||
import requests
|
||||
|
||||
@ -15,7 +14,7 @@ RSS_URL_POSTS = "https://habr.com/ru/rss/hubs/artificial_intelligence/news/top/d
|
||||
_session = requests.Session()
|
||||
|
||||
|
||||
async def fetch_rss(url: str) -> Optional[list[dict]]:
|
||||
async def fetch_rss(url):
|
||||
"""Скачать и распарсить RSS-ленту (RSS 2.0 / Atom)."""
|
||||
await habr_rss_limiter.acquire()
|
||||
from defusedxml.ElementTree import fromstring
|
||||
@ -75,7 +74,7 @@ async def fetch_rss(url: str) -> Optional[list[dict]]:
|
||||
return None
|
||||
|
||||
|
||||
def _parse_date(pub_date: Optional[str]) -> str:
|
||||
def _parse_date(pub_date):
|
||||
"""Парсить дату из RSS в строку 'дд.мм.гггг' или вернуть часть даты."""
|
||||
if not pub_date:
|
||||
return ""
|
||||
@ -87,7 +86,7 @@ def _parse_date(pub_date: Optional[str]) -> str:
|
||||
return pub_date[:10].replace("-", ".")
|
||||
|
||||
|
||||
def truncate_title(title: str, max_len: int = 60) -> str:
|
||||
def truncate_title(title, max_len=60):
|
||||
"""Обрезать заголовок, если он длиннее max_len."""
|
||||
if len(title) > max_len:
|
||||
return title[:max_len] + "..."
|
||||
@ -108,7 +107,7 @@ def truncate_embed_field(text: str, max_len: int = 1024) -> str:
|
||||
return text[:max_len - 3] + "..."
|
||||
|
||||
|
||||
def format_articles(articles: list[dict], title: str, link: str) -> list[str]:
|
||||
def format_articles(articles, title, link):
|
||||
"""Сформировать список строк для вывода статей/постов."""
|
||||
lines = [f"**{title}**\n<{link}>"]
|
||||
for i, article in enumerate(articles[:5], 1):
|
||||
|
||||
@ -1,7 +1,5 @@
|
||||
import asyncio
|
||||
import logging
|
||||
from typing import Any, Optional
|
||||
|
||||
import requests
|
||||
from requests.exceptions import ConnectionError, Timeout, SSLError
|
||||
|
||||
@ -14,7 +12,7 @@ API_URL_WEATHER = "https://wttr.in/Magnitogorsk?format=j1&lang=ru"
|
||||
_session = requests.Session()
|
||||
|
||||
|
||||
async def fetch_weather(api_url: str, timeout: int = 10, max_retries: int = 3) -> Optional[dict]:
|
||||
async def fetch_weather(api_url, timeout=10, max_retries=3):
|
||||
"""Получить данные о погоде с retry."""
|
||||
await weather_limiter.acquire()
|
||||
for attempt in range(max_retries):
|
||||
@ -37,7 +35,7 @@ async def fetch_weather(api_url: str, timeout: int = 10, max_retries: int = 3) -
|
||||
return await fetch_open_meteo()
|
||||
|
||||
|
||||
async def fetch_open_meteo(lat: float = 53.4069, lon: float = 58.9797, timeout: int = 10, max_retries: int = 3) -> Optional[dict]:
|
||||
async def fetch_open_meteo(lat=53.4069, lon=58.9797, timeout=10, max_retries=3):
|
||||
"""Fallback на Open-Meteo API."""
|
||||
await open_meteo_limiter.acquire()
|
||||
url = (
|
||||
@ -80,7 +78,7 @@ async def fetch_open_meteo(lat: float = 53.4069, lon: float = 58.9797, timeout:
|
||||
return None
|
||||
|
||||
|
||||
def wmo_to_russian(code: Optional[int]) -> str:
|
||||
def wmo_to_russian(code):
|
||||
"""Перевод WMO weather code в русский."""
|
||||
mapping = {
|
||||
0: "Ясно",
|
||||
@ -141,7 +139,7 @@ _WEATHER_MAPPING = [
|
||||
]
|
||||
|
||||
|
||||
def translate_weather(en: Optional[str]) -> str:
|
||||
def translate_weather(en):
|
||||
if not en:
|
||||
return "—"
|
||||
en_lower = en.lower()
|
||||
@ -151,7 +149,7 @@ def translate_weather(en: Optional[str]) -> str:
|
||||
return en
|
||||
|
||||
|
||||
def format_weather_data_for_console(data: Optional[dict]) -> Optional[list[str]]:
|
||||
def format_weather_data_for_console(data):
|
||||
"""
|
||||
Форматировать погодные данные для консольного вывода.
|
||||
|
||||
@ -188,7 +186,7 @@ def format_weather_data_for_console(data: Optional[dict]) -> Optional[list[str]]
|
||||
]
|
||||
|
||||
|
||||
def format_weather_for_embed(data: Optional[dict]) -> Optional[str]:
|
||||
def format_weather_for_embed(data):
|
||||
"""Форматировать погоду для Discord embed (с заголовком)."""
|
||||
if data is None:
|
||||
return None
|
||||
@ -198,7 +196,7 @@ def format_weather_for_embed(data: Optional[dict]) -> Optional[str]:
|
||||
return "**Погода в Магнитогорске:**\n" + "\n".join(lines)
|
||||
|
||||
|
||||
def pressure_to_mmhg(mb: Any) -> float | str:
|
||||
def pressure_to_mmhg(mb):
|
||||
if mb == "—" or mb is None or mb == "":
|
||||
return "—"
|
||||
try:
|
||||
|
||||
@ -53,7 +53,7 @@ class RateLimiter:
|
||||
await asyncio.sleep(token / self.rate)
|
||||
|
||||
|
||||
# --- Конфигурация лимитеров ---
|
||||
# --- Готовые лимитеры по API ---
|
||||
|
||||
# TheCatAPI: бесплатно, 1 req/sec, burst 3
|
||||
_CAT_RATE: Final[float] = float(os.getenv("CAT_API_RATE", "1"))
|
||||
@ -71,29 +71,8 @@ _OPEN_METEO_BURST: Final[int] = int(os.getenv("OPEN_METEO_API_BURST", "5"))
|
||||
_HABR_RSS_RATE: Final[float] = float(os.getenv("HABR_RSS_RATE", "1"))
|
||||
_HABR_RSS_BURST: Final[int] = int(os.getenv("HABR_RSS_BURST", "2"))
|
||||
|
||||
|
||||
def make_cat_limiter() -> RateLimiter:
|
||||
"""Создать лимитер для TheCatAPI."""
|
||||
return RateLimiter(_CAT_RATE, _CAT_BURST)
|
||||
|
||||
|
||||
def make_weather_limiter() -> RateLimiter:
|
||||
"""Создать лимитер для wttr.in."""
|
||||
return RateLimiter(_WEATHER_RATE, _WEATHER_BURST)
|
||||
|
||||
|
||||
def make_open_meteo_limiter() -> RateLimiter:
|
||||
"""Создать лимитер для Open-Meteo."""
|
||||
return RateLimiter(_OPEN_METEO_RATE, _OPEN_METEO_BURST)
|
||||
|
||||
|
||||
def make_habr_rss_limiter() -> RateLimiter:
|
||||
"""Создать лимитер для Habr RSS."""
|
||||
return RateLimiter(_HABR_RSS_RATE, _HABR_RSS_BURST)
|
||||
|
||||
|
||||
# Экземпляры лимитеров (глобальные, для production)
|
||||
cat_limiter: RateLimiter = make_cat_limiter()
|
||||
weather_limiter: RateLimiter = make_weather_limiter()
|
||||
open_meteo_limiter: RateLimiter = make_open_meteo_limiter()
|
||||
habr_rss_limiter: RateLimiter = make_habr_rss_limiter()
|
||||
# Экземпляры лимитеров
|
||||
cat_limiter: RateLimiter = RateLimiter(_CAT_RATE, _CAT_BURST)
|
||||
weather_limiter: RateLimiter = RateLimiter(_WEATHER_RATE, _WEATHER_BURST)
|
||||
open_meteo_limiter: RateLimiter = RateLimiter(_OPEN_METEO_RATE, _OPEN_METEO_BURST)
|
||||
habr_rss_limiter: RateLimiter = RateLimiter(_HABR_RSS_RATE, _HABR_RSS_BURST)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user