Spaces:
Running
Running
Jeremiah Lowin commited on
Commit ·
51adb4d
1
Parent(s): 92fd239
Update test_memory_client.py
Browse files
tests/client/test_memory_client.py
CHANGED
|
@@ -48,7 +48,6 @@ def fastmcp_server():
|
|
| 48 |
return server
|
| 49 |
|
| 50 |
|
| 51 |
-
@pytest.mark.asyncio
|
| 52 |
async def test_list_tools(fastmcp_server):
|
| 53 |
"""Test listing tools with InMemoryClient."""
|
| 54 |
client = InMemoryClient(server=fastmcp_server)
|
|
@@ -63,7 +62,6 @@ async def test_list_tools(fastmcp_server):
|
|
| 63 |
assert "error_tool" in tool_names
|
| 64 |
|
| 65 |
|
| 66 |
-
@pytest.mark.asyncio
|
| 67 |
async def test_call_tool(fastmcp_server):
|
| 68 |
"""Test calling a tool with InMemoryClient."""
|
| 69 |
client = InMemoryClient(server=fastmcp_server)
|
|
@@ -76,7 +74,6 @@ async def test_call_tool(fastmcp_server):
|
|
| 76 |
assert "Hello, World!" in content_str
|
| 77 |
|
| 78 |
|
| 79 |
-
@pytest.mark.asyncio
|
| 80 |
async def test_list_resources(fastmcp_server):
|
| 81 |
"""Test listing resources with InMemoryClient."""
|
| 82 |
client = InMemoryClient(server=fastmcp_server)
|
|
@@ -89,7 +86,6 @@ async def test_list_resources(fastmcp_server):
|
|
| 89 |
assert str(result.resources[0].uri) == "data://users"
|
| 90 |
|
| 91 |
|
| 92 |
-
@pytest.mark.asyncio
|
| 93 |
async def test_list_prompts(fastmcp_server):
|
| 94 |
"""Test listing prompts with InMemoryClient."""
|
| 95 |
client = InMemoryClient(server=fastmcp_server)
|
|
@@ -102,7 +98,6 @@ async def test_list_prompts(fastmcp_server):
|
|
| 102 |
assert result.prompts[0].name == "welcome"
|
| 103 |
|
| 104 |
|
| 105 |
-
@pytest.mark.asyncio
|
| 106 |
async def test_get_prompt(fastmcp_server):
|
| 107 |
"""Test getting a prompt with InMemoryClient."""
|
| 108 |
client = InMemoryClient(server=fastmcp_server)
|
|
@@ -115,7 +110,6 @@ async def test_get_prompt(fastmcp_server):
|
|
| 115 |
assert "Welcome to FastMCP, Developer!" in result_str
|
| 116 |
|
| 117 |
|
| 118 |
-
@pytest.mark.asyncio
|
| 119 |
async def test_read_resource(fastmcp_server):
|
| 120 |
"""Test reading a resource with InMemoryClient."""
|
| 121 |
client = InMemoryClient(server=fastmcp_server)
|
|
@@ -134,7 +128,6 @@ async def test_read_resource(fastmcp_server):
|
|
| 134 |
assert "Charlie" in contents_str
|
| 135 |
|
| 136 |
|
| 137 |
-
@pytest.mark.asyncio
|
| 138 |
async def test_client_connection(fastmcp_server):
|
| 139 |
"""Test that the client connects and disconnects properly."""
|
| 140 |
client = InMemoryClient(server=fastmcp_server)
|
|
@@ -150,7 +143,6 @@ async def test_client_connection(fastmcp_server):
|
|
| 150 |
assert not client.is_connected()
|
| 151 |
|
| 152 |
|
| 153 |
-
@pytest.mark.asyncio
|
| 154 |
async def test_resource_template(fastmcp_server):
|
| 155 |
"""Test using a resource template with InMemoryClient."""
|
| 156 |
client = InMemoryClient(server=fastmcp_server)
|
|
|
|
| 48 |
return server
|
| 49 |
|
| 50 |
|
|
|
|
| 51 |
async def test_list_tools(fastmcp_server):
|
| 52 |
"""Test listing tools with InMemoryClient."""
|
| 53 |
client = InMemoryClient(server=fastmcp_server)
|
|
|
|
| 62 |
assert "error_tool" in tool_names
|
| 63 |
|
| 64 |
|
|
|
|
| 65 |
async def test_call_tool(fastmcp_server):
|
| 66 |
"""Test calling a tool with InMemoryClient."""
|
| 67 |
client = InMemoryClient(server=fastmcp_server)
|
|
|
|
| 74 |
assert "Hello, World!" in content_str
|
| 75 |
|
| 76 |
|
|
|
|
| 77 |
async def test_list_resources(fastmcp_server):
|
| 78 |
"""Test listing resources with InMemoryClient."""
|
| 79 |
client = InMemoryClient(server=fastmcp_server)
|
|
|
|
| 86 |
assert str(result.resources[0].uri) == "data://users"
|
| 87 |
|
| 88 |
|
|
|
|
| 89 |
async def test_list_prompts(fastmcp_server):
|
| 90 |
"""Test listing prompts with InMemoryClient."""
|
| 91 |
client = InMemoryClient(server=fastmcp_server)
|
|
|
|
| 98 |
assert result.prompts[0].name == "welcome"
|
| 99 |
|
| 100 |
|
|
|
|
| 101 |
async def test_get_prompt(fastmcp_server):
|
| 102 |
"""Test getting a prompt with InMemoryClient."""
|
| 103 |
client = InMemoryClient(server=fastmcp_server)
|
|
|
|
| 110 |
assert "Welcome to FastMCP, Developer!" in result_str
|
| 111 |
|
| 112 |
|
|
|
|
| 113 |
async def test_read_resource(fastmcp_server):
|
| 114 |
"""Test reading a resource with InMemoryClient."""
|
| 115 |
client = InMemoryClient(server=fastmcp_server)
|
|
|
|
| 128 |
assert "Charlie" in contents_str
|
| 129 |
|
| 130 |
|
|
|
|
| 131 |
async def test_client_connection(fastmcp_server):
|
| 132 |
"""Test that the client connects and disconnects properly."""
|
| 133 |
client = InMemoryClient(server=fastmcp_server)
|
|
|
|
| 143 |
assert not client.is_connected()
|
| 144 |
|
| 145 |
|
|
|
|
| 146 |
async def test_resource_template(fastmcp_server):
|
| 147 |
"""Test using a resource template with InMemoryClient."""
|
| 148 |
client = InMemoryClient(server=fastmcp_server)
|