ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-24 06:24:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

Handle URL tests

Browse source
This commit is contained in:
Jeremiah Lowin 2024-12-03 12:54:25 -05:00
commit 83eb81ff7f
5 changed files with 104 additions and 158 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/resources/test_function_resources.py
View file

@ -1,4 +1,4 @@
from pydantic import BaseModel
from pydantic import BaseModel, AnyUrl
import pytest
from fastmcp.resources import FunctionResource
@ -13,7 +13,7 @@ class TestFunctionResource:
return "test content"
resource = FunctionResource(
uri="fn://test",
uri=AnyUrl("fn://test"),
name="test",
description="test function",
fn=my_func,
@ -31,7 +31,7 @@ class TestFunctionResource:
return "Hello, world!"
resource = FunctionResource(
uri="function://test",
uri=AnyUrl("function://test"),
name="test",
fn=get_data,
)
@ -46,7 +46,7 @@ class TestFunctionResource:
return b"Hello, world!"
resource = FunctionResource(
uri="function://test",
uri=AnyUrl("function://test"),
name="test",
fn=get_data,
)
@ -60,11 +60,12 @@ class TestFunctionResource:
return {"key": "value"}
resource = FunctionResource(
uri="function://test",
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):
@ -74,7 +75,7 @@ class TestFunctionResource:
raise ValueError("Test error")
resource = FunctionResource(
uri="function://test",
uri=AnyUrl("function://test"),
name="test",
fn=failing_func,
)
@ -88,7 +89,7 @@ class TestFunctionResource:
name: str
resource = FunctionResource(
uri="function://test",
uri=AnyUrl("function://test"),
name="test",
fn=lambda: MyModel(name="test"),
)
@ -106,7 +107,7 @@ class TestFunctionResource:
return CustomData()
resource = FunctionResource(
uri="function://test",
uri=AnyUrl("function://test"),
name="test",
fn=get_data,
)