mirror of
https://github.com/kuhyx/signal-bot.git
synced 2026-11-11 20:54:04 +01:00
Krzysztof kuhy Rudnicki
c9a90eb48a
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
# Copyright (c) 2026 Krzysztof Rudnicki. MIT licence.
|
|
"""The console entrypoint.
|
|
|
|
`python main.py` used to do literally nothing: there was no __main__ block
|
|
and no uvicorn.run() call anywhere in the project.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import runpy
|
|
|
|
import pytest
|
|
import uvicorn
|
|
|
|
import signal_bot.__main__ as entrypoint
|
|
from signal_bot.portcheck import PortInUseError
|
|
|
|
BASE_ENV = {
|
|
"PHONE_NUMBER": "+15550000000",
|
|
"GROUP_IDS": "group.watched==",
|
|
}
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _isolated_env(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""Keep the developer's own .env out of the entrypoint's decisions.
|
|
|
|
load_dotenv() does not overwrite variables that are already set, but it
|
|
happily *adds* ones that are not -- so a real .env carrying BIND_PORT
|
|
would silently decide the outcome of the default-port test.
|
|
"""
|
|
monkeypatch.setattr("signal_bot.config.load_dotenv", lambda *a, **k: None)
|
|
for key, value in BASE_ENV.items():
|
|
monkeypatch.setenv(key, value)
|
|
for key in ("BIND_HOST", "BIND_PORT", "GROUP_ID", "GROUP_ID_SEND"):
|
|
monkeypatch.delenv(key, raising=False)
|
|
|
|
|
|
@pytest.fixture
|
|
def captured(monkeypatch: pytest.MonkeyPatch) -> dict[str, object]:
|
|
seen: dict[str, object] = {}
|
|
|
|
def fake_run(target: str, **kwargs: object) -> None:
|
|
seen["target"] = target
|
|
seen.update(kwargs)
|
|
|
|
monkeypatch.setattr(entrypoint.uvicorn, "run", fake_run)
|
|
monkeypatch.setattr(entrypoint, "assert_port_free", lambda host, port: None)
|
|
return seen
|
|
|
|
|
|
def test_runs_the_asgi_app_on_loopback_by_default(captured: dict[str, object]) -> None:
|
|
entrypoint.main()
|
|
assert captured["target"] == "signal_bot.app:app"
|
|
assert captured["host"] == "127.0.0.1"
|
|
assert captured["port"] == 8137
|
|
|
|
|
|
def test_bind_overrides_are_honoured(
|
|
captured: dict[str, object], monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
# TEST-NET-1: proves the override without a bind-all literal.
|
|
monkeypatch.setenv("BIND_HOST", "192.0.2.10")
|
|
monkeypatch.setenv("BIND_PORT", "9001")
|
|
entrypoint.main()
|
|
assert captured["host"] == "192.0.2.10"
|
|
assert captured["port"] == 9001
|
|
|
|
|
|
def test_a_taken_port_exits_non_zero_instead_of_starting(
|
|
captured: dict[str, object], monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
"""uvicorn would log the collision and exit 0; the bot must not."""
|
|
|
|
def refuse(host: str, port: int) -> None:
|
|
msg = "cannot bind 127.0.0.1:8137 -- held by pid 1: /sbin/init"
|
|
raise PortInUseError(msg)
|
|
|
|
monkeypatch.setattr(entrypoint, "assert_port_free", refuse)
|
|
with pytest.raises(SystemExit, match="held by pid 1"):
|
|
entrypoint.main()
|
|
assert "target" not in captured
|
|
|
|
|
|
def test_module_runs_as_a_script(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""`python -m signal_bot` must actually start the server."""
|
|
seen: dict[str, object] = {}
|
|
monkeypatch.setattr(uvicorn, "run", lambda target, **kw: seen.update(target=target))
|
|
monkeypatch.setattr("signal_bot.portcheck.assert_port_free", lambda host, port: None)
|
|
runpy.run_module("signal_bot", run_name="__main__", alter_sys=True)
|
|
assert seen["target"] == "signal_bot.app:app"
|