71 lines
2.3 KiB
Rust
71 lines
2.3 KiB
Rust
|
//! 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();
|
||
|
|
}
|