"""
E2E tests for data explorer API.
"""
from datetime import date
from fastapi.testclient import TestClient
from sqlalchemy.orm import Session

from app.models.stock import Stock, ETF, Price, ETFPrice, Sector, Valuation, AssetClass


def _seed_stock(db: Session):
    """Add test stock data."""
    stock = Stock(
        ticker="005930", name="삼성전자", market="KOSPI",
        close_price=70000, market_cap=400000000000000,
        stock_type="common", base_date=date(2025, 1, 1),
    )
    db.add(stock)
    db.add(Price(ticker="005930", date=date(2025, 1, 2), open=69000, high=71000, low=68500, close=70000, volume=10000000))
    db.add(Price(ticker="005930", date=date(2025, 1, 3), open=70000, high=72000, low=69000, close=71000, volume=12000000))
    db.commit()


def _seed_etf(db: Session):
    """Add test ETF data."""
    etf = ETF(ticker="069500", name="TIGER 200", asset_class=AssetClass.EQUITY, market="ETF")
    db.add(etf)
    db.add(ETFPrice(ticker="069500", date=date(2025, 1, 2), close=43000, volume=500000))
    db.add(ETFPrice(ticker="069500", date=date(2025, 1, 3), close=43500, volume=600000))
    db.commit()


def _seed_sector(db: Session):
    db.add(Sector(ticker="005930", sector_code="G45", company_name="삼성전자", sector_name="반도체", base_date=date(2025, 1, 1)))
    db.commit()


def _seed_valuation(db: Session):
    db.add(Valuation(ticker="005930", base_date=date(2025, 1, 1), per=12.5, pbr=1.3, dividend_yield=2.1))
    db.commit()


def test_list_stocks(client: TestClient, auth_headers, db: Session):
    _seed_stock(db)
    resp = client.get("/api/data/stocks", headers=auth_headers)
    assert resp.status_code == 200
    data = resp.json()
    assert data["total"] >= 1
    assert len(data["items"]) >= 1
    assert data["items"][0]["ticker"] == "005930"


def test_list_stocks_search(client: TestClient, auth_headers, db: Session):
    _seed_stock(db)
    resp = client.get("/api/data/stocks?search=삼성", headers=auth_headers)
    assert resp.status_code == 200
    assert resp.json()["total"] >= 1


def test_stock_prices(client: TestClient, auth_headers, db: Session):
    _seed_stock(db)
    resp = client.get("/api/data/stocks/005930/prices", headers=auth_headers)
    assert resp.status_code == 200
    data = resp.json()
    assert len(data) == 2


def test_list_etfs(client: TestClient, auth_headers, db: Session):
    _seed_etf(db)
    resp = client.get("/api/data/etfs", headers=auth_headers)
    assert resp.status_code == 200
    assert resp.json()["total"] >= 1


def test_etf_prices(client: TestClient, auth_headers, db: Session):
    _seed_etf(db)
    resp = client.get("/api/data/etfs/069500/prices", headers=auth_headers)
    assert resp.status_code == 200
    assert len(resp.json()) == 2


def test_list_sectors(client: TestClient, auth_headers, db: Session):
    _seed_sector(db)
    resp = client.get("/api/data/sectors", headers=auth_headers)
    assert resp.status_code == 200
    assert resp.json()["total"] >= 1


def test_list_valuations(client: TestClient, auth_headers, db: Session):
    _seed_valuation(db)
    resp = client.get("/api/data/valuations", headers=auth_headers)
    assert resp.status_code == 200
    assert resp.json()["total"] >= 1