9 files changed, 406 insertions, 0 deletions
diff --git a/services/news-collector/tests/__init__.py b/services/news-collector/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/services/news-collector/tests/__init__.py
diff --git a/services/news-collector/tests/test_fear_greed.py b/services/news-collector/tests/test_fear_greed.py
new file mode 100644
index 0000000..e8bd8f0
--- /dev/null
+++ b/services/news-collector/tests/test_fear_greed.py
@@ -0,0 +1,49 @@
+"""Tests for CNN Fear & Greed Index collector."""
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from news_collector.collectors.fear_greed import FearGreedCollector
+
+
+@pytest.fixture
+def collector():
+    return FearGreedCollector()
+
+
+def test_collector_name(collector):
+    assert collector.name == "fear_greed"
+    assert collector.poll_interval == 3600
+
+
+async def test_is_available(collector):
+    assert await collector.is_available() is True
+
+
+async def test_collect_parses_api_response(collector):
+    mock_data = {
+        "fear_and_greed": {
+            "score": 45.0,
+            "rating": "Fear",
+            "timestamp": "2026-04-02T12:00:00+00:00",
+        }
+    }
+    with patch.object(collector, "_fetch_index", new_callable=AsyncMock, return_value=mock_data):
+        result = await collector.collect()
+    assert result.fear_greed == 45
+    assert result.fear_greed_label == "Fear"
+
+
+async def test_collect_returns_none_on_failure(collector):
+    with patch.object(collector, "_fetch_index", new_callable=AsyncMock, return_value=None):
+        result = await collector.collect()
+    assert result is None
+
+
+def test_classify_label():
+    c = FearGreedCollector()
+    assert c._classify(10) == "Extreme Fear"
+    assert c._classify(30) == "Fear"
+    assert c._classify(50) == "Neutral"
+    assert c._classify(70) == "Greed"
+    assert c._classify(85) == "Extreme Greed"
diff --git a/services/news-collector/tests/test_fed.py b/services/news-collector/tests/test_fed.py
new file mode 100644
index 0000000..7f1c46c
--- /dev/null
+++ b/services/news-collector/tests/test_fed.py
@@ -0,0 +1,38 @@
+"""Tests for Federal Reserve collector."""
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from news_collector.collectors.fed import FedCollector
+
+
+@pytest.fixture
+def collector():
+    return FedCollector()
+
+
+def test_collector_name(collector):
+    assert collector.name == "fed"
+    assert collector.poll_interval == 3600
+
+
+async def test_is_available(collector):
+    assert await collector.is_available() is True
+
+
+async def test_collect_parses_rss(collector):
+    mock_entries = [
+        {
+            "title": "Federal Reserve issues FOMC statement",
+            "link": "https://www.federalreserve.gov/newsevents/pressreleases/monetary20260402a.htm",
+            "published_parsed": (2026, 4, 2, 14, 0, 0, 0, 0, 0),
+            "summary": "The Federal Open Market Committee decided to maintain the target range...",
+        },
+    ]
+    with patch.object(
+        collector, "_fetch_fed_rss", new_callable=AsyncMock, return_value=mock_entries
+    ):
+        items = await collector.collect()
+    assert len(items) == 1
+    assert items[0].source == "fed"
+    assert items[0].category.value == "fed"
diff --git a/services/news-collector/tests/test_finnhub.py b/services/news-collector/tests/test_finnhub.py
new file mode 100644
index 0000000..3af65b8
--- /dev/null
+++ b/services/news-collector/tests/test_finnhub.py
@@ -0,0 +1,67 @@
+"""Tests for Finnhub news collector."""
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from news_collector.collectors.finnhub import FinnhubCollector
+
+
+@pytest.fixture
+def collector():
+    return FinnhubCollector(api_key="test_key")
+
+
+def test_collector_name(collector):
+    assert collector.name == "finnhub"
+    assert collector.poll_interval == 300
+
+
+async def test_is_available_with_key(collector):
+    assert await collector.is_available() is True
+
+
+async def test_is_available_without_key():
+    c = FinnhubCollector(api_key="")
+    assert await c.is_available() is False
+
+
+async def test_collect_parses_response(collector):
+    mock_response = [
+        {
+            "category": "top news",
+            "datetime": 1711929600,
+            "headline": "AAPL beats earnings",
+            "id": 12345,
+            "related": "AAPL",
+            "source": "MarketWatch",
+            "summary": "Apple reported better than expected...",
+            "url": "https://example.com/article",
+        },
+        {
+            "category": "top news",
+            "datetime": 1711929000,
+            "headline": "Fed holds rates steady",
+            "id": 12346,
+            "related": "",
+            "source": "Reuters",
+            "summary": "The Federal Reserve...",
+            "url": "https://example.com/fed",
+        },
+    ]
+
+    with patch.object(collector, "_fetch_news", new_callable=AsyncMock, return_value=mock_response):
+        items = await collector.collect()
+
+    assert len(items) == 2
+    assert items[0].source == "finnhub"
+    assert items[0].headline == "AAPL beats earnings"
+    assert items[0].symbols == ["AAPL"]
+    assert items[0].url == "https://example.com/article"
+    assert isinstance(items[0].sentiment, float)
+    assert items[1].symbols == []
+
+
+async def test_collect_handles_empty_response(collector):
+    with patch.object(collector, "_fetch_news", new_callable=AsyncMock, return_value=[]):
+        items = await collector.collect()
+    assert items == []
diff --git a/services/news-collector/tests/test_main.py b/services/news-collector/tests/test_main.py
new file mode 100644
index 0000000..f85569a
--- /dev/null
+++ b/services/news-collector/tests/test_main.py
@@ -0,0 +1,41 @@
+"""Tests for news collector scheduler."""
+
+from datetime import UTC, datetime
+from unittest.mock import AsyncMock, MagicMock
+
+from news_collector.main import run_collector_once
+
+from shared.models import NewsCategory, NewsItem
+
+
+async def test_run_collector_once_stores_and_publishes():
+    mock_item = NewsItem(
+        source="test",
+        headline="Test news",
+        published_at=datetime(2026, 4, 2, tzinfo=UTC),
+        sentiment=0.5,
+        category=NewsCategory.MACRO,
+    )
+    mock_collector = MagicMock()
+    mock_collector.name = "test"
+    mock_collector.collect = AsyncMock(return_value=[mock_item])
+    mock_db = MagicMock()
+    mock_db.insert_news_item = AsyncMock()
+    mock_broker = MagicMock()
+    mock_broker.publish = AsyncMock()
+
+    count = await run_collector_once(mock_collector, mock_db, mock_broker)
+    assert count == 1
+    mock_db.insert_news_item.assert_called_once_with(mock_item)
+    mock_broker.publish.assert_called_once()
+
+
+async def test_run_collector_once_handles_empty():
+    mock_collector = MagicMock()
+    mock_collector.name = "test"
+    mock_collector.collect = AsyncMock(return_value=[])
+    mock_db = MagicMock()
+    mock_broker = MagicMock()
+
+    count = await run_collector_once(mock_collector, mock_db, mock_broker)
+    assert count == 0
diff --git a/services/news-collector/tests/test_reddit.py b/services/news-collector/tests/test_reddit.py
new file mode 100644
index 0000000..31b1dc1
--- /dev/null
+++ b/services/news-collector/tests/test_reddit.py
@@ -0,0 +1,64 @@
+"""Tests for Reddit collector."""
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from news_collector.collectors.reddit import RedditCollector
+
+
+@pytest.fixture
+def collector():
+    return RedditCollector()
+
+
+def test_collector_name(collector):
+    assert collector.name == "reddit"
+    assert collector.poll_interval == 900
+
+
+async def test_is_available(collector):
+    assert await collector.is_available() is True
+
+
+async def test_collect_parses_posts(collector):
+    mock_posts = [
+        {
+            "data": {
+                "title": "NVDA to the moon! AI demand is insane",
+                "selftext": "Just loaded up on NVDA calls",
+                "url": "https://reddit.com/r/wallstreetbets/123",
+                "created_utc": 1711929600,
+                "score": 500,
+                "num_comments": 200,
+                "subreddit": "wallstreetbets",
+            }
+        },
+    ]
+    with patch.object(
+        collector, "_fetch_subreddit", new_callable=AsyncMock, return_value=mock_posts
+    ):
+        items = await collector.collect()
+    assert len(items) >= 1
+    assert items[0].source == "reddit"
+    assert items[0].category.value == "social"
+
+
+async def test_collect_filters_low_score(collector):
+    mock_posts = [
+        {
+            "data": {
+                "title": "Random question",
+                "selftext": "",
+                "url": "https://reddit.com/456",
+                "created_utc": 1711929600,
+                "score": 3,
+                "num_comments": 1,
+                "subreddit": "stocks",
+            }
+        },
+    ]
+    with patch.object(
+        collector, "_fetch_subreddit", new_callable=AsyncMock, return_value=mock_posts
+    ):
+        items = await collector.collect()
+    assert items == []
diff --git a/services/news-collector/tests/test_rss.py b/services/news-collector/tests/test_rss.py
new file mode 100644
index 0000000..7242c75
--- /dev/null
+++ b/services/news-collector/tests/test_rss.py
@@ -0,0 +1,47 @@
+"""Tests for RSS news collector."""
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from news_collector.collectors.rss import RSSCollector
+
+
+@pytest.fixture
+def collector():
+    return RSSCollector()
+
+
+def test_collector_name(collector):
+    assert collector.name == "rss"
+    assert collector.poll_interval == 600
+
+
+async def test_is_available(collector):
+    assert await collector.is_available() is True
+
+
+async def test_collect_parses_feed(collector):
+    mock_feed = {
+        "entries": [
+            {
+                "title": "NVDA surges on AI demand",
+                "link": "https://example.com/nvda",
+                "published_parsed": (2026, 4, 2, 12, 0, 0, 0, 0, 0),
+                "summary": "Nvidia stock jumped 5%...",
+            },
+            {
+                "title": "Markets rally on jobs data",
+                "link": "https://example.com/market",
+                "published_parsed": (2026, 4, 2, 11, 0, 0, 0, 0, 0),
+                "summary": "The S&P 500 rose...",
+            },
+        ],
+    }
+
+    with patch.object(collector, "_fetch_feeds", new_callable=AsyncMock, return_value=[mock_feed]):
+        items = await collector.collect()
+
+    assert len(items) == 2
+    assert items[0].source == "rss"
+    assert items[0].headline == "NVDA surges on AI demand"
+    assert isinstance(items[0].sentiment, float)
diff --git a/services/news-collector/tests/test_sec_edgar.py b/services/news-collector/tests/test_sec_edgar.py
new file mode 100644
index 0000000..b0faf18
--- /dev/null
+++ b/services/news-collector/tests/test_sec_edgar.py
@@ -0,0 +1,58 @@
+"""Tests for SEC EDGAR filing collector."""
+
+from datetime import UTC, datetime
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+from news_collector.collectors.sec_edgar import SecEdgarCollector
+
+
+@pytest.fixture
+def collector():
+    return SecEdgarCollector()
+
+
+def test_collector_name(collector):
+    assert collector.name == "sec_edgar"
+    assert collector.poll_interval == 1800
+
+
+async def test_is_available(collector):
+    assert await collector.is_available() is True
+
+
+async def test_collect_parses_filings(collector):
+    mock_response = {
+        "filings": {
+            "recent": {
+                "accessionNumber": ["0001234-26-000001"],
+                "filingDate": ["2026-04-02"],
+                "primaryDocument": ["filing.htm"],
+                "form": ["8-K"],
+                "primaryDocDescription": ["Current Report"],
+            }
+        },
+        "tickers": [{"ticker": "AAPL"}],
+        "name": "Apple Inc",
+    }
+
+    mock_datetime = MagicMock(spec=datetime)
+    mock_datetime.now.return_value = datetime(2026, 4, 2, tzinfo=UTC)
+    mock_datetime.strptime = datetime.strptime
+
+    with patch.object(
+        collector, "_fetch_recent_filings", new_callable=AsyncMock, return_value=[mock_response]
+    ):
+        with patch("news_collector.collectors.sec_edgar.datetime", mock_datetime):
+            items = await collector.collect()
+
+    assert len(items) == 1
+    assert items[0].source == "sec_edgar"
+    assert items[0].category.value == "filing"
+    assert "AAPL" in items[0].symbols
+
+
+async def test_collect_handles_empty(collector):
+    with patch.object(collector, "_fetch_recent_filings", new_callable=AsyncMock, return_value=[]):
+        items = await collector.collect()
+    assert items == []
diff --git a/services/news-collector/tests/test_truth_social.py b/services/news-collector/tests/test_truth_social.py
new file mode 100644
index 0000000..52f1e46
--- /dev/null
+++ b/services/news-collector/tests/test_truth_social.py
@@ -0,0 +1,42 @@
+"""Tests for Truth Social collector."""
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from news_collector.collectors.truth_social import TruthSocialCollector
+
+
+@pytest.fixture
+def collector():
+    return TruthSocialCollector()
+
+
+def test_collector_name(collector):
+    assert collector.name == "truth_social"
+    assert collector.poll_interval == 900
+
+
+async def test_is_available(collector):
+    assert await collector.is_available() is True
+
+
+async def test_collect_parses_posts(collector):
+    mock_posts = [
+        {
+            "content": "<p>We are imposing 25% tariffs on all steel imports!</p>",
+            "created_at": "2026-04-02T12:00:00.000Z",
+            "url": "https://truthsocial.com/@realDonaldTrump/12345",
+            "id": "12345",
+        },
+    ]
+    with patch.object(collector, "_fetch_posts", new_callable=AsyncMock, return_value=mock_posts):
+        items = await collector.collect()
+    assert len(items) == 1
+    assert items[0].source == "truth_social"
+    assert items[0].category.value == "policy"
+
+
+async def test_collect_handles_empty(collector):
+    with patch.object(collector, "_fetch_posts", new_callable=AsyncMock, return_value=[]):
+        items = await collector.collect()
+    assert items == []