"""
단위 테스트: 자율 비용 최적화 (AutonomousCostOps)
- cost_optimizer_ai 라우터 로직 검증
- 낭비 기준 상수 검증
- AI 텍스트 파싱 로직 검증
- Pydantic 스키마 검증
"""
from __future__ import annotations

import json
import pytest
from datetime import datetime


# ── 임포트 가능 여부 ──────────────────────────────────────────────────────────

def test_module_import():
    """cost_optimizer_ai 모듈이 오류 없이 임포트되는지 확인."""
    from routers import cost_optimizer_ai
    assert cost_optimizer_ai.router is not None


def test_models_import():
    """신규 ORM 모델 3개가 임포트되는지 확인."""
    from models import CostAIAnalysis, CostRecommendation, CostForecast
    assert CostAIAnalysis.__tablename__ == "tb_cost_ai_analysis"
    assert CostRecommendation.__tablename__ == "tb_cost_recommendation"
    assert CostForecast.__tablename__ == "tb_cost_forecast"


# ── 라우터 등록 확인 ──────────────────────────────────────────────────────────

def test_router_prefix():
    """라우터 prefix가 /api/cost-ai 인지 확인."""
    from routers.cost_optimizer_ai import router
    assert router.prefix == "/api/cost-ai"


def test_router_endpoints_registered():
    """8개 엔드포인트가 모두 등록되어 있는지 경로 목록으로 확인."""
    from routers.cost_optimizer_ai import router
    paths = {r.path for r in router.routes}
    assert "/api/cost-ai/analysis"          in paths
    assert "/api/cost-ai/analyze"           in paths
    assert "/api/cost-ai/forecast/{days}"   in paths
    assert "/api/cost-ai/recommendations"   in paths
    assert "/api/cost-ai/recommendations/{rec_id}/apply"  in paths
    assert "/api/cost-ai/recommendations/{rec_id}/reject" in paths
    assert "/api/cost-ai/waste"             in paths
    assert "/api/cost-ai/savings-report"    in paths


# ── 낭비 기준 상수 ────────────────────────────────────────────────────────────

def test_waste_thresholds():
    """낭비 감지 기준값이 스펙에 맞는지 확인."""
    from routers.cost_optimizer_ai import (
        _WASTE_CPU_THRESHOLD,
        _WASTE_MEM_THRESHOLD,
        _WASTE_SR_DAYS,
    )
    assert _WASTE_CPU_THRESHOLD == 10.0, "CPU 임계값은 10%"
    assert _WASTE_MEM_THRESHOLD == 20.0, "메모리 임계값은 20%"
    assert _WASTE_SR_DAYS       == 30,   "SR 미발생 기준은 30일"


# ── 절감 단가 상수 ────────────────────────────────────────────────────────────

def test_saving_unit_keys():
    """절감 단가 딕셔너리에 server, license, cloud 키가 있는지 확인."""
    from routers.cost_optimizer_ai import _SAVING_UNIT
    assert "server"  in _SAVING_UNIT
    assert "license" in _SAVING_UNIT
    assert "cloud"   in _SAVING_UNIT
    # 모두 양수
    for key, val in _SAVING_UNIT.items():
        assert val > 0, f"{key} 절감 단가는 양수여야 함"


# ── AI 텍스트 파싱 로직 ───────────────────────────────────────────────────────

@pytest.mark.asyncio
async def test_build_recommendations_from_ai_parses_numbered_list():
    """번호 목록 형식 AI 텍스트가 CostRecommendation 리스트로 변환되는지 확인."""
    from routers.cost_optimizer_ai import _build_recommendations_from_ai
    from unittest.mock import AsyncMock, MagicMock

    mock_db = MagicMock()

    ai_text = (
        "1. 유휴 서버 통합 가상화\n"
        "   CPU 사용률이 낮은 서버를 통합하세요.\n"
        "2. 미사용 라이선스 해지\n"
        "   분기마다 감사를 통해 절감하세요.\n"
        "3. 클라우드 스토리지 티어링\n"
        "   cold 티어로 전환하세요.\n"
    )
    recs = await _build_recommendations_from_ai(ai_text, mock_db)
    assert len(recs) == 3, f"3개 권고가 생성되어야 함, 실제: {len(recs)}"
    assert "유휴 서버 통합 가상화" in recs[0].title
    assert "미사용 라이선스 해지"  in recs[1].title
    assert "클라우드 스토리지 티어링" in recs[2].title


@pytest.mark.asyncio
async def test_build_recommendations_limits_to_5():
    """6개 이상 항목이 있을 때 최대 5개로 제한되는지 확인."""
    from routers.cost_optimizer_ai import _build_recommendations_from_ai
    from unittest.mock import MagicMock

    mock_db = MagicMock()

    ai_text = "\n".join(
        f"{i}. 권고 항목 {i}\n   설명 {i}." for i in range(1, 8)
    )
    recs = await _build_recommendations_from_ai(ai_text, mock_db)
    assert len(recs) <= 5, "최대 5개 제한 위반"


@pytest.mark.asyncio
async def test_build_recommendations_empty_text():
    """빈 AI 텍스트 시 빈 리스트를 반환하는지 확인."""
    from routers.cost_optimizer_ai import _build_recommendations_from_ai
    from unittest.mock import MagicMock

    mock_db = MagicMock()
    recs = await _build_recommendations_from_ai("", mock_db)
    assert recs == []


# ── Pydantic 스키마 검증 ──────────────────────────────────────────────────────

def test_recommendation_out_schema():
    """RecommendationOut 스키마가 올바른 필드를 갖는지 확인."""
    from routers.cost_optimizer_ai import RecommendationOut
    fields = RecommendationOut.model_fields
    required = {"id", "category", "title", "estimated_saving", "risk_level",
                "auto_applicable", "status", "created_at"}
    for f in required:
        assert f in fields, f"RecommendationOut 필드 누락: {f}"


def test_analysis_out_schema():
    """AnalysisOut 스키마가 올바른 필드를 갖는지 확인."""
    from routers.cost_optimizer_ai import AnalysisOut
    fields = AnalysisOut.model_fields
    required = {"id", "period", "total_cost", "created_at"}
    for f in required:
        assert f in fields, f"AnalysisOut 필드 누락: {f}"


def test_forecast_out_schema():
    """ForecastOut 스키마가 올바른 필드를 갖는지 확인."""
    from routers.cost_optimizer_ai import ForecastOut
    fields = ForecastOut.model_fields
    required = {"id", "forecast_date", "predicted_cost", "confidence", "created_at"}
    for f in required:
        assert f in fields, f"ForecastOut 필드 누락: {f}"


# ── ORM 모델 컬럼 검증 ────────────────────────────────────────────────────────

def test_cost_ai_analysis_columns():
    """CostAIAnalysis 모델에 필수 컬럼이 있는지 확인."""
    from models import CostAIAnalysis
    cols = {c.key for c in CostAIAnalysis.__table__.columns}
    required = {"id", "period", "total_cost", "breakdown", "ai_insights",
                "waste_detected", "created_at"}
    for c in required:
        assert c in cols, f"CostAIAnalysis 컬럼 누락: {c}"


def test_cost_recommendation_columns():
    """CostRecommendation 모델에 필수 컬럼이 있는지 확인."""
    from models import CostRecommendation
    cols = {c.key for c in CostRecommendation.__table__.columns}
    required = {"id", "category", "title", "description", "estimated_saving",
                "risk_level", "auto_applicable", "status", "created_at"}
    for c in required:
        assert c in cols, f"CostRecommendation 컬럼 누락: {c}"


def test_cost_forecast_columns():
    """CostForecast 모델에 필수 컬럼이 있는지 확인."""
    from models import CostForecast
    cols = {c.key for c in CostForecast.__table__.columns}
    required = {"id", "forecast_date", "predicted_cost", "confidence",
                "factors", "created_at"}
    for c in required:
        assert c in cols, f"CostForecast 컬럼 누락: {c}"


# ── 비즈니스 로직: 예측 수학적 검증 ─────────────────────────────────────────

def test_forecast_compound_growth():
    """복리 성장률 계산 공식이 올바른지 수동 검증."""
    base_cost   = 500.0   # 만원
    trend_rate  = 0.02    # 월 2%
    month_ahead = 3

    predicted = base_cost * ((1 + trend_rate) ** month_ahead)
    expected  = 500.0 * (1.02 ** 3)

    assert abs(predicted - expected) < 0.01, f"예측 계산 불일치: {predicted} != {expected}"


def test_confidence_bounds():
    """신뢰도가 0.3 ~ 0.95 범위 내인지 확인 (예측 공식 검증)."""
    import math

    def calc_confidence(m: int, history: int) -> float:
        return max(0.3, min(0.95, 0.95 - 0.1 * m - (0.05 if history < 4 else 0)))

    # 모든 케이스가 범위 내
    for m in range(1, 4):
        for h in (2, 4, 10):
            conf = calc_confidence(m, h)
            assert 0.3 <= conf <= 0.95, f"신뢰도 범위 초과: m={m} h={h} conf={conf}"
