diff --git a/hive-bash-mcp/src/bin/mcp.rs b/hive-bash-mcp/src/bin/mcp.rs index c7c5b23e..544d39d9 100644 --- a/hive-bash-mcp/src/bin/mcp.rs +++ b/hive-bash-mcp/src/bin/mcp.rs @@ -261,30 +261,29 @@ async fn main() -> Result<()> { #[cfg(test)] mod status_hint_tests { use super::{BASH_IDLE_WAIT_HINT, render_bash_status}; - use anyhow::Result; use hive_bash_mcp::protocol::DaemonResponse; - fn ok(status: &str) -> Result { - Ok(DaemonResponse::Ok { + fn status_resp(status: &str) -> DaemonResponse { + DaemonResponse::Ok { payload: serde_json::json!({ "status": status, "started_at": 1 }), - }) + } } #[test] fn running_task_after_wait_appends_idle_hint() { - let out = render_bash_status("t1", ok("running"), true); + let out = render_bash_status("t1", Ok(status_resp("running")), true); assert!(out.contains(BASH_IDLE_WAIT_HINT)); } #[test] fn running_task_without_wait_has_no_hint() { - let out = render_bash_status("t1", ok("running"), false); + let out = render_bash_status("t1", Ok(status_resp("running")), false); assert!(!out.contains(BASH_IDLE_WAIT_HINT)); } #[test] fn finished_task_after_wait_has_no_hint() { - let out = render_bash_status("t1", ok("done"), true); + let out = render_bash_status("t1", Ok(status_resp("done")), true); assert!(!out.contains(BASH_IDLE_WAIT_HINT)); } }