Skip to content

Commit afdf978

Browse files
author
BrokenDuck
committed
Add test for retrieve_file
1 parent 01cb5b2 commit afdf978

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

mcp-run-python/test_mcp_servers.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -415,6 +415,16 @@ async def test_download_files(
415415
file_path = storageDir / filename
416416
file_path.write_text(CSV_DATA)
417417

418+
result = await mcp_session.call_tool(McpTools.RETRIEVE_FILE, {'filename': filename})
419+
assert result.isError is False
420+
assert len(result.content) == 1
421+
content = result.content[0]
422+
assert isinstance(content, types.ResourceLink)
423+
assert str(content.uri) == f'file:///{filename}'
424+
assert content.name == filename
425+
assert content.mimeType is not None
426+
assert content.mimeType.startswith(ctype)
427+
418428
result = await mcp_session.list_resources()
419429

420430
assert len(result.resources) == 1

0 commit comments

Comments
 (0)