fix(telegram): clear send_path_degraded on successful reconnect (#35205) (#54076)

* fix(telegram): clear send_path_degraded on successful reconnect

_send_path_degraded was cleared only in _verify_polling_after_reconnect,
60s after reconnect and only if scheduled. A clean start_polling() reconnect
left the flag stuck True, short-circuiting send() and blocking all outbound
messages until the deferred probe ran (or forever if it never did).

Clear the flag the moment start_polling() succeeds — that is the recovery
signal. The deferred probe remains a defensive re-check that re-enters the
reconnect ladder (re-setting the flag) if it detects a silent wedge.

Fixes #35205.

* docs: add infographic for #35205 telegram send-path fix
This commit is contained in:
Teknium 2026-06-28 01:38:17 -07:00 committed by GitHub
parent 674e16e7c6
commit 2ecb6f7fe6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 64 additions and 6 deletions

View file

@ -66,8 +66,48 @@ async def test_send_short_circuits_when_path_degraded():
@pytest.mark.asyncio
async def test_reconnect_storm_sets_and_heartbeat_clears_flag(monkeypatch):
"""_handle_polling_network_error sets the flag; a successful heartbeat
probe in _verify_polling_after_reconnect clears it."""
"""_handle_polling_network_error sets the flag while reconnecting; if the
reconnect attempt itself raises (polling not yet healthy), the flag stays
True until a later successful heartbeat probe in
_verify_polling_after_reconnect clears it."""
adapter = _make_adapter()
adapter._app = MagicMock()
adapter._app.updater = MagicMock()
adapter._app.updater.running = True
adapter._app.updater.stop = AsyncMock()
# First start_polling attempt fails — the reconnect handler must leave the
# flag set (path still unhealthy) and not clear it prematurely.
adapter._app.updater.start_polling = AsyncMock(side_effect=OSError("still down"))
adapter._app.bot = MagicMock()
adapter._app.bot.get_me = AsyncMock(return_value=MagicMock())
adapter._polling_error_callback_ref = AsyncMock()
monkeypatch.setattr(
"plugins.platforms.telegram.adapter.Update", MagicMock(ALL_TYPES=[])
)
# Suppress the self-rescheduled retry so the test doesn't recurse.
monkeypatch.setattr(
"plugins.platforms.telegram.adapter.asyncio.ensure_future", MagicMock()
)
with patch("plugins.platforms.telegram.adapter.asyncio.sleep", new_callable=AsyncMock):
await adapter._handle_polling_network_error(OSError("Bad Gateway"))
# start_polling failed → path still degraded.
assert adapter._send_path_degraded is True
# Now the deferred probe runs against a recovered (running) updater and
# a responsive bot — it clears the flag.
adapter._app.updater.running = True
with patch("plugins.platforms.telegram.adapter.asyncio.sleep", new_callable=AsyncMock):
await adapter._verify_polling_after_reconnect()
assert adapter._send_path_degraded is False
@pytest.mark.asyncio
async def test_successful_reconnect_clears_flag_without_probe(monkeypatch):
"""Regression for #35205: a successful start_polling() clears
_send_path_degraded immediately, so outbound sends are not blocked for
the full HEARTBEAT_PROBE_DELAY window (and never get stuck True if the
deferred probe is never scheduled / never runs)."""
adapter = _make_adapter()
adapter._app = MagicMock()
adapter._app.updater = MagicMock()
@ -80,10 +120,17 @@ async def test_reconnect_storm_sets_and_heartbeat_clears_flag(monkeypatch):
monkeypatch.setattr(
"plugins.platforms.telegram.adapter.Update", MagicMock(ALL_TYPES=[])
)
await adapter._handle_polling_network_error(OSError("Bad Gateway"))
assert adapter._send_path_degraded is True
# Don't let the deferred probe run — prove the clear happens in the
# reconnect handler itself, not in _verify_polling_after_reconnect.
monkeypatch.setattr(
adapter, "_verify_polling_after_reconnect", AsyncMock()
)
with patch("plugins.platforms.telegram.adapter.asyncio.sleep", new_callable=AsyncMock):
await adapter._verify_polling_after_reconnect()
await adapter._handle_polling_network_error(OSError("Bad Gateway"))
assert adapter._send_path_degraded is False
assert adapter._polling_network_error_count == 0
# And send() works again right away.
result = await adapter.send("123", "hello")
assert result.success is True