# -*- coding: utf-8 -*-
"""Tests for data_tools preferring split services when configured."""

from backend.tools import data_tools
from shared.schema import CompanyNews, FinancialMetrics, InsiderTrade, LineItem, Price


def test_data_tools_prefers_trading_service(monkeypatch):
    monkeypatch.setenv("TRADING_SERVICE_URL", "http://localhost:8001")
    monkeypatch.setenv("SERVICE_NAME", "agent_service")
    monkeypatch.setattr(data_tools._cache, "get_prices", lambda key: None)
    monkeypatch.setattr(data_tools._cache, "get_financial_metrics", lambda key: None)
    monkeypatch.setattr(data_tools._cache, "get_insider_trades", lambda key: None)
    monkeypatch.setattr(data_tools._cache, "get_company_news", lambda key: None)

    def fake_service_get_json(base_url, path, *, params):
        if path == "/api/prices":
            return {
                "ticker": "AAPL",
                "prices": [
                    Price(
                        open=1,
                        close=2,
                        high=3,
                        low=1,
                        volume=10,
                        time="2026-03-16",
                    ).model_dump()
                ],
            }
        if path == "/api/financials":
            return {
                "financial_metrics": [
                    FinancialMetrics(
                        ticker="AAPL",
                        report_period="2026-03-16",
                        period="ttm",
                        currency="USD",
                        market_cap=123.0,
                        enterprise_value=None,
                        price_to_earnings_ratio=None,
                        price_to_book_ratio=None,
                        price_to_sales_ratio=None,
                        enterprise_value_to_ebitda_ratio=None,
                        enterprise_value_to_revenue_ratio=None,
                        free_cash_flow_yield=None,
                        peg_ratio=None,
                        gross_margin=None,
                        operating_margin=None,
                        net_margin=None,
                        return_on_equity=None,
                        return_on_assets=None,
                        return_on_invested_capital=None,
                        asset_turnover=None,
                        inventory_turnover=None,
                        receivables_turnover=None,
                        days_sales_outstanding=None,
                        operating_cycle=None,
                        working_capital_turnover=None,
                        current_ratio=None,
                        quick_ratio=None,
                        cash_ratio=None,
                        operating_cash_flow_ratio=None,
                        debt_to_equity=None,
                        debt_to_assets=None,
                        interest_coverage=None,
                        revenue_growth=None,
                        earnings_growth=None,
                        book_value_growth=None,
                        earnings_per_share_growth=None,
                        free_cash_flow_growth=None,
                        operating_income_growth=None,
                        ebitda_growth=None,
                        payout_ratio=None,
                        earnings_per_share=None,
                        book_value_per_share=None,
                        free_cash_flow_per_share=None,
                    ).model_dump()
                ]
            }
        if path == "/api/insider-trades":
            return {
                "insider_trades": [
                    InsiderTrade(ticker="AAPL", filing_date="2026-03-16").model_dump()
                ]
            }
        if path == "/api/news":
            return {
                "news": [
                    CompanyNews(
                        ticker="AAPL",
                        title="Title",
                        source="polygon",
                        url="https://example.com",
                    ).model_dump()
                ]
            }
        if path == "/api/market-cap":
            return {"ticker": "AAPL", "end_date": "2026-03-16", "market_cap": 2.5e12}
        if path == "/api/line-items":
            return {
                "search_results": [
                    LineItem(
                        ticker="AAPL",
                        report_period="2026-03-16",
                        period="ttm",
                        currency="USD",
                        free_cash_flow=321.0,
                    ).model_dump()
                ]
            }
        raise AssertionError(path)

    monkeypatch.setattr(data_tools, "_service_get_json", fake_service_get_json)

    prices = data_tools.get_prices("AAPL", "2026-03-01", "2026-03-16")
    metrics = data_tools.get_financial_metrics("AAPL", "2026-03-16")
    trades = data_tools.get_insider_trades("AAPL", "2026-03-16")
    news = data_tools.get_company_news("AAPL", "2026-03-16")
    market_cap = data_tools.get_market_cap("AAPL", "2026-03-16")
    line_items = data_tools.search_line_items(
        "AAPL",
        ["free_cash_flow"],
        "2026-03-16",
    )

    assert prices[0].close == 2
    assert metrics[0].ticker == "AAPL"
    assert trades[0].ticker == "AAPL"
    assert news[0].ticker == "AAPL"
    assert market_cap == 2.5e12
    assert line_items[0].free_cash_flow == 321.0


def test_data_tools_skips_self_recursion_for_trading_service(monkeypatch):
    monkeypatch.setenv("TRADING_SERVICE_URL", "http://localhost:8001")
    monkeypatch.setenv("SERVICE_NAME", "trading_service")

    assert data_tools._trading_service_url() is None