learningcircuit--local-deep-research/tests/security/test_log_sanitizer_edge_cases.py

"""Edge-case tests for sanitize_for_log() max_length boundary behavior.

The main test_log_sanitizer.py tests normal truncation (max_length > 3) but
misses the branch at line 42 where max_length <= 3 skips the ellipsis.
"""

from local_deep_research.security.log_sanitizer import sanitize_for_log


class TestSanitizeForLogMaxLengthBoundary:
    """Tests for max_length <= 3 branch in sanitize_for_log()."""

    def test_max_length_3_no_ellipsis(self):
        """max_length=3 with long input: max_length > 3 is False, so no ellipsis."""
        result = sanitize_for_log("abcdef", max_length=3)
        # max_length > 3 is False → falls to else → cleaned[:3]
        assert result == "abc"

    def test_max_length_2_truncates_without_ellipsis(self):
        result = sanitize_for_log("abcdef", max_length=2)
        assert result == "ab"

    def test_max_length_1_single_char(self):
        result = sanitize_for_log("abcdef", max_length=1)
        assert result == "a"

    def test_max_length_0_returns_empty(self):
        result = sanitize_for_log("abcdef", max_length=0)
        assert result == ""

    def test_max_length_4_gets_one_char_plus_ellipsis(self):
        """max_length=4 is > 3, so ellipsis branch applies."""
        result = sanitize_for_log("abcdef", max_length=4)
        assert result == "a..."
        assert len(result) == 4