From 4618cba00e85c328a7c556ea937567e01d3a2b4a Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 6 May 2026 19:30:11 -0700 Subject: [PATCH] =?UTF-8?q?refactor(tests):=20mirror=20warm=5Fpool=20move?= =?UTF-8?q?=20=E2=80=94=20test=5Fhydrate.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Tracks LIT-2879. --- .../vms}/warm_pool/test_hydrate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) rename tests/test_litellm/{proxy/agent_session_endpoints => managed_agents/vms}/warm_pool/test_hydrate.py (97%) diff --git a/tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_hydrate.py b/tests/test_litellm/managed_agents/vms/warm_pool/test_hydrate.py similarity index 97% rename from tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_hydrate.py rename to tests/test_litellm/managed_agents/vms/warm_pool/test_hydrate.py index cb7f65a5de7..6b4f8c7654d 100644 --- a/tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_hydrate.py +++ b/tests/test_litellm/managed_agents/vms/warm_pool/test_hydrate.py @@ -1,4 +1,4 @@ -"""Unit tests for `litellm/proxy/agent_session_endpoints/warm_pool/hydrate.py`. +"""Unit tests for `litellm/managed_agents/vms/warm_pool/hydrate.py`. Verify the payload builder: * respects scope filtering — out-of-scope secrets never land in the payload @@ -22,7 +22,7 @@ import pytest os.environ.setdefault("LITELLM_SALT_KEY", "sk-test-salt-key-do-not-use-in-prod") -from litellm.proxy.agent_session_endpoints.warm_pool.hydrate import ( +from litellm.managed_agents.vms.warm_pool.hydrate import ( build_hydrate_payload, ) from litellm.proxy.agent_settings_endpoints.encryption import encrypt_optional @@ -248,7 +248,7 @@ async def test_decrypt_failure_skips_secret_does_not_crash(prisma, jwt_expires_a raise ValueError("simulated decrypt fail") with patch( - "litellm.proxy.agent_session_endpoints.warm_pool.hydrate.decrypt_optional", + "litellm.managed_agents.vms.warm_pool.hydrate.decrypt_optional", new=fake_decrypt, ): payload = await build_hydrate_payload(