181 lines
6.2 KiB
Python
181 lines
6.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Tests for the extracted news service app surface."""
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from backend.apps.news_service import create_app
|
|
|
|
|
|
class _FakeStore:
|
|
def get_ticker_watermarks(self, symbol):
|
|
return {"symbol": symbol, "last_news_fetch": "2026-12-31"}
|
|
|
|
def get_news_timeline_enriched(self, symbol, start_date=None, end_date=None):
|
|
return [{"date": end_date, "count": 1}]
|
|
|
|
def get_news_items(self, symbol, start_date=None, end_date=None, limit=100):
|
|
return [{"id": "news-raw-1", "ticker": symbol, "title": "Raw Title", "date": end_date}]
|
|
|
|
def get_news_items_enriched(self, symbol, start_date=None, end_date=None, trade_date=None, limit=100):
|
|
return [{"id": "news-1", "ticker": symbol, "title": "Title", "date": trade_date or end_date}]
|
|
|
|
def upsert_news_analysis(self, symbol, rows):
|
|
return len(rows)
|
|
|
|
def get_analyzed_news_ids(self, symbol, start_date=None, end_date=None):
|
|
return set()
|
|
|
|
def get_news_categories_enriched(self, symbol, start_date=None, end_date=None, limit=200):
|
|
return {"market": {"label": "market", "count": 1, "article_ids": ["news-1"]}}
|
|
|
|
def get_news_by_ids_enriched(self, symbol, article_ids):
|
|
return [{"id": article_ids[0], "ticker": symbol, "title": "Picked"}]
|
|
|
|
|
|
def test_news_service_routes_are_exposed():
|
|
app = create_app()
|
|
paths = {route.path for route in app.routes}
|
|
|
|
assert "/health" in paths
|
|
assert "/api/enriched-news" in paths
|
|
assert "/api/news-for-date" in paths
|
|
assert "/api/news-timeline" in paths
|
|
assert "/api/categories" in paths
|
|
assert "/api/similar-days" in paths
|
|
assert "/api/stories/{ticker}" in paths
|
|
assert "/api/range-explain" in paths
|
|
|
|
|
|
def test_news_service_enriched_news_and_categories(monkeypatch):
|
|
app = create_app()
|
|
app.dependency_overrides.clear()
|
|
from backend.apps import news_service as news_service_module
|
|
|
|
app.dependency_overrides[news_service_module.get_market_store] = lambda: _FakeStore()
|
|
monkeypatch.setattr(
|
|
"backend.domains.news.enrich_news_for_symbol",
|
|
lambda *args, **kwargs: {"symbol": "AAPL", "analyzed": 1},
|
|
)
|
|
|
|
with TestClient(app) as client:
|
|
news_response = client.get(
|
|
"/api/enriched-news",
|
|
params={"ticker": "AAPL", "end_date": "2026-03-23"},
|
|
)
|
|
categories_response = client.get(
|
|
"/api/categories",
|
|
params={"ticker": "AAPL", "end_date": "2026-03-23"},
|
|
)
|
|
|
|
assert news_response.status_code == 200
|
|
assert news_response.json()["news"][0]["ticker"] == "AAPL"
|
|
assert categories_response.status_code == 200
|
|
assert categories_response.json()["categories"]["market"]["count"] == 1
|
|
|
|
|
|
def test_news_service_news_for_date_and_timeline(monkeypatch):
|
|
app = create_app()
|
|
from backend.apps import news_service as news_service_module
|
|
|
|
app.dependency_overrides[news_service_module.get_market_store] = lambda: _FakeStore()
|
|
monkeypatch.setattr(
|
|
"backend.domains.news.enrich_news_for_symbol",
|
|
lambda *args, **kwargs: {"symbol": "AAPL", "analyzed": 1},
|
|
)
|
|
|
|
with TestClient(app) as client:
|
|
date_response = client.get(
|
|
"/api/news-for-date",
|
|
params={"ticker": "AAPL", "date": "2026-03-23"},
|
|
)
|
|
timeline_response = client.get(
|
|
"/api/news-timeline",
|
|
params={
|
|
"ticker": "AAPL",
|
|
"start_date": "2026-03-01",
|
|
"end_date": "2026-03-23",
|
|
},
|
|
)
|
|
|
|
assert date_response.status_code == 200
|
|
assert date_response.json()["date"] == "2026-03-23"
|
|
assert timeline_response.status_code == 200
|
|
assert timeline_response.json()["timeline"][0]["count"] == 1
|
|
|
|
|
|
def test_news_service_similar_days_and_story(monkeypatch):
|
|
app = create_app()
|
|
from backend.apps import news_service as news_service_module
|
|
|
|
app.dependency_overrides[news_service_module.get_market_store] = lambda: _FakeStore()
|
|
monkeypatch.setattr(
|
|
"backend.domains.news.enrich_news_for_symbol",
|
|
lambda *args, **kwargs: {"symbol": "AAPL", "analyzed": 1},
|
|
)
|
|
monkeypatch.setattr(
|
|
"backend.domains.news.find_similar_days",
|
|
lambda store, symbol, target_date, top_k: {
|
|
"symbol": symbol,
|
|
"target_date": target_date,
|
|
"items": [{"date": "2026-03-20", "score": 0.9}],
|
|
},
|
|
)
|
|
monkeypatch.setattr(
|
|
"backend.domains.news.get_or_create_stock_story",
|
|
lambda store, symbol, as_of_date: {
|
|
"symbol": symbol,
|
|
"as_of_date": as_of_date,
|
|
"story": "story body",
|
|
"source": "local",
|
|
},
|
|
)
|
|
|
|
with TestClient(app) as client:
|
|
similar_response = client.get(
|
|
"/api/similar-days",
|
|
params={"ticker": "AAPL", "date": "2026-03-23", "n_similar": 3},
|
|
)
|
|
story_response = client.get(
|
|
"/api/stories/AAPL",
|
|
params={"as_of_date": "2026-03-23"},
|
|
)
|
|
|
|
assert similar_response.status_code == 200
|
|
assert similar_response.json()["items"][0]["score"] == 0.9
|
|
assert story_response.status_code == 200
|
|
assert story_response.json()["story"] == "story body"
|
|
|
|
|
|
def test_news_service_range_explain(monkeypatch):
|
|
app = create_app()
|
|
from backend.apps import news_service as news_service_module
|
|
|
|
app.dependency_overrides[news_service_module.get_market_store] = lambda: _FakeStore()
|
|
monkeypatch.setattr(
|
|
"backend.domains.news.enrich_news_for_symbol",
|
|
lambda *args, **kwargs: {"symbol": "AAPL", "analyzed": 1},
|
|
)
|
|
monkeypatch.setattr(
|
|
"backend.domains.news.build_range_explanation",
|
|
lambda ticker, start_date, end_date, news_rows: {
|
|
"symbol": ticker,
|
|
"news_count": len(news_rows),
|
|
"start_date": start_date,
|
|
"end_date": end_date,
|
|
},
|
|
)
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get(
|
|
"/api/range-explain",
|
|
params={
|
|
"ticker": "AAPL",
|
|
"start_date": "2026-03-01",
|
|
"end_date": "2026-03-23",
|
|
"article_ids": ["news-7"],
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.json()["result"]["news_count"] == 1
|