fix: remove cost calculation from dograh codebase

api/tests/test_organization_usage_billing.py

@@ -0,0 +1,33 @@
+from types import SimpleNamespace
+from unittest.mock import AsyncMock
+
+import pytest
+
+from api.routes import organization_usage
+
+
+def test_is_mps_billing_v2_depends_only_on_account_mode():
+    assert organization_usage._is_mps_billing_v2({"billing_mode": "v2"}) is True
+    assert organization_usage._is_mps_billing_v2({"billing_mode": "v1"}) is False
+    assert organization_usage._is_mps_billing_v2({"billing_mode": "shadow"}) is False
+    assert organization_usage._is_mps_billing_v2(None) is False
+
+
+@pytest.mark.asyncio
+async def test_get_mps_billing_account_status_uses_user_provider_id(monkeypatch):
+    get_status = AsyncMock(return_value={"billing_mode": "v2"})
+    monkeypatch.setattr(
+        organization_usage.mps_service_key_client,
+        "get_billing_account_status",
+        get_status,
+    )
+
+    user = SimpleNamespace(provider_id="provider-123")
+
+    assert await organization_usage._get_mps_billing_account_status(user, 42) == {
+        "billing_mode": "v2"
+    }
+    get_status.assert_awaited_once_with(
+        organization_id=42,
+        created_by="provider-123",
+    )