100% Coverage

tests/integration/test_db_seed.py

@@ -2,10 +2,15 @@
 
 from __future__ import annotations
 
+import contextlib
+from contextlib import asynccontextmanager
 from typing import TYPE_CHECKING
+from unittest.mock import AsyncMock, patch
+from uuid import uuid4
 
 import pytest
 from roboco.db.seed import (
+    bootstrap_database,
     create_agents,
     create_channel_memberships,
     create_channels,
@@ -66,9 +71,86 @@ async def test_create_initial_messages(db_session: AsyncSession) -> None:
     agent_ids = await create_agents(db_session)
     await create_channel_memberships(db_session, channel_ids, agent_ids)
     # Initial messages may or may not be seeded depending on config — just
-    # confirm the call doesn't raise.
-    try:
+    # confirm the call doesn't raise. Some setups may not have everything
+    # wired; accept silent skip.
+    with contextlib.suppress(Exception):
         await create_initial_messages(db_session, channel_ids, agent_ids)
-    except Exception:
-        # Some setups may not have everything wired; accept silent skip.
-        pass
+
+
+@pytest.mark.asyncio
+async def test_create_initial_messages_skips_when_channel_missing(
+    db_session: AsyncSession,
+) -> None:
+    """Missing channel/agent IDs cause continue branches (line 226)."""
+    # Empty mappings → every iteration hits the "skip" continue.
+    await create_initial_messages(db_session, {}, {})
+
+
+@pytest.mark.asyncio
+async def test_create_initial_messages_skips_when_already_present(
+    db_session: AsyncSession,
+) -> None:
+    """A second pass triggers the duplicate-detection skip (lines 238-239)."""
+    channel_ids = await create_channels(db_session)
+    agent_ids = await create_agents(db_session)
+    await create_channel_memberships(db_session, channel_ids, agent_ids)
+    # First pass seeds messages; second pass hits the existing-message skip.
+    await create_initial_messages(db_session, channel_ids, agent_ids)
+    await db_session.flush()
+    await create_initial_messages(db_session, channel_ids, agent_ids)
+
+
+@pytest.mark.asyncio
+async def test_create_channel_memberships_skips_unknown_channels(
+    db_session: AsyncSession,
+) -> None:
+    """Unknown channel slugs hit the continue branches (lines 159, 163, 179, 183)."""
+    agent_ids = await create_agents(db_session)
+    # Pass an empty channel map — every membership entry skips at line 159
+    # and every auditor-access entry skips at line 178.
+    await create_channel_memberships(db_session, {}, agent_ids)
+
+
+@pytest.mark.asyncio
+async def test_create_channel_memberships_with_orphan_channel_id(
+    db_session: AsyncSession,
+) -> None:
+    """Channel slug pointing at a non-existent UUID hits lines 163/183."""
+
+    agent_ids = await create_agents(db_session)
+    # Real channels exist; we lie about their UUID so _get_channel returns None.
+    bogus = {slug: str(uuid4()) for slug in ("backend-cell", "announcements")}
+    await create_channel_memberships(db_session, bogus, agent_ids)
+
+
+@pytest.mark.asyncio
+async def test_bootstrap_database_invokes_full_pipeline() -> None:
+    """bootstrap_database wires up init_db + seeding (lines 284-298)."""
+    fake_session = AsyncMock()
+    fake_session.commit = AsyncMock()
+
+    @asynccontextmanager
+    async def _ctx():
+        yield fake_session
+
+    with (
+        patch("roboco.db.seed.init_db", AsyncMock()) as mock_init,
+        patch("roboco.db.seed.get_db_context", _ctx),
+        patch(
+            "roboco.db.seed.create_channels",
+            AsyncMock(return_value={"backend-cell": "id"}),
+        ) as mock_ch,
+        patch(
+            "roboco.db.seed.create_agents",
+            AsyncMock(return_value={"be-dev-1": "id"}),
+        ) as mock_ag,
+        patch("roboco.db.seed.create_channel_memberships", AsyncMock()) as mock_mem,
+        patch("roboco.db.seed.create_initial_messages", AsyncMock()) as mock_msg,
+    ):
+        await bootstrap_database()
+    mock_init.assert_awaited_once()
+    mock_ch.assert_awaited_once()
+    mock_ag.assert_awaited_once()
+    mock_mem.assert_awaited_once()
+    mock_msg.assert_awaited_once()
+    fake_session.commit.assert_awaited_once()