beginning of userSubmitPrompt persistance test

Author: ekang7

crates/agent/tests/common/mod.rs

@@ -47,8 +47,8 @@ use serde::Serialize;
 
 type MockResponseStreams = Vec<Vec<StreamResult>>;
 
-#[derive(Default)]
 pub struct TestCaseBuilder {
+    test_base: TestBase,
     test_name: Option<String>,
     agent_config: Option<AgentConfig>,
     files: Vec<Box<dyn TestFile>>,
@@ -58,6 +58,10 @@ pub struct TestCaseBuilder {
 }
 
 impl TestCaseBuilder {
+    pub async fn new() -> Self {
+Self { test_base: TestBase::new().await, test_name: Default::default(), agent_config: todo!(), files: todo!(), mock_responses: todo!(), trust_all_tools: todo!(), tool_use_approvals: todo!() }
+    }
+
     pub fn test_name<'a>(mut self, name: impl Into<Cow<'a, str>>) -> Self {
         self.test_name = Some(name.into().to_string());
         self
@@ -92,6 +96,10 @@ impl TestCaseBuilder {
         self
     }
 
+    pub fn with_test_perprompt_hook(mut self) -> Self {
+        
+    }
+
     pub async fn build(self) -> Result<TestCase> {
         let snapshot = AgentSnapshot::new_empty(self.agent_config.unwrap_or_default());
 
@@ -102,7 +110,6 @@ impl TestCaseBuilder {
 
         let mut agent = Agent::new(snapshot, Arc::new(model), McpManager::new().spawn()).await?;
 
-        let mut test_base = TestBase::new().await;
         for file in self.files {
             test_base = test_base.with_file(file).await;
         }

crates/agent/tests/test_hooks.rs

@@ -120,4 +120,54 @@ async fn test_user_prompt_hook_single() {
     let req = test.requests().first().expect("should have one request");
     let first_msg = req.messages().first().expect("first message should exist").text(); 
     assert_contains(&first_msg, "submitted!");
-}
+}
+
+#[tokio::test]
+async fn test_user_prompt_hook_persistence() {
+    let _ = tracing_subscriber::fmt::try_init();
+    
+    let hooks = HashMap::from([
+        (
+            HookTrigger::UserPromptSubmit,
+            vec![HookConfig::ShellCommand(CommandHook {
+                command: "printf 'a' >> turns.txt; printf \"char count: $(wc -c < turns.txt | tr -d ' ')\"".to_string(),
+                opts: BaseHookConfig {
+                    timeout_ms: 5000,
+                    max_output_size: 1024,
+                    cache_ttl_seconds: 0,
+                    matcher: None,
+                },
+            })],
+        ),
+    ]);
+
+    let agent_config = AgentConfig::V2025_08_22(AgentConfigV2025_08_22 {
+        hooks,
+        ..Default::default()
+    });
+    let mut test: TestCase = TestCase::builder()
+        .test_name("user prompt submit hook behavior for single shell command and one turn")
+        .with_agent_config(agent_config)
+        .with_test_perprompt_hook()
+        .with_file(("turns.txt", ""))
+        .with_responses(
+            parse_response_streams(include_str!("./mock_responses/simple_two_turn.jsonl"))
+                .await
+                .unwrap(),
+        )
+        .build()
+        .await
+        .unwrap();
+    
+    test.wait_until_agent_initializes(Duration::from_millis(100)).await;
+    test.send_prompt("hello".to_string()).await;
+    test.wait_until_agent_stop(Duration::from_millis(100)).await;
+    test.send_prompt("bye".to_string()).await;
+    test.wait_until_agent_stop(Duration::from_millis(100)).await;
+    let req = test.requests().first().expect("first request should exist");
+    let first_msg = req.messages().first().expect("first message should exist").text(); 
+    assert_contains(&first_msg, "char count: 1"); 
+    let req = test.requests().get(1).expect("second request should exist"); 
+    let first_msg = req.messages().first().expect("first message should exist").text(); 
+    assert_contains(&first_msg, "char count: 2"); 
+}