chore: rustfmt, clippy lints and code tidying

This commit is contained in:
Riz Ashraf committed 2026-09-17 15:35:41 +01:00
1 parent 3716c3e698
commit 1752753fcc
19 files changed
+1273 -738

No files matched your search

+36 -16
View File
@@ -10,10 +10,14 @@ fn get_stub_exe() -> std::path::PathBuf {
#[tokio::test]
async fn test_stub_connection_refused() {
let _ = std::process::Command::new("cargo").arg("build").arg("--bin").arg("mcp-memory-stub").status();
let _ = std::process::Command::new("cargo")
.arg("build")
.arg("--bin")
.arg("mcp-memory-stub")
.status();
let target = "http://127.0.0.1:49999";
let start = Instant::now();
let mut child = Command::new(get_stub_exe())
.arg("--target")
@@ -21,19 +25,27 @@ async fn test_stub_connection_refused() {
.stdin(Stdio::null()) // close stdin immediately to simulate EOF
.spawn()
.expect("Failed to execute stub");
let res = tokio::time::timeout(Duration::from_secs(5), child.wait()).await;
let elapsed = start.elapsed();
assert!(res.is_ok(), "Stub hung on connection refused! Took {:?}", elapsed);
assert!(
res.is_ok(),
"Stub hung on connection refused! Took {:?}",
elapsed
);
}
#[tokio::test]
async fn test_stub_handles_eof_cleanly() {
let _ = std::process::Command::new("cargo").arg("build").arg("--bin").arg("mcp-memory-stub").status();
let _ = std::process::Command::new("cargo")
.arg("build")
.arg("--bin")
.arg("mcp-memory-stub")
.status();
let target = "http://127.0.0.1:49998";
let mut child = Command::new(get_stub_exe())
.arg("--target")
.arg(target)
@@ -42,28 +54,32 @@ async fn test_stub_handles_eof_cleanly() {
.stderr(Stdio::piped())
.spawn()
.expect("Failed to execute stub");
if let Some(mut stdin) = child.stdin.take() {
use tokio::io::AsyncWriteExt;
let msg = "Content-Length: 51\r\n\r\n{\"jsonrpc\":\"2.0\",\"method\":\"tools/list\",\"params\":{},\"id\":1}";
stdin.write_all(msg.as_bytes()).await.unwrap();
} // stdin dropped here
let start = Instant::now();
let res = tokio::time::timeout(Duration::from_secs(5), child.wait()).await;
let elapsed = start.elapsed();
assert!(res.is_ok(), "Stub hung after EOF! Took {:?}", elapsed);
}
#[tokio::test]
async fn test_stub_sse_fallback_failure() {
let _ = std::process::Command::new("cargo").arg("build").arg("--bin").arg("mcp-memory-stub").status();
let _ = std::process::Command::new("cargo")
.arg("build")
.arg("--bin")
.arg("mcp-memory-stub")
.status();
let listener = tokio::net::TcpListener::bind("127.0.0.1:0").await.unwrap();
let local_addr = listener.local_addr().unwrap();
let target = format!("http://127.0.0.1:{}", local_addr.port());
tokio::spawn(async move {
while let Ok((mut socket, _)) = listener.accept().await {
use tokio::io::AsyncReadExt;
@@ -77,12 +93,16 @@ async fn test_stub_sse_fallback_failure() {
let mut child = Command::new(get_stub_exe())
.arg("--target")
.arg(target)
.stdin(Stdio::null())
.stdin(Stdio::null())
.spawn()
.expect("Failed to execute stub");
let res = tokio::time::timeout(Duration::from_secs(5), child.wait()).await;
let elapsed = start.elapsed();
assert!(res.is_ok(), "Stub hung on fallback failure! Took {:?}", elapsed);
assert!(
res.is_ok(),
"Stub hung on fallback failure! Took {:?}",
elapsed
);
}