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

+5 -3
View File
@@ -836,7 +836,8 @@ mod tests {
async fn test_read_message_jsonl() {
let input = "{\"jsonrpc\": \"2.0\", \"id\": 1, \"method\": \"test\"}\n";
let mut reader = BufReader::new(input.as_bytes());
let req = read_message(&mut reader).await.unwrap();
let raw = mcp_stdio::read_mcp_message(&mut reader).await.unwrap();
let req: JsonRpcRequest = serde_json::from_str(&raw).unwrap();
assert_eq!(req.method, "test");
}
@@ -845,7 +846,8 @@ mod tests {
let payload = "{\"jsonrpc\": \"2.0\", \"id\": 2, \"method\": \"test2\"}";
let input = format!("Content-Length: {}\r\n\r\n{}", payload.len(), payload);
let mut reader = BufReader::new(input.as_bytes());
let req = read_message(&mut reader).await.unwrap();
let raw = mcp_stdio::read_mcp_message(&mut reader).await.unwrap();
let req: JsonRpcRequest = serde_json::from_str(&raw).unwrap();
assert_eq!(req.method, "test2");
}
@@ -853,7 +855,7 @@ mod tests {
async fn test_read_message_malformed() {
let input = "Content-Length: abc\r\n\r\n{}";
let mut reader = BufReader::new(input.as_bytes());
let req = read_message(&mut reader).await;
let req = mcp_stdio::read_mcp_message(&mut reader).await;
assert!(req.is_none());
}
}