Marius Mutu
42a1a0e7fd
- Telegram /h /help — listă comenzi în română - atm.bat — pornire cu venv local automat, pip install la primul run - tzdata adăugat în deps principale cu marker sys_platform==win32 - README: secțiuni dev, instalare Windows, flow-uri calibrare Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
"""Tests for atm.commands — /pause /resume parsing (Commit 5)."""
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from atm.commands import Command, TelegramPoller
|
|
|
|
|
|
def _make_poller() -> TelegramPoller:
|
|
cfg = MagicMock()
|
|
cfg.bot_token = "tok"
|
|
cfg.chat_id = "123"
|
|
cfg.allowed_chat_ids = ("123",)
|
|
cfg.poll_timeout_s = 1
|
|
return TelegramPoller(cfg, MagicMock(), MagicMock())
|
|
|
|
|
|
def test_parse_pause():
|
|
p = _make_poller()
|
|
assert p._parse_command("pause") == Command(action="pause")
|
|
assert p._parse_command("/pause") == Command(action="pause")
|
|
|
|
|
|
def test_parse_resume_plain():
|
|
p = _make_poller()
|
|
assert p._parse_command("resume") == Command(action="resume")
|
|
assert p._parse_command("/resume") == Command(action="resume")
|
|
|
|
|
|
def test_parse_resume_force():
|
|
p = _make_poller()
|
|
# "resume force" → value=1 signals force-resume of canary drift
|
|
cmd = p._parse_command("resume force")
|
|
assert cmd is not None
|
|
assert cmd.action == "resume"
|
|
assert cmd.value == 1
|
|
|
|
|
|
def test_parse_help():
|
|
p = _make_poller()
|
|
assert p._parse_command("h") == Command(action="help")
|
|
assert p._parse_command("/h") == Command(action="help")
|
|
assert p._parse_command("help") == Command(action="help")
|
|
assert p._parse_command("/help") == Command(action="help")
|
|
|
|
|
|
def test_parse_existing_commands_still_work():
|
|
"""Regression: adding pause/resume must not break stop/status/ss/interval."""
|
|
p = _make_poller()
|
|
assert p._parse_command("stop") == Command(action="stop")
|
|
assert p._parse_command("status") == Command(action="status")
|
|
assert p._parse_command("ss") == Command(action="ss")
|
|
assert p._parse_command("3") == Command(action="set_interval", value=180)
|