o2o-site-AEO/solution/backend/tests/test_cost.py

"""사이트 1건 원가 미터 — $1 예산이 실제로 호출을 막는지.

★ 이 테스트가 지키는 제품 제약: 업소 1곳의 사이트 생성 외부 API 비용 <= $1(1,400원).
  예산이 "문서에 적힌 목표" 로만 남지 않게, 가드가 실제로 예외를 던지는지 확인한다.
"""
import pytest

from common.cost import (
    RATES,
    SITE_BUDGET_KRW,
    USD_KRW,
    BudgetExceeded,
    CostMeter,
    Provider,
    UnconfirmedRate,
    assert_rates_confirmed,
    estimate_krw,
)


def test_budget_is_one_dollar():
    """검증: 예산 상한값.
    기대결과: 1,400원 = $1 — 환율 상수와 예산 상수가 어긋나지 않는다."""
    assert SITE_BUDGET_KRW == USD_KRW
    assert CostMeter().budget_krw == SITE_BUDGET_KRW


def test_kakao_keyword_search_is_2won():
    """검증: 레포에 확정된 카카오 단가.
    기대결과: 키워드 검색 2원/건, 좌표 변환 0.5원/건 — 좌표가 4배 싸다."""
    assert estimate_krw(Provider.KAKAO, calls=1) == 2.0
    assert estimate_krw(Provider.KAKAO, kakao_coord_calls=1) == 0.5
    assert estimate_krw(Provider.KAKAO, calls=10) == 20.0


def test_free_providers_cost_nothing():
    """검증: 무료 API.
    기대결과: TourAPI·Open-Meteo 는 아무리 불러도 0원."""
    assert estimate_krw(Provider.TOUR_API, calls=100) == 0.0
    assert estimate_krw(Provider.OPEN_METEO, calls=100) == 0.0


def test_guard_blocks_before_the_call():
    """검증: 예산을 넘기는 호출을 '하기 전' 에 막는가.
    기대결과: guard 가 BudgetExceeded — 호출측은 이 호출을 하지 않는다."""
    meter = CostMeter(place_id=1)
    meter.charge(Provider.KAKAO, calls=699)          # 1,398원 — 상한 코앞
    assert meter.remaining_krw == pytest.approx(2.0)

    meter.guard(Provider.KAKAO, calls=1)             # 2원 — 딱 맞으므로 통과
    with pytest.raises(BudgetExceeded):
        meter.guard(Provider.KAKAO, calls=2)         # 4원 — 넘으므로 차단

    # ★ guard 는 확인만 한다. 막혔다고 이미 쓴 금액이 변하면 안 된다.
    assert meter.spent_krw == pytest.approx(1398.0)


def test_charge_raises_after_exceeding_to_stop_the_next_call():
    """검증: 실측을 적었더니 예산을 넘긴 경우.
    기대결과: 금액은 그대로 누적하되 BudgetExceeded — 이미 나간 호출은 못 되돌리고 다음을 막는다."""
    meter = CostMeter(place_id=2)
    with pytest.raises(BudgetExceeded):
        meter.charge(Provider.KAKAO, calls=1000)     # 2,000원 — 한 번에 초과
    assert meter.spent_krw == pytest.approx(2000.0)  # 실제로 나간 돈은 그대로 기록된다
    assert meter.remaining_krw == 0.0


def test_meter_tracks_spend_per_provider():
    """검증: 공급자별 누적.
    기대결과: 합계와 공급자별 금액이 일치하고, $ 환산이 맞는다."""
    meter = CostMeter(place_id=3)
    meter.charge(Provider.KAKAO, calls=5)                        # 10원
    meter.charge(Provider.PERPLEXITY, calls=1, tokens=3000)      # 7 + 4.2 = 11.2원
    assert meter.by_provider["kakao"] == pytest.approx(10.0)
    assert meter.by_provider["perplexity"] == pytest.approx(11.2)
    assert meter.spent_krw == pytest.approx(21.2)
    assert meter.spent_usd == pytest.approx(21.2 / USD_KRW)
    assert meter.calls == 2


def test_unconfirmed_rates_block_a_real_batch():
    """검증: 추정 단가로 실배치를 돌리려 할 때.
    기대결과: UnconfirmedRate — 추정치 위에 세운 '예산 안' 결론을 막는다."""
    assert_rates_confirmed(
        Provider.KAKAO, Provider.PERPLEXITY, Provider.TOUR_API, Provider.OPEN_METEO
    )  # 확정분은 통과
    with pytest.raises(UnconfirmedRate):
        assert_rates_confirmed()  # 전체 — 미확인이 섞여 있으므로 막힌다


def test_rate_table_declares_its_source():
    """검증: 단가표의 출처 표기.
    기대결과: 확정 단가에는 출처가 적혀 있다 — 어디서 온 숫자인지 추적 가능해야 한다."""
    for provider, rate in RATES.items():
        if rate.confirmed:
            assert rate.source, f"{provider.value} 단가가 확정인데 출처가 비어 있다"