From fdaf3deb95de88fb185a363def948bcb3eff77cb Mon Sep 17 00:00:00 2001
From: kerry <kerry@berri.ai>
Date: Tue, 22 Sep 2026 21:34:28 +0000
Subject: [PATCH] test(integration): charge a team member added without any
 budget on its membership row (Pylon #7363)

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../spend/test_team_member_spend.py           | 54 +++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 tests/integration/spend/test_team_member_spend.py

diff --git a/tests/integration/spend/test_team_member_spend.py b/tests/integration/spend/test_team_member_spend.py
new file mode 100644
index 00000000000..89eb2a24bcf
--- /dev/null
+++ b/tests/integration/spend/test_team_member_spend.py
@@ -0,0 +1,54 @@
+import uuid
+from typing import Final
+
+import pytest
+from integration._support.client import Gateway, eventually, object_value
+from integration._support.database import read_rows
+
+
+@pytest.mark.covers("spend.team_member.member_without_budget_gets_membership_row_and_spend")
+def test_member_added_without_any_budget_is_charged_on_its_membership_row(gateway: Gateway) -> None:
+    with gateway.scenario() as scenario:
+        model: Final = scenario.model(input_cost_per_token=0.001, output_cost_per_token=0.002)
+        team: Final = scenario.team(models=[model])
+        user: Final = scenario.user()
+        added: Final = gateway.request(
+            "POST", "/team/member_add", {"team_id": team, "member": {"user_id": user, "role": "user"}}
+        )
+        assert added.status_code == 200, added.text
+        memberships: Final = added.json()["updated_team_memberships"]
+        assert [
+            {"user_id": row["user_id"], "team_id": row["team_id"], "budget_id": row["budget_id"], "spend": row["spend"]}
+            for row in memberships
+        ] == [{"user_id": user, "team_id": team, "budget_id": None, "spend": 0}], added.text
+        assert read_rows(
+            'SELECT budget_id, spend, total_spend FROM "LiteLLM_TeamMembership" WHERE team_id=%s AND user_id=%s',
+            (team, user),
+        ) == [{"budget_id": None, "spend": 0.0, "total_spend": 0.0}]
+        key: Final = scenario.key(team_id=team, user_id=user, models=[model])
+        assert gateway.chat(model, key=key, text=f"member spend {uuid.uuid4().hex}")["usage"]["total_tokens"] == 40
+        charged: Final = eventually(
+            lambda: read_rows(
+                'SELECT spend, total_spend FROM "LiteLLM_TeamMembership" WHERE team_id=%s AND user_id=%s',
+                (team, user),
+            ),
+            lambda values: len(values) == 1 and float(values[0]["spend"]) >= 0.06,
+            seconds=70,
+        )
+        assert float(charged[0]["spend"]) == pytest.approx(0.06)
+        assert float(charged[0]["total_spend"]) == pytest.approx(0.06)
+        team_rows: Final = eventually(
+            lambda: read_rows('SELECT spend FROM "LiteLLM_TeamTable" WHERE team_id=%s', (team,)),
+            lambda values: len(values) == 1 and float(values[0]["spend"]) >= 0.06,
+            seconds=70,
+        )
+        assert float(team_rows[0]["spend"]) == pytest.approx(0.06)
+        info: Final = gateway.get("/team/info", {"team_id": team})
+        listed: Final = info["team_memberships"]
+        assert isinstance(listed, list)
+        exposed: Final = [
+            (object_value(row)["user_id"], object_value(row)["spend"])
+            for row in listed
+            if object_value(row)["user_id"] == user
+        ]
+        assert len(exposed) == 1 and exposed[0][1] == pytest.approx(0.06), info