Initial public release

tests/test_supplemental_sentiment.py

@@ -0,0 +1,70 @@
+# tests/test_supplemental_sentiment.py
+import asyncio
+import json
+import pytest
+from unittest.mock import AsyncMock, MagicMock, patch
+from hydra.supplemental.sentiment import SentimentPoller
+
+
+@pytest.fixture
+def mock_redis():
+    r = MagicMock()
+    r.keys = MagicMock(return_value=["hydra:indicator:binance:BTC/USDT:1m"])
+    r.set = AsyncMock()
+    return r
+
+
+def test_score_positive_headline():
+    r = MagicMock()
+    poller = SentimentPoller(redis_client=r)
+    score = poller._score(["Bitcoin is great and prices are rising strongly today!"])
+    assert score > 0.0
+
+
+def test_score_empty_headlines_returns_zero():
+    r = MagicMock()
+    poller = SentimentPoller(redis_client=r)
+    score = poller._score([])
+    assert score == 0.0
+
+
+@pytest.mark.asyncio
+async def test_fetch_news_returns_headlines():
+    r = MagicMock()
+    poller = SentimentPoller(redis_client=r, api_key="test_key")
+    mock_response = {
+        "results": [
+            {"title": "Bitcoin price rises strongly"},
+            {"title": "Market looks very bullish today"},
+        ]
+    }
+    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)
+        headlines = await poller._fetch_news("BTC")
+    assert len(headlines) == 2
+    assert "Bitcoin price rises strongly" in headlines
+
+
+@pytest.mark.asyncio
+async def test_run_writes_redis(mock_redis):
+    poller = SentimentPoller(redis_client=mock_redis)
+    poller._fetch_news = AsyncMock(return_value=["Bitcoin is surging higher today!"])
+    with patch("hydra.supplemental.sentiment.asyncio.sleep",
+               side_effect=asyncio.CancelledError):
+        try:
+            await poller.run()
+        except asyncio.CancelledError:
+            pass
+    assert mock_redis.set.call_count >= 1
+    key, value = mock_redis.set.call_args_list[0][0]
+    assert key == "hydra:sentiment:BTC"
+    data = json.loads(value)
+    assert "score" in data
+    assert "article_count" in data
+    assert data["article_count"] == 1