From fcdff6ed11f5bfa27cc2a67c494ea5fc2288c0ce Mon Sep 17 00:00:00 2001 From: Dominic White Date: Mon, 7 Sep 2026 12:09:43 +0200 Subject: [PATCH] test: cover idempotent safety identifier enforcement --- tests/proxy_unit_tests/test_safety_identifier.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/proxy_unit_tests/test_safety_identifier.py b/tests/proxy_unit_tests/test_safety_identifier.py index e4b79ef77ef..b57ffc09ad4 100644 --- a/tests/proxy_unit_tests/test_safety_identifier.py +++ b/tests/proxy_unit_tests/test_safety_identifier.py @@ -5,6 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, patch import pytest import litellm +from litellm.litellm_core_utils.safety_identifier import enforce_safety_identifier from litellm.llms.perplexity.responses.transformation import PerplexityResponsesConfig from litellm.proxy._types import UserAPIKeyAuth from litellm.proxy.common_request_processing import ProxyBaseLLMRequestProcessing @@ -24,6 +25,16 @@ def test_enforce_safety_identifier_hashes_authenticated_user(monkeypatch: pytest assert data["safety_identifier"] == hashlib.sha256(b"user-123").hexdigest() +def test_enforce_safety_identifier_is_idempotent(): + safety_identifier = hashlib.sha256(b"user-123").hexdigest() + data = {"safety_identifier": safety_identifier} + + modified = enforce_safety_identifier(data=data, user_id="user-123", enabled=True) + + assert modified is False + assert data == {"safety_identifier": safety_identifier} + + @pytest.mark.parametrize("setting", [None, "false"]) def test_enforce_safety_identifier_is_opt_in(monkeypatch: pytest.MonkeyPatch, setting: str | None): if setting is None: