@@ -322,46 +322,6 @@ pub(crate) fn agent_start(
322322 Ok ( AgentStartResult { started : true } )
323323}
324324
325- #[ cfg( test) ]
326- mod tests {
327- use super :: * ;
328-
329- #[ test]
330- fn fallback_agent_lifecycle_marks_first_output_as_tool_started_once ( ) {
331- let mut state = AgentLifecycleFallbackState :: default ( ) ;
332-
333- assert_eq ! (
334- fallback_agent_lifecycle_from_output( & mut state, "fixture-running\n " ) ,
335- Some ( (
336- "tool_started" ,
337- "AgentProcessOutput" ,
338- r#"{"source":"agent_process_output"}"# ,
339- ) ) ,
340- ) ;
341- assert_eq ! (
342- fallback_agent_lifecycle_from_output( & mut state, "fixture-still-running\n " ) ,
343- None
344- ) ;
345- }
346-
347- #[ test]
348- fn fallback_agent_lifecycle_only_emits_completion_after_output_started ( ) {
349- let mut state = AgentLifecycleFallbackState :: default ( ) ;
350- assert_eq ! ( fallback_agent_lifecycle_from_exit( & mut state) , None ) ;
351-
352- let _ = fallback_agent_lifecycle_from_output ( & mut state, "fixture-running\n " ) ;
353- assert_eq ! (
354- fallback_agent_lifecycle_from_exit( & mut state) ,
355- Some ( (
356- "turn_completed" ,
357- "AgentProcessExit" ,
358- r#"{"source":"agent_process_exit"}"# ,
359- ) ) ,
360- ) ;
361- assert_eq ! ( fallback_agent_lifecycle_from_exit( & mut state) , None ) ;
362- }
363- }
364-
365325pub ( crate ) fn agent_send (
366326 workspace_id : String ,
367327 session_id : String ,
@@ -477,3 +437,43 @@ pub(crate) fn agent_resize(
477437 } )
478438 . map_err ( |e| e. to_string ( ) )
479439}
440+
441+ #[ cfg( test) ]
442+ mod tests {
443+ use super :: * ;
444+
445+ #[ test]
446+ fn fallback_agent_lifecycle_marks_first_output_as_tool_started_once ( ) {
447+ let mut state = AgentLifecycleFallbackState :: default ( ) ;
448+
449+ assert_eq ! (
450+ fallback_agent_lifecycle_from_output( & mut state, "fixture-running\n " ) ,
451+ Some ( (
452+ "tool_started" ,
453+ "AgentProcessOutput" ,
454+ r#"{"source":"agent_process_output"}"# ,
455+ ) ) ,
456+ ) ;
457+ assert_eq ! (
458+ fallback_agent_lifecycle_from_output( & mut state, "fixture-still-running\n " ) ,
459+ None
460+ ) ;
461+ }
462+
463+ #[ test]
464+ fn fallback_agent_lifecycle_only_emits_completion_after_output_started ( ) {
465+ let mut state = AgentLifecycleFallbackState :: default ( ) ;
466+ assert_eq ! ( fallback_agent_lifecycle_from_exit( & mut state) , None ) ;
467+
468+ let _ = fallback_agent_lifecycle_from_output ( & mut state, "fixture-running\n " ) ;
469+ assert_eq ! (
470+ fallback_agent_lifecycle_from_exit( & mut state) ,
471+ Some ( (
472+ "turn_completed" ,
473+ "AgentProcessExit" ,
474+ r#"{"source":"agent_process_exit"}"# ,
475+ ) ) ,
476+ ) ;
477+ assert_eq ! ( fallback_agent_lifecycle_from_exit( & mut state) , None ) ;
478+ }
479+ }
0 commit comments