diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/auth/test_user_api_key_auth_mcp.py b/tests/test_litellm/proxy/_experimental/mcp_server/auth/test_user_api_key_auth_mcp.py index 980b93faa3f..729c1b42aea 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/auth/test_user_api_key_auth_mcp.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/auth/test_user_api_key_auth_mcp.py @@ -2249,9 +2249,12 @@ async def test_get_org_object_includes_object_permission_with_separate_cache_key include_object_permission=True, ) - assert result == org_row + assert result is not None + assert result.organization_id == "org-123" + assert result.object_permission_id == "org-perm-123" mock_cache.async_get_cache.assert_awaited_once_with( - key="org_id:org-123:with_object_permission" + key="org_id:org-123:with_object_permission", + model_type=type(result), ) mock_prisma.db.litellm_organizationtable.find_unique.assert_awaited_once_with( where={"organization_id": "org-123"},