awoooi/apps/api/tests/test_knowledge_service_readback_degraded.py

import pytest

from src.services import knowledge_service as knowledge_service_module
from src.services.knowledge_service import KnowledgeService


class _BrokenDbContext:
    async def __aenter__(self):
        raise RuntimeError("db pool exhausted")

    async def __aexit__(self, exc_type, exc, tb):
        return False


@pytest.mark.asyncio
async def test_knowledge_list_entries_fails_soft_when_readback_breaks(monkeypatch) -> None:
    monkeypatch.setattr(
        knowledge_service_module,
        "get_db_context",
        lambda: _BrokenDbContext(),
    )
    service = KnowledgeService.__new__(KnowledgeService)

    response = await service.list_entries(limit=50)

    assert response.total == 13
    assert len(response.items) == 13
    assert response.items[0].id == "source-backed-project-awoooi"
    assert response.items[0].source == "ai_extracted"
    assert response.items[0].status == "approved"
    assert [row.category for row in response.categories] == [
        "project",
        "product",
        "website",
        "service",
        "package",
        "tool",
        "log",
        "alert",
        "playbook",
        "rag",
        "mcp",
        "schedule",
        "general",
    ]
    assert all(row.count == 1 for row in response.categories)
    assert [row.key for row in response.asset_taxonomy] == [
        "project",
        "product",
        "website",
        "service",
        "package",
        "tool",
        "log",
        "alert",
        "playbook",
        "rag",
        "mcp",
        "schedule",
    ]
    assert all(row.count >= 1 for row in response.asset_taxonomy)
    assert response.readback_status == "source_backed_degraded"
    assert response.operator_stage == "knowledge_readback_source_backed_ai_controlled_repair"
    assert response.next_step == "repair_primary_km_db_readback_then_promote_source_backed_receipts_to_persistent_km"
    assert response.writes_on_read is False
    assert response.manual_review_required is False


@pytest.mark.asyncio
async def test_knowledge_list_entries_source_backed_filter_and_search(monkeypatch) -> None:
    monkeypatch.setattr(
        knowledge_service_module,
        "get_db_context",
        lambda: _BrokenDbContext(),
    )
    service = KnowledgeService.__new__(KnowledgeService)

    response = await service.list_entries(category="alert", q="Telegram", limit=50)

    assert response.total == 1
    assert response.items[0].id == "source-backed-alert-telegram-monitoring-coverage"
    assert response.categories[[row.category for row in response.categories].index("alert")].count == 1
    assert response.asset_taxonomy[[row.key for row in response.asset_taxonomy].index("alert")].count == 1


@pytest.mark.asyncio
async def test_knowledge_categories_fails_soft_when_readback_breaks(monkeypatch) -> None:
    monkeypatch.setattr(
        knowledge_service_module,
        "get_db_context",
        lambda: _BrokenDbContext(),
    )
    service = KnowledgeService.__new__(KnowledgeService)

    categories = await service.get_categories()

    assert [row.category for row in categories] == [
        "project",
        "product",
        "website",
        "service",
        "package",
        "tool",
        "log",
        "alert",
        "playbook",
        "rag",
        "mcp",
        "schedule",
        "general",
    ]
    assert all(row.count == 1 for row in categories)


@pytest.mark.asyncio
async def test_knowledge_asset_taxonomy_fails_soft_when_readback_breaks(monkeypatch) -> None:
    monkeypatch.setattr(
        knowledge_service_module,
        "get_db_context",
        lambda: _BrokenDbContext(),
    )
    service = KnowledgeService.__new__(KnowledgeService)

    taxonomy = await service.get_asset_taxonomy()

    assert [row.key for row in taxonomy] == [
        "project",
        "product",
        "website",
        "service",
        "package",
        "tool",
        "log",
        "alert",
        "playbook",
        "rag",
        "mcp",
        "schedule",
    ]
    assert all(row.count >= 1 for row in taxonomy)