o2o-site-AEO/solution/backend/tests/test_cost.py
Mina Choi 11d30bb3d1 [chore] solution,admin,ontology: 코드 주석을 한 줄로 — 히스토리 주석 삭제
여러 줄 주석이 설명보다 경위(예전·실측·지적)를 적고 있어 읽는 사람이 결론을 찾기 어려웠다.

- ts·tsx·js·mjs·css·py 478개: 여러 줄 주석은 첫 문장 한 줄로, 과거형·날짜 문장은 삭제
- 주석 위치는 TypeScript 파서·파이썬 tokenize/ast 로 찾는다 — 문자열 안의 # · /* 는 건드리지 않는다
- eslint·ts·noqa·type: ignore 같은 지시 주석은 그대로 둔다

파이썬 275개 정리 전후 AST 동일, TS 298개 주석 뺀 토큰 동일(빈 JSX 주석 10곳만 차이).
site·frontend·admin tsc, site vitest 105 passed

Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
2026-09-28 16:05:19 +09:00

84 lines
3.0 KiB
Python

"""사이트 1건 원가 미터 — $1 예산이 실제로 호출을 막는지."""
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():
"""검증: 예산 상한값."""
assert SITE_BUDGET_KRW == USD_KRW
assert CostMeter().budget_krw == SITE_BUDGET_KRW
def test_kakao_keyword_search_is_2won():
"""검증: 레포에 확정된 카카오 단가."""
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."""
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():
"""검증: 예산을 넘기는 호출을 '하기 전' 에 막는가."""
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():
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():
"""검증: 추정 단가로 실배치를 돌리려 할 때."""
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} 단가가 확정인데 출처가 비어 있다"