Add health endpoint

This commit is contained in:
Jack Kingsman
2026-03-12 19:47:21 -07:00
parent 338f632514
commit 2710cafb21
18 changed files with 607 additions and 73 deletions
+120
View File
@@ -5,6 +5,7 @@ Uses httpx.AsyncClient or direct function calls with real in-memory SQLite.
"""
import hashlib
import logging
import time
from unittest.mock import AsyncMock, MagicMock, patch
@@ -109,6 +110,125 @@ class TestHealthEndpoint:
assert data["connection_info"] is None
class TestDebugEndpoint:
"""Test the debug support snapshot endpoint."""
@pytest.mark.asyncio
async def test_support_snapshot_returns_logs_and_live_radio_audits(self, test_db, client):
"""Debug snapshot should include recent logs plus live radio/contact/channel state."""
from meshcore import EventType
from app.config import clear_recent_log_lines
from app.routers.debug import DebugApplicationInfo
clear_recent_log_lines()
contact_key = "ab" * 32
channel_key = "CD" * 16
await _insert_contact(contact_key, "Alice", last_contacted=1700000000)
await ChannelRepository.upsert(key=channel_key, name="#flightless", on_radio=False)
radio_manager.max_channels = 2
radio_manager.path_hash_mode = 1
radio_manager.path_hash_mode_supported = True
radio_manager._connection_info = "Serial: /dev/ttyUSB0"
radio_manager.note_channel_slot_loaded(channel_key, 0)
mock_mc = MagicMock()
mock_mc.self_info = {"name": "TestNode", "radio_freq": 910.525}
mock_mc.stop_auto_message_fetching = AsyncMock()
mock_mc.start_auto_message_fetching = AsyncMock()
mock_mc.commands.send_device_query = AsyncMock(
return_value=MagicMock(type=EventType.DEVICE_INFO, payload={"max_channels": 2})
)
mock_mc.commands.get_stats_core = AsyncMock(
return_value=MagicMock(type=EventType.STATS_CORE, payload={"heap_free": 1234})
)
mock_mc.commands.get_stats_radio = AsyncMock(
return_value=MagicMock(type=EventType.STATS_RADIO, payload={"tx_good": 9})
)
mock_mc.commands.get_contacts = AsyncMock(
return_value=MagicMock(
type=EventType.OK,
payload={contact_key: {"adv_name": "Alice"}},
)
)
def _channel_event(slot: int) -> MagicMock:
if slot == 0:
return MagicMock(
type=EventType.CHANNEL_INFO,
payload={
"channel_name": "#flightless",
"channel_secret": bytes.fromhex(channel_key),
},
)
return MagicMock(type=EventType.OK, payload={})
mock_mc.commands.get_channel = AsyncMock(side_effect=_channel_event)
radio_manager._meshcore = mock_mc
logging.getLogger("tests.debug").warning("support snapshot marker")
with patch(
"app.routers.debug._build_application_info",
return_value=DebugApplicationInfo(
version="3.2.0",
commit_hash="deadbeef",
git_branch="main",
git_dirty=False,
python_version="3.12.0",
),
):
response = await client.get("/api/debug")
assert response.status_code == 200
payload = response.json()
assert payload["application"]["commit_hash"] == "deadbeef"
assert payload["runtime"]["channel_slot_reuse_enabled"] is True
assert payload["runtime"]["channel_send_cache"] == [{"channel_key": channel_key, "slot": 0}]
assert any("support snapshot marker" in line for line in payload["logs"])
radio_probe = payload["radio_probe"]
assert radio_probe["performed"] is True
assert radio_probe["device_info"] == {"max_channels": 2}
assert radio_probe["stats_core"] == {"heap_free": 1234}
assert radio_probe["stats_radio"] == {"tx_good": 9}
assert radio_probe["contacts"]["expected_and_found"] == 1
assert radio_probe["contacts"]["expected_but_not_found"] == []
assert radio_probe["contacts"]["found_but_not_expected"] == []
assert radio_probe["channels"]["matched_slots"] == 2
assert radio_probe["channels"]["wrong_slots"] == []
@pytest.mark.asyncio
async def test_support_snapshot_returns_runtime_when_disconnected(self, test_db, client):
"""Debug snapshot should still return logs and runtime state when radio is disconnected."""
from app.config import clear_recent_log_lines
from app.routers.debug import DebugApplicationInfo
clear_recent_log_lines()
radio_manager._meshcore = None
radio_manager._connection_info = None
with patch(
"app.routers.debug._build_application_info",
return_value=DebugApplicationInfo(
version="3.2.0",
commit_hash="deadbeef",
git_branch="main",
git_dirty=False,
python_version="3.12.0",
),
):
response = await client.get("/api/debug")
assert response.status_code == 200
payload = response.json()
assert payload["radio_probe"]["performed"] is False
assert payload["radio_probe"]["errors"] == ["Radio not connected"]
class TestRadioDisconnectedHandler:
"""Test that RadioDisconnectedError maps to 503."""
+1 -1
View File
@@ -915,7 +915,7 @@ class TestOnNewContact:
contact = await ContactRepository.get_by_key("cc" * 32)
assert contact is not None
assert contact.name == "Charlie"
assert contact.on_radio is True
assert contact.on_radio is False
assert contact.last_seen == 1700000000
mock_broadcast.assert_called_once()
+35 -23
View File
@@ -20,6 +20,7 @@ from app.radio_sync import (
ensure_contact_on_radio,
is_polling_paused,
pause_polling,
sync_and_offload_all,
sync_radio_time,
sync_recent_contacts_to_radio,
)
@@ -211,11 +212,6 @@ class TestSyncRecentContactsToRadio:
result = await sync_recent_contacts_to_radio()
assert result["loaded"] == 2
# Verify contacts are now marked as on_radio in DB
alice = await ContactRepository.get_by_key(KEY_A)
bob = await ContactRepository.get_by_key(KEY_B)
assert alice.on_radio is True
assert bob.on_radio is True
@pytest.mark.asyncio
async def test_fills_remaining_slots_with_recently_contacted_then_advertised(self, test_db):
@@ -272,6 +268,39 @@ class TestSyncRecentContactsToRadio:
]
assert loaded_keys == favorite_keys
class TestSyncAndOffloadAll:
"""Test session-local contact radio residency reset behavior."""
@pytest.mark.asyncio
async def test_clears_stale_contact_on_radio_flags_before_reload(self, test_db):
await _insert_contact(KEY_A, "Alice", on_radio=True)
await _insert_contact(KEY_B, "Bob", on_radio=True)
mock_mc = MagicMock()
with (
patch(
"app.radio_sync.sync_and_offload_contacts",
new=AsyncMock(return_value={"synced": 0, "removed": 0}),
),
patch(
"app.radio_sync.sync_and_offload_channels",
new=AsyncMock(return_value={"synced": 0, "cleared": 0}),
),
patch("app.radio_sync.ensure_default_channels", new=AsyncMock()),
patch(
"app.radio_sync.sync_recent_contacts_to_radio",
new=AsyncMock(return_value={"loaded": 0, "already_on_radio": 0, "failed": 0}),
),
):
await sync_and_offload_all(mock_mc)
alice = await ContactRepository.get_by_key(KEY_A)
bob = await ContactRepository.get_by_key(KEY_B)
assert alice is not None and alice.on_radio is False
assert bob is not None and bob.on_radio is False
@pytest.mark.asyncio
async def test_advert_fill_skips_repeaters(self, test_db):
"""Recent advert fallback only considers non-repeaters."""
@@ -325,7 +354,7 @@ class TestSyncRecentContactsToRadio:
@pytest.mark.asyncio
async def test_skips_contacts_already_on_radio(self, test_db):
"""Contacts already on radio are counted but not re-added."""
await _insert_contact(KEY_A, "Alice", on_radio=True)
await _insert_contact(KEY_A, "Alice", on_radio=False)
await AppSettingsRepository.update(favorites=[Favorite(type="contact", id=KEY_A)])
mock_mc = MagicMock()
@@ -382,23 +411,6 @@ class TestSyncRecentContactsToRadio:
assert result["loaded"] == 0
assert "error" in result
@pytest.mark.asyncio
async def test_marks_on_radio_when_found_but_not_flagged(self, test_db):
"""Contact found on radio but not flagged gets set_on_radio(True)."""
await _insert_contact(KEY_A, "Alice", on_radio=False)
await AppSettingsRepository.update(favorites=[Favorite(type="contact", id=KEY_A)])
mock_mc = MagicMock()
mock_mc.get_contact_by_key_prefix = MagicMock(return_value=MagicMock()) # Found
radio_manager._meshcore = mock_mc
result = await sync_recent_contacts_to_radio()
assert result["already_on_radio"] == 1
# Should update the flag since contact.on_radio was False
contact = await ContactRepository.get_by_key(KEY_A)
assert contact.on_radio is True
@pytest.mark.asyncio
async def test_handles_add_failure(self, test_db):
"""Failed add_contact increments the failed counter."""
-3
View File
@@ -161,9 +161,6 @@ class TestOutgoingDMBroadcast:
assert contact_payload["out_path"] == "aa00bb00"
assert contact_payload["out_path_len"] == 2
assert contact_payload["out_path_hash_mode"] == 1
contact = await ContactRepository.get_by_key(pub_key)
assert contact is not None
assert contact.on_radio is True
@pytest.mark.asyncio
async def test_send_dm_prefers_route_override_over_learned_path(self, test_db):