signal-bot/tests/test_help_command.py
Krzysztof kuhy Rudnicki 6ec746fad2 feat: a word the bot does not know gets an answer, and gets kept
An unrecognised "!word" was dropped in silence, which made a typo
indistinguishable from the bot being down and threw away the best available
list of commands worth writing. Three things now happen at the one place in
dispatcher.py that used to be a bare `return`.

A near miss is corrected -- "!cta" asks whether you meant "!kot". The
distance is optimal string alignment rather than difflib's ratio: a
transposition is the typo people actually make, it costs one edit here, and
it shreds the matching blocks a ratio counts ("!cta"/"!cat" scores 0.75 but
"!cta"/"!kot" scores 0.50, so no cutoff separates them on principle).

A word with no near miss points at !zaproponuj, which files an idea from
inside the group. Either way the word and the message are recorded, and the
management page grows an Ideas tab that lists them.

Three things stay silent, each for its own reason. "!kot ladny" is a real
trigger with words after it and always has been silent. "!!" is punctuation.
And a command claimed in some *other* group is neither answered nor recorded
-- the album's triggers are real people's names, and pack_scoping.yaml says
it out loud: a refusal would itself announce that the album exists. The
existence check is therefore global, which is what pack_add already asks.

Measured rather than guessed, against the shipped aliases.json:

- a minimum candidate length, because "!ja" is one edit from ~40 of the 676
  two-letter words somebody could type; 69 of them got a suggestion before
  the rule and 18 after.
- the two-edit budget starts at six characters, not five. At five it put
  four pairs of *different* commands within reach of each other --
  !pies/!popes, !days/!cats, !dzis/!dni, !mnie/!dni -- and costs nothing to
  give up, because every real typo this was written for is one edit away.
  test_no_two_commands_are_within_reach_of_each_other fails if a future
  alias reopens the gap.

Both new tables are keyed on the envelope's own sent_at_ms, never the wall
clock: signal-cli replays undelivered frames on reconnect, and a clock-keyed
table collects a duplicate row per reconnect instead of ignoring one.

The branch is gated on should_count -- which excludes the bot's own envelope,
and that is the loop guard, since handle() checks is_own nowhere else -- and
wrapped in its own try/except, because it runs above the one that wraps a
handler and Bot.listen has no exception handling of its own.

Suggestions live in SQLite, not in a tracked file, and the export goes to the
state directory. check_personal_data learns the names it refuses from the
live database, so it catches a name somebody has typed and cannot catch a
misspelling of one -- and an unrecognised-command corpus is made of
misspellings. A group-writable file in a public tree is both a leak and a
wedge. The export therefore carries
deliberate !zaproponuj notes in full and unrecognised words as counts only;
bodies stay on the loopback-only page.

Verified on the live bot: restarted, both tables created with every existing
row count unchanged, !cta and !zaproponuj !pogoda delivered through
/_selftest, both rows read back out of the live database, and the journal
names neither the word nor the body.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01416TegJ3sa8xL5qGGgrygP
2026-08-31 16:48:26 +02:00

243 lines
8.7 KiB
Python

# Copyright (c) 2026 Krzysztof Rudnicki. MIT licence.
"""!pomoc, and the two gates that keep it from going stale."""
from __future__ import annotations
from typing import TYPE_CHECKING
import httpx
import pytest
from signal_bot.animals import register_animal_commands
from signal_bot.config import DEFAULT_COMMONS_API_URL
from signal_bot.context import CommandContext
from signal_bot.help_command import (
ALBUM_HEADING,
COMMAND_NAME,
GROUPS,
HEADING,
PACK_HEADING,
SPELLINGS,
register_help_command,
render_help,
)
from signal_bot.http_client import ImageFetcher
from signal_bot.models import parse_envelope
from signal_bot.pack_add import register_add_command
from signal_bot.pack_command import register_pack_commands
from signal_bot.pack_holder import PackHolder
from signal_bot.pack_load import load as load_pack
from signal_bot.pack_random import register_random_command
from signal_bot.popes_command import register_pope_command
from signal_bot.portraits import PortraitCache
from signal_bot.registry import CommandRegistry
from signal_bot.stats.commands import register_stats_commands
from signal_bot.suggest_command import register_suggest_command
from tests import pack_fixture
from tests.conftest import SECOND_GROUP, WATCHED_GROUP, envelope
if TYPE_CHECKING:
from pathlib import Path
from signal_bot.config import Config
from signal_bot.media_sniff import MediaKind
from signal_bot.store import Store
from signal_bot.transport_fake import FakeTransport
def _offline_fetcher() -> ImageFetcher:
"""A fetcher wired to a transport that refuses to be called.
!pomoc never fetches anything, and the other commands are registered here
only so that the help text has something to list.
"""
def handler(request: httpx.Request) -> httpx.Response:
raise AssertionError(request.url)
return ImageFetcher(httpx.AsyncClient(transport=httpx.MockTransport(handler)))
def full_registry() -> CommandRegistry:
"""Every command the live bot registers, wired the way build_bot wires it."""
registry = CommandRegistry()
fetcher = _offline_fetcher()
register_animal_commands(registry, fetcher)
register_pope_command(registry, PortraitCache(fetcher, DEFAULT_COMMONS_API_URL))
register_stats_commands(registry)
register_suggest_command(registry)
register_help_command(registry)
return registry
async def _noop(ctx: CommandContext) -> None:
"""A handler that does nothing."""
def test_every_registered_command_carries_a_description() -> None:
"""The whole point of making description a required argument."""
for command in full_registry().commands:
assert command.description.strip(), command.name
def test_help_lists_every_command_exactly_once() -> None:
"""A command missing from !pomoc, or listed twice, fails here.
Asserted on the whole line rather than the bare trigger, so prose that
happens to mention "!stats" can neither satisfy nor break it.
"""
registry = full_registry()
text = render_help(registry.commands, WATCHED_GROUP)
for command in registry.commands:
assert text.count(f"\n{command.primary} — {command.description}") == 1, command.name
def test_help_prints_the_polish_spelling_not_the_folded_one() -> None:
"""`triggers` holds skeletons; only `primary` still carries the diacritics."""
text = render_help(full_registry().commands, WATCHED_GROUP)
assert "!papież" in text
assert "!papiez" not in text
assert "!tydzień" in text
def test_headings_appear_in_order() -> None:
"""Obrazki, Papież, Statystyki, then the ungrouped block, then the note."""
text = render_help(full_registry().commands, WATCHED_GROUP)
positions = [text.index(f"\n{title}\n") for title, _ in GROUPS]
assert positions == sorted(positions)
assert text.startswith(HEADING)
assert text.endswith(SPELLINGS)
def test_a_command_outside_the_groups_is_still_listed() -> None:
"""The safety net: forgetting to add a command to GROUPS cannot hide it."""
registry = full_registry()
registry.register("ping", ["!ping"], _noop, "odpowiada pong")
assert "\n!ping — odpowiada pong" in render_help(registry.commands, WATCHED_GROUP)
def test_a_group_naming_an_unknown_command_fails_loudly() -> None:
"""A typo in GROUPS raises rather than silently dropping a line."""
registry = CommandRegistry()
register_help_command(registry)
with pytest.raises(KeyError):
render_help(registry.commands, WATCHED_GROUP)
async def test_the_handler_replies_with_the_rendered_list(
transport: FakeTransport, store: Store, config: Config
) -> None:
"""!pomoc answers in the group that asked, quoting the asker."""
registry = full_registry()
command = registry.resolve("!pomoc")
assert command is not None
assert command.name == COMMAND_NAME
message = parse_envelope(envelope(text="!pomoc"))
assert message is not None
await command.handler(
CommandContext(
message=message,
group_id=WATCHED_GROUP,
transport=transport,
store=store,
config=config,
command_name=COMMAND_NAME,
now=100.0,
day="2026-08-30",
)
)
assert transport.sent[0].recipient == WATCHED_GROUP
assert transport.sent[0].text == render_help(registry.commands, WATCHED_GROUP)
@pytest.mark.parametrize("trigger", ["!pomoc", "!pomocy", "!help", "!komendy", "!commands"])
def test_every_alias_resolves(trigger: str) -> None:
"""All five spellings reach the same command."""
command = full_registry().resolve(trigger)
assert command is not None
assert command.name == COMMAND_NAME
def test_an_album_gets_its_own_heading_and_a_hint(tmp_path: Path) -> None:
"""And the hint is built from real triggers, because they are the example."""
registry = full_registry()
register_pack_commands(registry, PackHolder(load_pack(pack_fixture.build(tmp_path / "pack"))))
text = render_help(registry.commands, WATCHED_GROUP)
assert f"\n{PACK_HEADING}\n" in text
assert "!alfa — losowe zdjęcie tej osoby" in text
assert "Kilka naraz (!alfa !beta)" in text
def test_the_album_is_invisible_in_every_other_group(tmp_path: Path) -> None:
"""The list is what would announce a private command family to strangers."""
registry = full_registry()
register_pack_commands(registry, PackHolder(load_pack(pack_fixture.build(tmp_path / "pack"))))
text = render_help(registry.commands, SECOND_GROUP)
assert PACK_HEADING not in text
assert "alfa" not in text
assert "!kot" in text
def test_a_pack_of_one_prints_no_hint(tmp_path: Path) -> None:
""" "Combine them" needs two things to combine."""
root = pack_fixture.build(
tmp_path / "pack",
people=[{"key": "p1", "triggers": ["!alfa"]}],
combos=[],
tags={},
)
registry = full_registry()
register_pack_commands(registry, PackHolder(load_pack(root)))
text = render_help(registry.commands, WATCHED_GROUP)
assert PACK_HEADING in text
assert "Kilka naraz" not in text
class _NoStripper:
"""Registration needs a stripper; rendering help never calls one."""
def strip(self, payload: bytes, kind: MediaKind) -> bytes | None:
return payload
def test_the_album_commands_get_their_own_block(tmp_path: Path) -> None:
"""Their own heading, so the "kilka naraz" hint keeps describing people.
And neither description may name anybody: !pomoc prints them, and the
trigger is the only place a name is allowed to appear.
"""
holder = PackHolder(load_pack(pack_fixture.build(tmp_path / "pack")))
registry = full_registry()
register_pack_commands(registry, holder)
register_random_command(registry, holder, ("!losowe",))
register_add_command(registry, holder, _NoStripper(), ("!dodaj",))
text = render_help(registry.commands, WATCHED_GROUP)
assert f"\n{ALBUM_HEADING}\n" in text
assert "!dodaj — dodaje załączone zdjęcie do albumu z podanymi etykietami" in text
assert "!losowe — losowe zdjęcie z albumu, opcjonalnie po etykiecie" in text
# The people's block is still its own, with its own hint.
assert f"\n{PACK_HEADING}\n" in text
assert "Kilka naraz (!alfa !beta)" in text
def test_the_album_commands_are_invisible_in_every_other_group(tmp_path: Path) -> None:
holder = PackHolder(load_pack(pack_fixture.build(tmp_path / "pack")))
registry = full_registry()
register_pack_commands(registry, holder)
register_random_command(registry, holder, ("!losowe",))
register_add_command(registry, holder, _NoStripper(), ("!dodaj",))
text = render_help(registry.commands, SECOND_GROUP)
assert ALBUM_HEADING not in text
assert "!dodaj" not in text
assert "!losowe" not in text