mirror of
https://github.com/Alishahryar1/free-claude-code.git
synced 2026-04-28 11:30:03 +00:00
109 lines
3.3 KiB
Python
109 lines
3.3 KiB
Python
"""Tests for extract_text_from_content helper functions."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from providers.common.text import extract_text_from_content
|
|
|
|
|
|
class TestExtractTextFromContent:
|
|
"""Tests for providers.common.text.extract_text_from_content."""
|
|
|
|
def test_string_content(self):
|
|
"""Return string content as-is."""
|
|
assert extract_text_from_content("hello world") == "hello world"
|
|
|
|
def test_empty_string(self):
|
|
"""Return empty string for empty string input."""
|
|
assert extract_text_from_content("") == ""
|
|
|
|
def test_list_single_block(self):
|
|
"""Extract text from a single content block."""
|
|
block = MagicMock()
|
|
block.text = "some text"
|
|
assert extract_text_from_content([block]) == "some text"
|
|
|
|
def test_list_multiple_blocks(self):
|
|
"""Concatenate text from multiple content blocks."""
|
|
b1 = MagicMock()
|
|
b1.text = "hello "
|
|
b2 = MagicMock()
|
|
b2.text = "world"
|
|
assert extract_text_from_content([b1, b2]) == "hello world"
|
|
|
|
def test_list_with_non_text_block(self):
|
|
"""Skip blocks without text attribute."""
|
|
b1 = MagicMock()
|
|
b1.text = "hello"
|
|
b2 = MagicMock(spec=[]) # No attributes
|
|
assert extract_text_from_content([b1, b2]) == "hello"
|
|
|
|
def test_list_with_empty_text(self):
|
|
"""Skip blocks with empty text."""
|
|
b1 = MagicMock()
|
|
b1.text = ""
|
|
b2 = MagicMock()
|
|
b2.text = "world"
|
|
assert extract_text_from_content([b1, b2]) == "world"
|
|
|
|
def test_list_with_none_text(self):
|
|
"""Skip blocks with None text."""
|
|
b1 = MagicMock()
|
|
b1.text = None
|
|
b2 = MagicMock()
|
|
b2.text = "world"
|
|
assert extract_text_from_content([b1, b2]) == "world"
|
|
|
|
def test_empty_list(self):
|
|
"""Return empty string for empty list."""
|
|
assert extract_text_from_content([]) == ""
|
|
|
|
def test_non_string_non_list(self):
|
|
"""Return empty string for unexpected types."""
|
|
assert extract_text_from_content(None) == ""
|
|
assert extract_text_from_content(42) == ""
|
|
|
|
def test_list_with_non_string_text_attr(self):
|
|
"""Skip blocks where text is not a string."""
|
|
b1 = MagicMock()
|
|
b1.text = 123 # Not a string
|
|
b2 = MagicMock()
|
|
b2.text = "valid"
|
|
assert extract_text_from_content([b1, b2]) == "valid"
|
|
|
|
|
|
# --- Parametrized Edge Case Tests ---
|
|
|
|
|
|
def _make_block(text_val):
|
|
b = MagicMock()
|
|
b.text = text_val
|
|
return b
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"content,expected",
|
|
[
|
|
("hello world", "hello world"),
|
|
("", ""),
|
|
(None, ""),
|
|
(42, ""),
|
|
([], ""),
|
|
(" ", " "),
|
|
],
|
|
ids=["string", "empty_str", "none", "int", "empty_list", "whitespace_only"],
|
|
)
|
|
def test_extract_text_scalar_and_empty_parametrized(content, expected):
|
|
"""Parametrized scalar and empty input handling."""
|
|
assert extract_text_from_content(content) == expected
|
|
|
|
|
|
def test_extract_functions_whitespace_only():
|
|
"""extract_text_from_content handles whitespace-only string."""
|
|
assert extract_text_from_content(" ") == " "
|
|
|
|
|
|
def test_extract_functions_unicode():
|
|
"""extract_text_from_content handles unicode content."""
|
|
assert extract_text_from_content("日本語テスト") == "日本語テスト"
|