Create rag-client package

tests/conftest.py

@@ -0,0 +1,74 @@
+"""Shared pytest fixtures and configuration.
+
+Environment variables for testing are configured in pytest.ini via pytest-env plugin.
+"""
+
+from collections.abc import AsyncGenerator
+
+import pytest
+import pytest_asyncio
+
+from capa_de_integracion.config import Settings
+from capa_de_integracion.services.firestore_service import FirestoreService
+
+# Configure pytest-asyncio
+pytest_plugins = ("pytest_asyncio",)
+
+
+@pytest.fixture(scope="session")
+def emulator_settings() -> Settings:
+    """Create settings configured for Firestore emulator."""
+    # Settings will be loaded from environment variables set at module level
+    return Settings.model_validate({})
+
+
+@pytest_asyncio.fixture
+async def firestore_service(
+    emulator_settings: Settings,
+) -> AsyncGenerator[FirestoreService, None]:
+    """Create FirestoreService instance connected to emulator."""
+    service = FirestoreService(emulator_settings)
+
+    yield service
+
+    # Cleanup: Close the service
+    await service.close()
+
+
+@pytest_asyncio.fixture
+async def clean_firestore(firestore_service: FirestoreService) -> AsyncGenerator[FirestoreService, None]:
+    """Provide a clean Firestore service and cleanup after test."""
+    # Cleanup before test
+    await _cleanup_collections(firestore_service)
+
+    yield firestore_service
+
+    # Cleanup after test
+    await _cleanup_collections(firestore_service)
+
+
+async def _cleanup_collections(service: FirestoreService) -> None:
+    """Delete all documents from test collections."""
+    # Clean conversations collection
+    conversations_ref = service.db.collection(service.conversations_collection)
+    async for doc in conversations_ref.stream():
+        # Delete subcollection entries first
+        entries_ref = doc.reference.collection(service.entries_subcollection)
+        async for entry_doc in entries_ref.stream():
+            await entry_doc.reference.delete()
+        # Delete session document
+        await doc.reference.delete()
+
+    # Clean notifications collection
+    notifications_ref = service.db.collection(service.notifications_collection)
+    async for doc in notifications_ref.stream():
+        await doc.reference.delete()
+
+
+def pytest_recording_configure(config, vcr):
+    """Configure pytest-recording for Firestore emulator."""
+    # Don't filter requests to the emulator
+    vcr.ignore_hosts = []
+
+    # Match on method and path for emulator requests
+    vcr.match_on = ["method", "path", "body"]