test: Fix broken mcp-stdio module imports in unit tests
This commit is contained in:
1 parent
368f28e593
commit
4fe6ed7791
2 files changed
+17
-3
No files matched your search
@@ -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());
|
||||
}
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user