refactored to have common assert_contains function and finished verification for first AgentSpawn hook integration test

Author: ekang7

crates/agent/tests/common/mod.rs

@@ -302,3 +302,12 @@ pub async fn parse_response_streams(content: impl AsRef<str>) -> Result<MockResp
     }
     Ok(stream)
 }
+
+pub fn assert_contains(text: &str, expected: &str) {
+    assert!(
+        text.contains(expected), 
+        "expected to find '{}' inside content: '{}'",
+        expected,
+        text 
+    );
+}

crates/agent/tests/mod.rs

@@ -58,18 +58,10 @@ async fn test_agent_defaults() {
 
     for req in test.requests() {
         let first_msg = req.messages().first().expect("first message should exist").text();
-        let assert_contains = |expected: &str| {
-            assert!(
-                first_msg.contains(expected),
-                "expected to find '{}' inside content: '{}'",
-                expected,
-                first_msg
-            );
-        };
-        assert_contains(AMAZON_Q_MD_CONTENT);
-        assert_contains(AGENTS_MD_CONTENT);
-        assert_contains(README_MD_CONTENT);
-        assert_contains(LOCAL_RULE_MD_CONTENT);
-        assert_contains(SUB_LOCAL_RULE_MD_CONTENT);
+        assert_contains(&first_msg, AMAZON_Q_MD_CONTENT);
+        assert_contains(&first_msg,AGENTS_MD_CONTENT);
+        assert_contains(&first_msg,README_MD_CONTENT);
+        assert_contains(&first_msg,LOCAL_RULE_MD_CONTENT);
+        assert_contains(&first_msg,SUB_LOCAL_RULE_MD_CONTENT);
     }
 }

crates/agent/tests/test_hooks.rs

@@ -45,7 +45,8 @@ async fn test_agent_spawn_hook() {
     info!("before stop");
     test.wait_until_agent_stop(Duration::from_secs(10)).await;
     info!("after stop");
-    for (i, req) in test.requests().iter().enumerate() {
-        info!("Request {}: {:#?}", i, req);
+    for req in test.requests() {
+        let first_msg = req.messages().first().expect("first message should exist").text(); 
+        assert_contains(&first_msg, "Agent initialized");
     }
 }