From 2858a74f3843f39071629c550dcdc1e5007ffbd3 Mon Sep 17 00:00:00 2001 From: jesus Date: Fri, 18 Sep 2026 00:18:21 +0000 Subject: [PATCH] test(keys): assert accepted project assignment flows into update payload Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../management_endpoints/test_key_management_endpoints.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py index bd8bf4316fb..1b9642fcb94 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py @@ -19697,13 +19697,17 @@ async def test_project_assignment_to_unassigned_key_on_same_team(): token="project-assign-token", project_id=None, team_id="team-lit-5823", models=["model-orbit"] ) cache: Final = await _cache_with_project("project-orbit", ["model-orbit"]) + data: Final = UpdateKeyRequest(key=existing.token, project_id="project-orbit") await _validate_update_key_data( - UpdateKeyRequest(key=existing.token, project_id="project-orbit"), existing, + data, existing, UserAPIKeyAuth(user_role=LitellmUserRoles.PROXY_ADMIN), None, False, _project_assignment_prisma(), cache, ) + update: Final = await prepare_key_update_data(data=data, existing_key_row=existing) + assert update["project_id"] == "project-orbit" + @pytest.mark.asyncio async def test_project_assignment_rejects_project_on_another_team():