completed local setup with compose

tests/integration/events/test_debug.py

@@ -0,0 +1,98 @@
+import asyncio
+
+import pytest
+
+from libs.events.base import EventPayload
+from libs.events.nats_bus import NATSEventBus
+from libs.schemas.events import DocumentIngestedEventData
+
+
+@pytest.mark.asyncio
+async def test_nats_bus_class():
+    """Test NATSEventBus class within pytest."""
+
+    import time
+
+    unique_suffix = int(time.time())
+    stream_name = f"PYTEST_DEBUG_STREAM_{unique_suffix}"
+
+    print(f"\nStarting NATSEventBus with stream {stream_name}...")
+    bus = NATSEventBus(
+        servers="nats://localhost:4222",
+        stream_name=stream_name,
+        consumer_group="test-debug-group",
+    )
+
+    await bus.start()
+    print("Bus started.")
+
+    # Clean up (just in case)
+    try:
+        await bus.js.delete_stream(stream_name)
+    except Exception:
+        pass
+    await bus._ensure_stream_exists()
+
+    # Wait for stream to be ready
+    await asyncio.sleep(2)
+
+    try:
+        info = await bus.js.stream_info(stream_name)
+        print(f"Stream info: {info.config.subjects}")
+    except Exception as e:
+        print(f"Failed to get stream info: {e}")
+
+    # Setup subscriber
+    received_event = asyncio.Future()
+
+    async def handler(topic, event):
+        print(f"Handler received event: {event.event_id}")
+        if not received_event.done():
+            received_event.set_result(event)
+
+    await bus.subscribe("doc.ingested", handler)
+
+    print("Publishing message...")
+
+    data = DocumentIngestedEventData(
+        doc_id="test-doc-123",
+        filename="test.pdf",
+        mime_type="application/pdf",
+        size_bytes=1024,
+        source="upload",
+        kind="invoice",
+        storage_path="s3://test-bucket/test.pdf",
+        checksum_sha256="a" * 64,
+    )
+
+    payload = EventPayload(
+        data=data.model_dump(mode="json"),
+        actor="tester",
+        tenant_id="tenant-1",
+        schema_version="1.0",
+    )
+    payload.event_id = "evt-debug-1"
+
+    success = await bus.publish("doc.ingested", payload)
+    print(f"Published: {success}")
+
+    try:
+        result = await asyncio.wait_for(received_event, timeout=5.0)
+        print(f"Received event: {result.event_id}")
+        assert result.event_id == "evt-debug-1"
+        assert result.data["doc_id"] == "test-doc-123"
+    except TimeoutError:
+        print("Timeout waiting for event")
+        raise
+
+    await bus.stop()
+    print("Bus stopped.")
+
+    # Cleanup stream
+    try:
+        nc = await nats.connect("nats://localhost:4222")
+        js = nc.jetstream()
+        await js.delete_stream(stream_name)
+        await nc.close()
+    except Exception:
+        pass