"""Tests for news-related SQLAlchemy models."""

from shared.sa_models import MarketSentimentRow, NewsItemRow, StockSelectionRow, SymbolScoreRow


def test_news_item_row_tablename():
    assert NewsItemRow.__tablename__ == "news_items"


def test_symbol_score_row_tablename():
    assert SymbolScoreRow.__tablename__ == "symbol_scores"


def test_market_sentiment_row_tablename():
    assert MarketSentimentRow.__tablename__ == "market_sentiment"


def test_stock_selection_row_tablename():
    assert StockSelectionRow.__tablename__ == "stock_selections"


def test_news_item_row_columns():
    cols = {c.name for c in NewsItemRow.__table__.columns}
    assert cols >= {"id", "source", "headline", "published_at", "sentiment", "category"}


def test_symbol_score_row_columns():
    cols = {c.name for c in SymbolScoreRow.__table__.columns}
    assert cols >= {"id", "symbol", "news_score", "composite", "updated_at"}