summaryrefslogtreecommitdiff
path: root/services/news-collector/tests
diff options
context:
space:
mode:
authorTheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com>2026-04-02 15:54:55 +0900
committerTheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com>2026-04-02 15:54:55 +0900
commitbf4afbc0a3cc4e847ef01840365fd6a6ae9c142f (patch)
treec8634b3b21534f550e2d255d98c4a068a1b567d0 /services/news-collector/tests
parentec8b6fea5a4a710df4b2ae18f3f399d165c8ffd4 (diff)
style: auto-fix lint violations from enhanced ruff rules
Diffstat (limited to 'services/news-collector/tests')
-rw-r--r--services/news-collector/tests/test_fear_greed.py2
-rw-r--r--services/news-collector/tests/test_fed.py3
-rw-r--r--services/news-collector/tests/test_finnhub.py2
-rw-r--r--services/news-collector/tests/test_main.py8
-rw-r--r--services/news-collector/tests/test_reddit.py3
-rw-r--r--services/news-collector/tests/test_rss.py2
-rw-r--r--services/news-collector/tests/test_sec_edgar.py8
-rw-r--r--services/news-collector/tests/test_truth_social.py3
8 files changed, 18 insertions, 13 deletions
diff --git a/services/news-collector/tests/test_fear_greed.py b/services/news-collector/tests/test_fear_greed.py
index d483aa6..e8bd8f0 100644
--- a/services/news-collector/tests/test_fear_greed.py
+++ b/services/news-collector/tests/test_fear_greed.py
@@ -1,8 +1,8 @@
"""Tests for CNN Fear & Greed Index collector."""
-import pytest
from unittest.mock import AsyncMock, patch
+import pytest
from news_collector.collectors.fear_greed import FearGreedCollector
diff --git a/services/news-collector/tests/test_fed.py b/services/news-collector/tests/test_fed.py
index d1a736b..7f1c46c 100644
--- a/services/news-collector/tests/test_fed.py
+++ b/services/news-collector/tests/test_fed.py
@@ -1,7 +1,8 @@
"""Tests for Federal Reserve collector."""
-import pytest
from unittest.mock import AsyncMock, patch
+
+import pytest
from news_collector.collectors.fed import FedCollector
diff --git a/services/news-collector/tests/test_finnhub.py b/services/news-collector/tests/test_finnhub.py
index a4cf169..3af65b8 100644
--- a/services/news-collector/tests/test_finnhub.py
+++ b/services/news-collector/tests/test_finnhub.py
@@ -1,8 +1,8 @@
"""Tests for Finnhub news collector."""
-import pytest
from unittest.mock import AsyncMock, patch
+import pytest
from news_collector.collectors.finnhub import FinnhubCollector
diff --git a/services/news-collector/tests/test_main.py b/services/news-collector/tests/test_main.py
index 66190dc..f85569a 100644
--- a/services/news-collector/tests/test_main.py
+++ b/services/news-collector/tests/test_main.py
@@ -1,16 +1,18 @@
"""Tests for news collector scheduler."""
+from datetime import UTC, datetime
from unittest.mock import AsyncMock, MagicMock
-from datetime import datetime, timezone
-from shared.models import NewsCategory, NewsItem
+
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=timezone.utc),
+ published_at=datetime(2026, 4, 2, tzinfo=UTC),
sentiment=0.5,
category=NewsCategory.MACRO,
)
diff --git a/services/news-collector/tests/test_reddit.py b/services/news-collector/tests/test_reddit.py
index 440b173..31b1dc1 100644
--- a/services/news-collector/tests/test_reddit.py
+++ b/services/news-collector/tests/test_reddit.py
@@ -1,7 +1,8 @@
"""Tests for Reddit collector."""
-import pytest
from unittest.mock import AsyncMock, patch
+
+import pytest
from news_collector.collectors.reddit import RedditCollector
diff --git a/services/news-collector/tests/test_rss.py b/services/news-collector/tests/test_rss.py
index e03250a..7242c75 100644
--- a/services/news-collector/tests/test_rss.py
+++ b/services/news-collector/tests/test_rss.py
@@ -1,8 +1,8 @@
"""Tests for RSS news collector."""
-import pytest
from unittest.mock import AsyncMock, patch
+import pytest
from news_collector.collectors.rss import RSSCollector
diff --git a/services/news-collector/tests/test_sec_edgar.py b/services/news-collector/tests/test_sec_edgar.py
index 5d4f69f..b0faf18 100644
--- a/services/news-collector/tests/test_sec_edgar.py
+++ b/services/news-collector/tests/test_sec_edgar.py
@@ -1,9 +1,9 @@
"""Tests for SEC EDGAR filing collector."""
-import pytest
-from datetime import datetime, timezone
-from unittest.mock import AsyncMock, patch, MagicMock
+from datetime import UTC, datetime
+from unittest.mock import AsyncMock, MagicMock, patch
+import pytest
from news_collector.collectors.sec_edgar import SecEdgarCollector
@@ -37,7 +37,7 @@ async def test_collect_parses_filings(collector):
}
mock_datetime = MagicMock(spec=datetime)
- mock_datetime.now.return_value = datetime(2026, 4, 2, tzinfo=timezone.utc)
+ mock_datetime.now.return_value = datetime(2026, 4, 2, tzinfo=UTC)
mock_datetime.strptime = datetime.strptime
with patch.object(
diff --git a/services/news-collector/tests/test_truth_social.py b/services/news-collector/tests/test_truth_social.py
index 91ddb9d..52f1e46 100644
--- a/services/news-collector/tests/test_truth_social.py
+++ b/services/news-collector/tests/test_truth_social.py
@@ -1,7 +1,8 @@
"""Tests for Truth Social collector."""
-import pytest
from unittest.mock import AsyncMock, patch
+
+import pytest
from news_collector.collectors.truth_social import TruthSocialCollector