jason/mempalace
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: ruff format with CI-pinned version (0.4.x)

Browse Source
This commit is contained in:
eblander
2026-04-13 18:29:48 -04:00
parent 1e86892e62
commit 8dc5970ca9
6 changed files with 53 additions and 175 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_mcp_server.py
+23 -70
View File
@@ -32,9 +32,7 @@ def _get_collection(palace_path, create=False):
if create:
return (
client,
client.get_or_create_collection(
"mempalace_drawers", metadata={"hnsw:space": "cosine"}
),
client.get_or_create_collection("mempalace_drawers", metadata={"hnsw:space": "cosine"}),
)
return client, client.get_collection("mempalace_drawers")
@@ -97,9 +95,7 @@ class TestHandleRequest:
def test_notifications_initialized_returns_none(self):
from mempalace.mcp_server import handle_request
resp = handle_request(
{"method": "notifications/initialized", "id": None, "params": {}}
)
resp = handle_request({"method": "notifications/initialized", "id": None, "params": {}})
assert resp is None
def test_ping_returns_empty_result(self):
@@ -120,9 +116,7 @@ class TestHandleRequest:
assert "mempalace_add_drawer" in names
assert "mempalace_kg_add" in names
def test_null_arguments_does_not_hang(
self, monkeypatch, config, palace_path, seeded_kg
):
def test_null_arguments_does_not_hang(self, monkeypatch, config, palace_path, seeded_kg):
"""Sending arguments: null should return a result, not hang (#394)."""
_patch_mcp_server(monkeypatch, config, seeded_kg)
from mempalace.mcp_server import handle_request
@@ -227,9 +221,7 @@ class TestReadTools:
assert result["total_drawers"] == 0
assert result["wings"] == {}
def test_status_with_data(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_status_with_data(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_status
@@ -246,9 +238,7 @@ class TestReadTools:
assert result["wings"]["project"] == 3
assert result["wings"]["notes"] == 1
def test_list_rooms_all(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_list_rooms_all(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_list_rooms
@@ -257,9 +247,7 @@ class TestReadTools:
assert "frontend" in result["rooms"]
assert "planning" in result["rooms"]
def test_list_rooms_filtered(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_list_rooms_filtered(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_list_rooms
@@ -267,9 +255,7 @@ class TestReadTools:
assert "backend" in result["rooms"]
assert "planning" not in result["rooms"]
def test_get_taxonomy(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_get_taxonomy(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_get_taxonomy
@@ -290,9 +276,7 @@ class TestReadTools:
class TestSearchTool:
def test_search_basic(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_search_basic(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_search
@@ -303,18 +287,14 @@ class TestSearchTool:
top = result["results"][0]
assert "JWT" in top["text"] or "authentication" in top["text"].lower()
def test_search_with_wing_filter(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_search_with_wing_filter(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_search
result = tool_search(query="planning", wing="notes")
assert all(r["wing"] == "notes" for r in result["results"])
def test_search_with_room_filter(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_search_with_room_filter(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_search
@@ -333,9 +313,7 @@ class TestSearchTool:
assert "results" in result
# Old name takes precedence when both provided
result_strict = tool_search(
query="JWT", max_distance=999.0, min_similarity=0.01
)
result_strict = tool_search(query="JWT", max_distance=999.0, min_similarity=0.01)
result_loose = tool_search(query="JWT", max_distance=0.01, min_similarity=999.0)
assert len(result_strict["results"]) <= len(result_loose["results"])
@@ -427,9 +405,7 @@ class TestWriteTools:
assert result2["success"] is True
assert result2["reason"] == "already_exists"
def test_add_drawer_shared_header_no_collision(
self, monkeypatch, config, palace_path, kg
):
def test_add_drawer_shared_header_no_collision(self, monkeypatch, config, palace_path, kg):
"""Documents sharing a >100-char header must get distinct IDs (full-content hash)."""
_patch_mcp_server(monkeypatch, config, kg)
_client, _col = _get_collection(palace_path, create=True)
@@ -441,10 +417,7 @@ class TestWriteTools:
header
+ "Decision: Use PostgreSQL for primary storage. Rationale: ACID compliance required."
)
doc2 = (
header
+ "Decision: Use Redis for session caching. Rationale: sub-ms latency needed."
)
doc2 = header + "Decision: Use Redis for session caching. Rationale: sub-ms latency needed."
result1 = tool_add_drawer(wing="work", room="decisions", content=doc1)
result2 = tool_add_drawer(wing="work", room="decisions", content=doc2)
@@ -455,9 +428,7 @@ class TestWriteTools:
result1["drawer_id"] != result2["drawer_id"]
), "Documents with shared header but different content must have distinct drawer IDs"
def test_delete_drawer(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_delete_drawer(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_delete_drawer
@@ -465,18 +436,14 @@ class TestWriteTools:
assert result["success"] is True
assert seeded_collection.count() == 3
def test_delete_drawer_not_found(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_delete_drawer_not_found(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_delete_drawer
result = tool_delete_drawer("nonexistent_drawer")
assert result["success"] is False
def test_check_duplicate(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_check_duplicate(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_check_duplicate
@@ -505,18 +472,14 @@ class TestWriteTools:
assert result["room"] == "backend"
assert "JWT tokens" in result["content"]
def test_get_drawer_not_found(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_get_drawer_not_found(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_get_drawer
result = tool_get_drawer("nonexistent_drawer")
assert "error" in result
def test_list_drawers(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_list_drawers(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_list_drawers
@@ -544,9 +507,7 @@ class TestWriteTools:
assert result["count"] == 2
assert all(d["room"] == "backend" for d in result["drawers"])
def test_list_drawers_pagination(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_list_drawers_pagination(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_list_drawers
@@ -564,9 +525,7 @@ class TestWriteTools:
result = tool_list_drawers(offset=-5)
assert result["offset"] == 0
def test_update_drawer_content(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_update_drawer_content(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_update_drawer, tool_get_drawer
@@ -584,25 +543,19 @@ class TestWriteTools:
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_update_drawer
result = tool_update_drawer(
"drawer_proj_backend_aaa", wing="new_wing", room="new_room"
)
result = tool_update_drawer("drawer_proj_backend_aaa", wing="new_wing", room="new_room")
assert result["success"] is True
assert result["wing"] == "new_wing"
assert result["room"] == "new_room"
def test_update_drawer_not_found(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_update_drawer_not_found(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_update_drawer
result = tool_update_drawer("nonexistent_drawer", content="hello")
assert result["success"] is False
def test_update_drawer_noop(
self, monkeypatch, config, palace_path, seeded_collection, kg
):
def test_update_drawer_noop(self, monkeypatch, config, palace_path, seeded_collection, kg):
_patch_mcp_server(monkeypatch, config, kg)
from mempalace.mcp_server import tool_update_drawer