wehub-resource-sync
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
"""Tests for session cost source rollup helpers."""
|
|
|
|
from opensquilla.session.cost_rollup import (
|
|
normalize_event_cost_source,
|
|
rollup_cost_source,
|
|
)
|
|
|
|
|
|
def test_normalize_event_cost_source_prefers_provider_billed_signals() -> None:
|
|
assert normalize_event_cost_source("provider_billed") == "provider_billed"
|
|
assert normalize_event_cost_source("openrouter_usage") == "provider_billed"
|
|
assert normalize_event_cost_source(None, billed_cost_usd=0.01) == "provider_billed"
|
|
|
|
|
|
def test_normalize_event_cost_source_preserves_mixed_components() -> None:
|
|
assert normalize_event_cost_source("mixed") == "mixed"
|
|
assert (
|
|
normalize_event_cost_source(None, cost_usd=0.03, billed_cost_usd=0.01)
|
|
== "mixed"
|
|
)
|
|
|
|
|
|
def test_normalize_event_cost_source_distinguishes_estimate_unavailable_and_none() -> None:
|
|
assert normalize_event_cost_source("opensquilla_estimate") == "opensquilla_estimate"
|
|
assert normalize_event_cost_source(None, cost_usd=0.02) == "opensquilla_estimate"
|
|
assert normalize_event_cost_source(None, input_tokens=100) == "unavailable"
|
|
assert normalize_event_cost_source("unpriced") == "unavailable"
|
|
assert normalize_event_cost_source(None) == "none"
|
|
|
|
|
|
def test_rollup_cost_source_classifies_single_and_mixed_components() -> None:
|
|
assert (
|
|
rollup_cost_source(
|
|
billed_cost_usd=0.03,
|
|
estimated_cost_component_usd=0.0,
|
|
missing_cost_entries=0,
|
|
)
|
|
== "provider_billed"
|
|
)
|
|
assert (
|
|
rollup_cost_source(
|
|
billed_cost_usd=0.0,
|
|
estimated_cost_component_usd=0.02,
|
|
missing_cost_entries=0,
|
|
)
|
|
== "opensquilla_estimate"
|
|
)
|
|
assert (
|
|
rollup_cost_source(
|
|
billed_cost_usd=0.0,
|
|
estimated_cost_component_usd=0.0,
|
|
missing_cost_entries=1,
|
|
)
|
|
== "unavailable"
|
|
)
|
|
assert (
|
|
rollup_cost_source(
|
|
billed_cost_usd=0.03,
|
|
estimated_cost_component_usd=0.02,
|
|
missing_cost_entries=0,
|
|
)
|
|
== "mixed"
|
|
)
|
|
assert (
|
|
rollup_cost_source(
|
|
billed_cost_usd=0.0,
|
|
estimated_cost_component_usd=0.0,
|
|
missing_cost_entries=0,
|
|
)
|
|
== "none"
|
|
)
|