Skip to content

Commit 0f93c61

Browse files
committed
update IT tests for new get_schema format
1 parent c49f4df commit 0f93c61

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

servers/mcp-neo4j-cypher/tests/integration/test_server_IT.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,12 +10,14 @@ async def test_get_neo4j_schema(mcp_server: FastMCP, init_data: Any):
1010
response = await mcp_server.call_tool("get_neo4j_schema", dict())
1111

1212
temp_parsed = json.loads(response[0].text)['content'][0]['text']
13-
schema = json.loads(temp_parsed)[0]
14-
13+
schema = json.loads(temp_parsed)
14+
1515
# Verify the schema result
16-
assert "label" in schema
17-
assert "attributes" in schema
18-
assert "relationships" in schema
16+
assert "Person" in schema
17+
assert schema['Person']['count'] == 3
18+
assert len(schema['Person']['properties']) == 2
19+
assert "FRIEND" in schema['Person']['relationships']
20+
1921

2022

2123
@pytest.mark.asyncio(loop_scope="function")

0 commit comments

Comments
 (0)