Initial public release

tests/test_supplemental_events.py

@@ -0,0 +1,59 @@
+# tests/test_supplemental_events.py
+import asyncio
+import json
+import pytest
+from unittest.mock import AsyncMock, MagicMock, patch
+from hydra.supplemental.events import EventCalendarPoller
+
+
+@pytest.mark.asyncio
+async def test_fetch_returns_empty_without_api_key():
+    r = MagicMock()
+    poller = EventCalendarPoller(redis_client=r, api_key="")
+    result = await poller._fetch()
+    assert result == []
+
+
+@pytest.mark.asyncio
+async def test_fetch_parses_response():
+    r = MagicMock()
+    poller = EventCalendarPoller(redis_client=r, api_key="test_key")
+    mock_response = {
+        "body": [
+            {
+                "title": "Bitcoin Halving",
+                "coins": [{"symbol": "BTC"}],
+                "date_event": "2024-04-20T00:00:00Z",
+            }
+        ]
+    }
+    mock_resp = MagicMock()
+    mock_resp.json.return_value = mock_response
+    mock_resp.raise_for_status = MagicMock()
+    mock_http = AsyncMock()
+    mock_http.get = AsyncMock(return_value=mock_resp)
+    with patch("httpx.AsyncClient") as mock_client_cls:
+        mock_client_cls.return_value.__aenter__ = AsyncMock(return_value=mock_http)
+        mock_client_cls.return_value.__aexit__ = AsyncMock(return_value=False)
+        result = await poller._fetch()
+    assert len(result) == 1
+    assert result[0]["title"] == "Bitcoin Halving"
+    assert result[0]["symbol"] == "BTC"
+    assert result[0]["source"] == "coinmarketcal"
+
+
+@pytest.mark.asyncio
+async def test_run_writes_redis():
+    r = MagicMock()
+    r.set = AsyncMock()
+    poller = EventCalendarPoller(redis_client=r, api_key="")
+    with patch("hydra.supplemental.events.asyncio.sleep",
+               side_effect=asyncio.CancelledError):
+        try:
+            await poller.run()
+        except asyncio.CancelledError:
+            pass
+    r.set.assert_called_once()
+    key, value = r.set.call_args[0]
+    assert key == "hydra:events:upcoming"
+    assert json.loads(value) == []