mirror of
https://github.com/kuhyx/signal-bot.git
synced 2026-11-11 19:14:04 +01:00
Krzysztof kuhy Rudnicki
4b457da8f3
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
# Copyright (c) 2026 Krzysztof Rudnicki. MIT licence.
|
|
"""Scoping and tally predicates."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
|
|
from signal_bot.counting import format_tally, should_count, watched_group
|
|
from signal_bot.models import parse_envelope
|
|
from tests.conftest import OTHER_GROUP, SECOND_GROUP, WATCHED_GROUP, envelope
|
|
|
|
if TYPE_CHECKING:
|
|
from signal_bot.config import Config
|
|
|
|
|
|
def _message(**kwargs: object) -> object:
|
|
parsed = parse_envelope(envelope(**kwargs))
|
|
assert parsed is not None
|
|
return parsed
|
|
|
|
|
|
@pytest.mark.parametrize("group", [WATCHED_GROUP, SECOND_GROUP])
|
|
def test_every_watched_group_is_accepted_as_itself(config: Config, group: str) -> None:
|
|
"""Each watched group resolves to itself, never to the first one listed."""
|
|
assert watched_group(_message(group_id=group), config) == group
|
|
|
|
|
|
def test_other_group_is_rejected(config: Config) -> None:
|
|
assert watched_group(_message(group_id=OTHER_GROUP), config) is None
|
|
|
|
|
|
def test_direct_message_is_rejected(config: Config) -> None:
|
|
"""The dangerous case: a linked device receives every DM on the account."""
|
|
assert watched_group(_message(group_id=None), config) is None
|
|
|
|
|
|
def test_plain_text_counts() -> None:
|
|
assert should_count(_message(text="hello"))
|
|
|
|
|
|
@pytest.mark.parametrize("kwargs", [{"sticker": True}, {"reaction": True}])
|
|
def test_stickers_and_reactions_do_not_count(kwargs: dict[str, bool]) -> None:
|
|
assert not should_count(_message(**kwargs))
|
|
|
|
|
|
@pytest.mark.parametrize("text", [None, ""])
|
|
def test_empty_text_does_not_count(text: str | None) -> None:
|
|
assert not should_count(_message(text=text))
|
|
|
|
|
|
def test_format_tally_pads_names() -> None:
|
|
rendered = format_tally([("Alice", 5), ("Bo", 2)])
|
|
assert rendered == "Alice 5\nBo 2"
|
|
|
|
|
|
def test_format_tally_when_nobody_spoke() -> None:
|
|
assert format_tally([]) == "Brak wiadomosci dzisiaj."
|