From c8b0c2d0e1584692ac41e999f339127d71e4349f Mon Sep 17 00:00:00 2001
From: Shalmankursheeth <nailashalman001@gmail.com>
Date: Sun, 1 Nov 2026 08:34:04 +0000
Subject: [PATCH] test(utils): add edge-case coverage for is_base64_encoded

Adds parametrized cases for malformed input (empty string, missing comma,
non-base64 payload) and a valid text data URI. Test-only change; the
existing PNG and "Dog" cases are already covered by
test_is_base64_encoded_2.

Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01UqediyVH1ZRdT3eJqoz2bY
---
 tests/unit/test_utils.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
index 5cf872f4ea7..7c54ef309cf 100644
--- a/tests/unit/test_utils.py
+++ b/tests/unit/test_utils.py
@@ -9872,6 +9872,26 @@ def test_is_base64_encoded_2():
     assert is_base64_encoded(s="Dog") is False
 
 
+@pytest.mark.parametrize(
+    "value",
+    [
+        "",
+        "data:text/plain;base64",
+        "data:text/plain;base64,!!!not-base64!!!",
+    ],
+)
+def test_is_base64_encoded_rejects_malformed_input(value):
+    from litellm.utils import is_base64_encoded
+
+    assert is_base64_encoded(s=value) is False
+
+
+def test_is_base64_encoded_accepts_text_data_uri():
+    from litellm.utils import is_base64_encoded
+
+    assert is_base64_encoded(s="data:text/plain;base64,aGVsbG8gd29ybGQ=") is True
+
+
 @pytest.mark.parametrize(
     "messages, expected_bool",
     [