mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-08-09 15:19:10 +02:00
0600834da4
Single-element lists like [1] were being incorrectly unwrapped to "1" in unstructured content while multi-element lists remained as lists. This created inconsistent behavior where the structure was lost for single items. This fix ensures lists always preserve their structure in unstructured content regardless of length, making behavior consistent and predictable. Also removes pretty-printing from JSON serialization for more compact output across tools, prompts, and resources. Fixes #1064 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
131 lines
3.7 KiB
Python
131 lines
3.7 KiB
Python
import pytest
|
|
from pydantic import AnyUrl, BaseModel
|
|
|
|
from fastmcp.resources.resource import FunctionResource
|
|
|
|
|
|
class TestFunctionResource:
|
|
"""Test FunctionResource functionality."""
|
|
|
|
def test_function_resource_creation(self):
|
|
"""Test creating a FunctionResource."""
|
|
|
|
def my_func() -> str:
|
|
return "test content"
|
|
|
|
resource = FunctionResource(
|
|
uri=AnyUrl("fn://test"),
|
|
name="test",
|
|
description="test function",
|
|
fn=my_func,
|
|
)
|
|
assert str(resource.uri) == "fn://test"
|
|
assert resource.name == "test"
|
|
assert resource.description == "test function"
|
|
assert resource.mime_type == "text/plain" # default
|
|
assert resource.fn == my_func
|
|
|
|
async def test_read_text(self):
|
|
"""Test reading text from a FunctionResource."""
|
|
|
|
def get_data() -> str:
|
|
return "Hello, world!"
|
|
|
|
resource = FunctionResource(
|
|
uri=AnyUrl("function://test"),
|
|
name="test",
|
|
fn=get_data,
|
|
)
|
|
content = await resource.read()
|
|
assert content == "Hello, world!"
|
|
assert resource.mime_type == "text/plain"
|
|
|
|
async def test_read_binary(self):
|
|
"""Test reading binary data from a FunctionResource."""
|
|
|
|
def get_data() -> bytes:
|
|
return b"Hello, world!"
|
|
|
|
resource = FunctionResource(
|
|
uri=AnyUrl("function://test"),
|
|
name="test",
|
|
fn=get_data,
|
|
)
|
|
content = await resource.read()
|
|
assert content == b"Hello, world!"
|
|
|
|
async def test_json_conversion(self):
|
|
"""Test automatic JSON conversion of non-string results."""
|
|
|
|
def get_data() -> dict:
|
|
return {"key": "value"}
|
|
|
|
resource = FunctionResource(
|
|
uri=AnyUrl("function://test"),
|
|
name="test",
|
|
fn=get_data,
|
|
)
|
|
content = await resource.read()
|
|
assert isinstance(content, str)
|
|
assert '"key":"value"' in content
|
|
|
|
async def test_error_handling(self):
|
|
"""Test error handling in FunctionResource."""
|
|
|
|
def failing_func() -> str:
|
|
raise ValueError("Test error")
|
|
|
|
resource = FunctionResource(
|
|
uri=AnyUrl("function://test"),
|
|
name="test",
|
|
fn=failing_func,
|
|
)
|
|
with pytest.raises(ValueError, match="Test error"):
|
|
await resource.read()
|
|
|
|
async def test_basemodel_conversion(self):
|
|
"""Test handling of BaseModel types."""
|
|
|
|
class MyModel(BaseModel):
|
|
name: str
|
|
|
|
resource = FunctionResource(
|
|
uri=AnyUrl("function://test"),
|
|
name="test",
|
|
fn=lambda: MyModel(name="test"),
|
|
)
|
|
content = await resource.read()
|
|
assert content == '{"name":"test"}'
|
|
|
|
async def test_custom_type_conversion(self):
|
|
"""Test handling of custom types."""
|
|
|
|
class CustomData:
|
|
def __str__(self) -> str:
|
|
return "custom data"
|
|
|
|
def get_data() -> CustomData:
|
|
return CustomData()
|
|
|
|
resource = FunctionResource(
|
|
uri=AnyUrl("function://test"),
|
|
name="test",
|
|
fn=get_data,
|
|
)
|
|
content = await resource.read()
|
|
assert isinstance(content, str)
|
|
|
|
async def test_async_read_text(self):
|
|
"""Test reading text from async FunctionResource."""
|
|
|
|
async def get_data() -> str:
|
|
return "Hello, world!"
|
|
|
|
resource = FunctionResource(
|
|
uri=AnyUrl("function://test"),
|
|
name="test",
|
|
fn=get_data,
|
|
)
|
|
content = await resource.read()
|
|
assert content == "Hello, world!"
|
|
assert resource.mime_type == "text/plain"
|