86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
"""Smoke tests for fin-hub model registration."""
|
|
|
|
from datetime import date
|
|
from decimal import Decimal
|
|
|
|
import pytest
|
|
|
|
from fin_hub.models import (
|
|
Budget,
|
|
BurnRate,
|
|
Commitment,
|
|
EngagementPrice,
|
|
RunwayProjection,
|
|
ServiceCost,
|
|
TokenSpend,
|
|
)
|
|
from fin_hub.models.engagement_price import _validate_engagement_price
|
|
from fin_hub.models.service_cost import _normalize_service_cost_attribution
|
|
from hub_core.models.base import Base
|
|
|
|
|
|
def test_fin_models_register_on_metadata():
|
|
tables = {table.name for table in Base.metadata.sorted_tables}
|
|
assert "fin_budgets" in tables
|
|
assert "fin_commitments" in tables
|
|
assert "fin_burn_rates" in tables
|
|
assert "fin_runway_projections" in tables
|
|
assert "fin_token_spends" in tables
|
|
assert "fin_service_costs" in tables
|
|
assert "fin_engagement_prices" in tables
|
|
|
|
|
|
def test_model_classes_importable():
|
|
assert Budget.__tablename__ == "fin_budgets"
|
|
assert Commitment.__tablename__ == "fin_commitments"
|
|
assert BurnRate.__tablename__ == "fin_burn_rates"
|
|
assert RunwayProjection.__tablename__ == "fin_runway_projections"
|
|
assert TokenSpend.__tablename__ == "fin_token_spends"
|
|
assert ServiceCost.__tablename__ == "fin_service_costs"
|
|
assert EngagementPrice.__tablename__ == "fin_engagement_prices"
|
|
|
|
|
|
def test_service_cost_has_external_attribution_seam():
|
|
columns = ServiceCost.__table__.columns
|
|
assert columns["client_id"].nullable is True
|
|
assert columns["application_id"].nullable is True
|
|
assert columns["app_instance_id"].nullable is True
|
|
assert columns["cost_attribution_key"].nullable is True
|
|
|
|
|
|
def test_service_cost_orm_uses_shared_money_and_period_invariants():
|
|
cost = ServiceCost(
|
|
service_id="cluster",
|
|
environment="production",
|
|
period_month="2026-07",
|
|
amount=Decimal("10.005"),
|
|
currency="eur",
|
|
source=" invoice ",
|
|
)
|
|
_normalize_service_cost_attribution(None, None, cost)
|
|
assert cost.amount == Decimal("10.00")
|
|
assert cost.currency == "EUR"
|
|
assert cost.source == "invoice"
|
|
|
|
cost.period_month = "2026-13"
|
|
with pytest.raises(ValueError, match="YYYY-MM"):
|
|
_normalize_service_cost_attribution(None, None, cost)
|
|
|
|
|
|
def test_engagement_price_orm_normalizes_same_basis_as_sqlite():
|
|
price = EngagementPrice(
|
|
client_id="acme",
|
|
application_id="portal",
|
|
app_instance_id="prod-01",
|
|
cost_attribution_key=None,
|
|
period_month="placeholder",
|
|
effective_from=date(2026, 7, 15),
|
|
amount=Decimal("100.005"),
|
|
currency="eur",
|
|
source=" agreement ",
|
|
)
|
|
_validate_engagement_price(None, None, price)
|
|
assert price.period_month == "2026-07"
|
|
assert price.amount == Decimal("100.00")
|
|
assert price.currency == "EUR"
|
|
assert price.source == "agreement"
|