From c18f411a5eb17e2c526beccec1e4a5ee07ab0e6e Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 14 Nov 2025 17:28:15 -0800 Subject: [PATCH] test_encrypt_response_id_success --- .../test_responses_id_security.py | 30 +++++++++++-------- 1 file changed, 18 insertions(+), 12 deletions(-) diff --git a/tests/test_litellm/test_responses_id_security.py b/tests/test_litellm/test_responses_id_security.py index 16e57f73cf..e72a09ee0d 100644 --- a/tests/test_litellm/test_responses_id_security.py +++ b/tests/test_litellm/test_responses_id_security.py @@ -115,14 +115,17 @@ class TestEncryptResponseId: "litellm.proxy.hooks.responses_id_security.encrypt_value_helper" ) as mock_encrypt: mock_encrypt.return_value = "encrypted_base64_value" + + with patch.object( + responses_id_security, "_get_signing_key", return_value="test-key" + ): + result = responses_id_security._encrypt_response_id( + mock_response, mock_user_api_key_dict + ) - result = responses_id_security._encrypt_response_id( - mock_response, mock_user_api_key_dict - ) - - assert result.id == "resp_encrypted_base64_value" - assert result.id.startswith("resp_") - mock_encrypt.assert_called_once() + assert result.id == "resp_encrypted_base64_value" + assert result.id.startswith("resp_") + mock_encrypt.assert_called_once() def test_encrypt_response_id_maintains_prefix( self, responses_id_security, mock_user_api_key_dict @@ -136,12 +139,15 @@ class TestEncryptResponseId: "litellm.proxy.hooks.responses_id_security.encrypt_value_helper" ) as mock_encrypt: mock_encrypt.return_value = "encrypted_value_456" + + with patch.object( + responses_id_security, "_get_signing_key", return_value="test-key" + ): + result = responses_id_security._encrypt_response_id( + mock_response, mock_user_api_key_dict + ) - result = responses_id_security._encrypt_response_id( - mock_response, mock_user_api_key_dict - ) - - assert result.id.startswith("resp_") + assert result.id.startswith("resp_") class TestCheckUserAccessToResponseId: