import unittest
from unittest.mock import AsyncMock

from rag.events import publish_knowledge_update


class KnowledgeEventTests(unittest.IsolatedAsyncioTestCase):
    async def test_publishes_update_event_after_successful_ingestion(self):
        publisher = AsyncMock()

        await publish_knowledge_update(
            publisher, doc_id="doc-1", collection_name="fin_faq", chunk_count=4
        )

        publisher.assert_awaited_once_with(
            "event:knowledge_update",
            {"doc_id": "doc-1", "collection_name": "fin_faq", "chunk_count": 4},
        )


if __name__ == "__main__":
    unittest.main()