review

codex-rs/mcp-server/tests/common/lib.rs

@@ -1,7 +1,9 @@
+mod config;
 mod mcp_process;
 mod mock_model_server;
 mod responses;
 
+pub use config::create_config_toml;
 pub use mcp_process::McpProcess;
 pub use mock_model_server::create_mock_chat_completions_server;
 pub use responses::create_apply_patch_sse_response;

codex-rs/mcp-server/tests/common/mcp_process.rs

@@ -2,6 +2,7 @@ use std::path::Path;
 use std::process::Stdio;
 use std::sync::atomic::AtomicI64;
 use std::sync::atomic::Ordering;
+use std::time::Duration;
 use tokio::io::AsyncBufReadExt;
 use tokio::io::AsyncWriteExt;
 use tokio::io::BufReader;
@@ -295,7 +296,7 @@ impl McpProcess {
     /// Wait for an agent_message with a bounded timeout. Returns Some(params) if received, None on timeout.
     pub async fn maybe_wait_for_agent_message(
         &mut self,
-        dur: std::time::Duration,
+        dur: Duration,
     ) -> anyhow::Result<Option<serde_json::Value>> {
         match tokio::time::timeout(dur, self.wait_for_agent_message()).await {
             Ok(Ok(v)) => Ok(Some(v)),