diff --git a/crates/agent/src/test_support.rs b/crates/agent/src/test_support.rs index e37ed3502..2701c7a33 100644 --- a/crates/agent/src/test_support.rs +++ b/crates/agent/src/test_support.rs @@ -46,10 +46,6 @@ impl MockExecutionEnvironment { } impl MockExecutionEnvironment { - pub fn set_event_callback(&mut self, cb: crate::execution_env::ExecEnvEventCallback) { - self.event_callback = Some(cb); - } - fn emit(&self, event: crate::execution_env::ExecutionEnvEvent) { if let Some(ref cb) = self.event_callback { cb(event); diff --git a/crates/attractor/src/handler/script.rs b/crates/attractor/src/handler/script.rs index 07d328bbd..c9988ee87 100644 --- a/crates/attractor/src/handler/script.rs +++ b/crates/attractor/src/handler/script.rs @@ -389,7 +389,7 @@ mod tests { let timing_path = logs_root.path().join("script_node").join("script_timing.json"); let content = std::fs::read_to_string(&timing_path).unwrap(); let json: serde_json::Value = serde_json::from_str(&content).unwrap(); - assert!(json["duration_ms"].as_u64().unwrap() >= 0); + assert!(json["duration_ms"].is_u64()); assert_eq!(json["exit_code"], 0); assert_eq!(json["timed_out"], false); } @@ -442,7 +442,7 @@ mod tests { let timing_path = logs_root.path().join("script_node").join("script_timing.json"); let content = std::fs::read_to_string(&timing_path).unwrap(); let json: serde_json::Value = serde_json::from_str(&content).unwrap(); - assert!(json["duration_ms"].as_u64().unwrap() >= 0); + assert!(json["duration_ms"].is_u64()); assert_eq!(json["exit_code"], serde_json::Value::Null); assert_eq!(json["timed_out"], true); }