test(budget): add spend threshold evaluation and budget reset arithmetic tests [skip ci]

This commit is contained in:
Kartik Chilkoti 2026-09-16 17:06:14 +05:30
parent a8979fe054
commit 1998fedd4a

View file

@ -0,0 +1,30 @@
\"\"\"
Unit tests for user and team budget tracking arithmetic and threshold evaluation in LiteLLM proxy.
\"\"\"
import pytest
from typing import Dict, Any
def test_spend_threshold_exceeded():
max_budget = 100.0
current_spend = 95.50
incoming_cost = 5.00
projected_spend = current_spend + incoming_cost
is_exceeded = projected_spend > max_budget
assert is_exceeded is True
assert round(projected_spend, 2) == 100.50
def test_spend_threshold_within_limit():
max_budget = 50.0
current_spend = 20.0
incoming_cost = 12.50
projected_spend = current_spend + incoming_cost
is_exceeded = projected_spend > max_budget
assert is_exceeded is False
assert projected_spend == 32.50
def test_daily_budget_reset_calculation():
budget_duration = "1d"
seconds_in_day = 86400
assert seconds_in_day == 24 * 60 * 60