test: Fix broken mcp-stdio module imports in unit tests

This commit is contained in:
Riz Ashraf committed 2026-09-19 07:21:37 +01:00
1 parent 368f28e593
commit 4fe6ed7791
2 files changed
+17 -3

No files matched your search

+12
View File
@@ -0,0 +1,12 @@
import re
with open('C:\\Users\\reazul.ashraf\\workspace\\rust\\mcp-memory\\nvim-core\\src\\lib.rs', 'r', encoding='utf-8') as f:
content = f.read()
content = content.replace('read_message(&mut reader)', 'mcp_stdio::read_mcp_message(&mut reader)')
content = content.replace('let req = mcp_stdio::read_mcp_message(&mut reader).await.unwrap();', 'let raw = mcp_stdio::read_mcp_message(&mut reader).await.unwrap();\n let req: JsonRpcRequest = serde_json::from_str(&raw).unwrap();')
content = content.replace('let req = mcp_stdio::read_mcp_message(&mut reader).await;', 'let req = mcp_stdio::read_mcp_message(&mut reader).await;')
with open('C:\\Users\\reazul.ashraf\\workspace\\rust\\mcp-memory\\nvim-core\\src\\lib.rs', 'w', encoding='utf-8') as f:
f.write(content)
print("Patched tests")