"""사이트 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} 단가가 확정인데 출처가 비어 있다"