huskies: merge 783

server/src/agents/pool/auto_assign/watchdog/tests/mod.rs

@@ -0,0 +1,70 @@
+//! Shared test helpers for the watchdog module.
+
+use std::path::Path;
+
+mod limits_tests;
+mod orphan_tests;
+
+/// Write a fake session log file with `n` assistant turn entries.
+///
+/// The file is named `{agent_name}-{session_id}.log` to match the
+/// real naming convention used by `AgentLogWriter`.
+pub(super) fn write_fake_session_log(
+    project_root: &Path,
+    story_id: &str,
+    agent_name: &str,
+    session_id: &str,
+    n_turns: u64,
+) {
+    let log_dir = project_root.join(".huskies").join("logs").join(story_id);
+    std::fs::create_dir_all(&log_dir).unwrap();
+    let log_path = log_dir.join(format!("{agent_name}-{session_id}.log"));
+    let mut content = String::new();
+    for _ in 0..n_turns {
+        content.push_str(
+            &serde_json::to_string(&serde_json::json!({
+                "timestamp": "2026-04-25T00:00:00Z",
+                "type": "agent_json",
+                "story_id": story_id,
+                "agent_name": agent_name,
+                "data": { "type": "assistant", "message": {} }
+            }))
+            .unwrap(),
+        );
+        content.push('\n');
+    }
+    std::fs::write(log_path, content).unwrap();
+}
+
+/// Write a fake session log containing a `result` event with the given cost.
+///
+/// Used to test budget enforcement via the watchdog's per-session log
+/// reading (not `token_usage.jsonl`).
+pub(super) fn write_fake_budget_session_log(
+    project_root: &Path,
+    story_id: &str,
+    agent_name: &str,
+    session_id: &str,
+    cost_usd: f64,
+) {
+    let log_dir = project_root.join(".huskies").join("logs").join(story_id);
+    std::fs::create_dir_all(&log_dir).unwrap();
+    let log_path = log_dir.join(format!("{agent_name}-{session_id}.log"));
+    let content = serde_json::to_string(&serde_json::json!({
+        "timestamp": "2026-04-25T00:00:00Z",
+        "type": "agent_json",
+        "story_id": story_id,
+        "agent_name": agent_name,
+        "data": { "type": "result", "total_cost_usd": cost_usd }
+    }))
+    .unwrap()
+        + "\n";
+    std::fs::write(log_path, content).unwrap();
+}
+
+/// Write a minimal project.toml with the given agent config.
+pub(super) fn write_project_config(project_root: &Path, config_toml: &str) {
+    let huskies_dir = project_root.join(".huskies");
+    std::fs::create_dir_all(&huskies_dir).unwrap();
+    std::fs::write(huskies_dir.join("project.toml"), config_toml).unwrap();
+}