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

Fix tests

Browse source
This commit is contained in:
Jeremiah Lowin 2025-04-15 10:11:06 -04:00
commit faa8944cfd
1 changed files with 16 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/resources/test_resource_manager.py
View file

@ -696,13 +696,17 @@ class TestCustomResourceKeys:
def test_add_resource_with_custom_key(self, temp_file: Path):
"""Test adding a resource with a custom key different from its URI."""
manager = ResourceManager()
original_uri = f"file://{temp_file}"
original_uri = "data://test/resource"
custom_key = "custom://resource/key"
resource = FileResource(
uri=FileUrl(original_uri),
# Create a function resource instead of file resource to avoid path issues
async def get_data():
return "Test data"
resource = FunctionResource(
uri=AnyUrl(original_uri),
name="test_resource",
path=temp_file,
fn=get_data,
)
manager.add_resource(resource, key=custom_key)
@ -743,13 +747,17 @@ class TestCustomResourceKeys:
async def test_get_resource_with_custom_key(self, temp_file: Path):
"""Test that get_resource works with resources added with custom keys."""
manager = ResourceManager()
original_uri = f"file://{temp_file}"
original_uri = "data://test/resource"
custom_key = "custom://resource/path"
resource = FileResource(
uri=FileUrl(original_uri),
# Create a function resource instead of file resource to avoid path issues
async def get_data():
return "Test data"
resource = FunctionResource(
uri=AnyUrl(original_uri),
name="test_resource",
path=temp_file,
fn=get_data,
)
manager.add_resource(resource, key=custom_key)