feat: add Kurigram property notification bot

tests/test_consumer.py

@@ -0,0 +1,72 @@
+import json
+import asyncio
+
+import pytest
+from websockets.asyncio.server import serve
+
+from property_bot.config import PropertyApiConfig
+from property_bot.consumer import ListingEventConsumer
+
+
+@pytest.mark.asyncio
+async def test_consumer_ignores_service_ready_and_forwards_listing() -> None:
+    received = []
+
+    async def sink(listing: object) -> None:
+        received.append(listing)
+
+    consumer = ListingEventConsumer(
+        PropertyApiConfig("ws://localhost:3000/ws", 1, 1, 2), sink  # type: ignore[arg-type]
+    )
+    assert await consumer.handle_message(json.dumps({"type": "service.ready"})) is False
+    assert (
+        await consumer.handle_message(
+            json.dumps(
+                {
+                    "type": "listing.new",
+                    "listing": {
+                        "externalId": "abc",
+                        "url": "https://example.test/a",
+                        "title": "A",
+                        "images": [],
+                    },
+                }
+            )
+        )
+    ) is True
+    assert len(received) == 1
+    assert consumer.status.last_event_at is not None
+
+
+@pytest.mark.asyncio
+async def test_consumer_reads_listing_from_websocket_server(unused_tcp_port: int) -> None:
+    received = asyncio.Event()
+
+    async def sink(_: object) -> None:
+        received.set()
+
+    async def handler(websocket: object) -> None:
+        await websocket.send(  # type: ignore[attr-defined]
+            json.dumps(
+                {
+                    "type": "listing.new",
+                    "listing": {
+                        "externalId": "from-socket",
+                        "url": "https://example.test/a",
+                        "title": "From socket",
+                    },
+                }
+            )
+        )
+
+    async with serve(handler, "127.0.0.1", unused_tcp_port):
+        consumer = ListingEventConsumer(
+            PropertyApiConfig(f"ws://127.0.0.1:{unused_tcp_port}", 1, 0.01, 0.02), sink  # type: ignore[arg-type]
+        )
+        task = asyncio.create_task(consumer.run())
+        try:
+            await asyncio.wait_for(received.wait(), timeout=2)
+        finally:
+            task.cancel()
+            with pytest.raises(asyncio.CancelledError):
+                await task