From f972fddafcb5c0da1966ab82583a9bab333bc8ab Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Thu, 17 Sep 2026 18:10:00 +0000
Subject: [PATCH] test(proxy): include temp budget fields in customer budget
 table fixture

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../proxy/management_endpoints/test_customer_endpoints.py       | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/test_litellm/proxy/management_endpoints/test_customer_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_customer_endpoints.py
index 1510d8f671d..77e52f30bb7 100644
--- a/tests/test_litellm/proxy/management_endpoints/test_customer_endpoints.py
+++ b/tests/test_litellm/proxy/management_endpoints/test_customer_endpoints.py
@@ -806,6 +806,8 @@ _EXPECTED_CUSTOMER = {
         "model_max_budget": None,
         "budget_duration": "30d",
         "allowed_models": [],
+        "temp_budget_increase": None,
+        "temp_budget_expiry": None,
         "budget_reset_at": "2024-02-01T00:00:00",
         "created_at": "2024-01-01T00:00:00",
     },