summaryrefslogtreecommitdiff
path: root/services/news-collector/tests/test_main.py
diff options
context:
space:
mode:
Diffstat (limited to 'services/news-collector/tests/test_main.py')
-rw-r--r--services/news-collector/tests/test_main.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/services/news-collector/tests/test_main.py b/services/news-collector/tests/test_main.py
index 3ebb094..66190dc 100644
--- a/services/news-collector/tests/test_main.py
+++ b/services/news-collector/tests/test_main.py
@@ -1,5 +1,5 @@
"""Tests for news collector scheduler."""
-import pytest
+
from unittest.mock import AsyncMock, MagicMock
from datetime import datetime, timezone
from shared.models import NewsCategory, NewsItem
@@ -8,9 +8,11 @@ from news_collector.main import run_collector_once
async def test_run_collector_once_stores_and_publishes():
mock_item = NewsItem(
- source="test", headline="Test news",
+ source="test",
+ headline="Test news",
published_at=datetime(2026, 4, 2, tzinfo=timezone.utc),
- sentiment=0.5, category=NewsCategory.MACRO,
+ sentiment=0.5,
+ category=NewsCategory.MACRO,
)
mock_collector = MagicMock()
mock_collector.name = "test"