use std::path::Path;
use assert_cmd::Command;
use predicates::prelude::*;
fn rigger(data: &Path) -> Command {
let mut cmd = Command::cargo_bin("rigger").unwrap();
cmd.env("RIGGER_DATA_DIR", data);
cmd
}
fn repo(root: &Path) {
std::fs::create_dir_all(root).unwrap();
std::fs::write(root.join("Cargo.toml"), "[package]\nname = \"declared-elsewhere\"\nversion = \"0.1.0\"\n").unwrap();
}
#[test]
fn version_prints_the_crate_version() {
Command::cargo_bin("rigger")
.unwrap()
.arg("--version")
.assert()
.success()
.stdout(predicate::str::contains(env!("CARGO_PKG_VERSION")));
}
#[test]
fn help_describes_the_product() {
Command::cargo_bin("rigger")
.unwrap()
.arg("--help")
.assert()
.success()
.stdout(predicate::str::contains("One seat for all your projects and tasks"));
}
#[test]
fn init_creates_the_database_and_is_idempotent() {
let data = tempfile::tempdir().unwrap();
rigger(data.path())
.arg("init")
.assert()
.success()
.stdout(predicate::str::contains("Created").and(predicate::str::contains("schema version 1")));
assert!(data.path().join("rigger.db").exists());
rigger(data.path())
.arg("init")
.assert()
.success()
.stdout(predicate::str::contains("Already initialised"));
}
#[test]
fn commands_before_init_point_at_init() {
let data = tempfile::tempdir().unwrap();
rigger(data.path())
.args(["project", "list"])
.assert()
.failure()
.stderr(predicate::str::contains("run `rigger init` first"));
}
#[test]
fn project_add_names_the_project_after_its_directory() {
let data = tempfile::tempdir().unwrap();
let root = data.path().join("repos").join("folder-name");
repo(&root);
rigger(data.path()).arg("init").assert().success();
rigger(data.path())
.args(["project", "add"])
.arg(&root)
.assert()
.success()
.stdout(predicate::str::contains("Recorded 'folder-name'").and(predicate::str::contains("remote: none")));
rigger(data.path())
.args(["project", "list"])
.assert()
.success()
.stdout(predicate::str::starts_with("folder-name"));
}
#[test]
fn project_add_reads_the_origin_remote() {
let data = tempfile::tempdir().unwrap();
let root = data.path().join("remoted");
repo(&root);
std::fs::create_dir(root.join(".git")).unwrap();
std::fs::write(
root.join(".git").join("config"),
"[remote \"origin\"]\n\turl = https://example.com/remoted.git\n",
)
.unwrap();
rigger(data.path()).arg("init").assert().success();
rigger(data.path())
.args(["project", "add"])
.arg(&root)
.assert()
.success()
.stdout(predicate::str::contains("remote: https://example.com/remoted.git"));
rigger(data.path())
.args(["project", "show", "remoted", "--json"])
.assert()
.success()
.stdout(predicate::str::contains("\"remote\": \"https://example.com/remoted.git\""));
}
#[test]
fn name_can_be_overridden_and_must_be_unique() {
let data = tempfile::tempdir().unwrap();
let first = data.path().join("a").join("same");
let second = data.path().join("b").join("same");
repo(&first);
repo(&second);
rigger(data.path()).arg("init").assert().success();
rigger(data.path()).args(["project", "add"]).arg(&first).assert().success();
rigger(data.path())
.args(["project", "add"])
.arg(&second)
.assert()
.failure()
.stderr(predicate::str::contains("a project named 'same' already exists"));
rigger(data.path())
.args(["project", "add", "--name", "other"])
.arg(&second)
.assert()
.success()
.stdout(predicate::str::contains("Recorded 'other'"));
}
#[test]
fn the_same_path_is_not_recorded_twice() {
let data = tempfile::tempdir().unwrap();
let root = data.path().join("once");
repo(&root);
rigger(data.path()).arg("init").assert().success();
rigger(data.path()).args(["project", "add"]).arg(&root).assert().success();
rigger(data.path())
.args(["project", "add"])
.arg(&root)
.assert()
.failure()
.stderr(predicate::str::contains("already recorded as project 'once'"));
}
#[test]
fn show_of_an_unknown_project_fails_with_a_pointer() {
let data = tempfile::tempdir().unwrap();
rigger(data.path()).arg("init").assert().success();
rigger(data.path())
.args(["project", "show", "ghost"])
.assert()
.failure()
.stderr(predicate::str::contains("no project named 'ghost'"));
}
fn hub(dir: &Path) {
std::fs::create_dir_all(dir).unwrap();
std::fs::write(
dir.join("План.md"),
"# План разработки\n\n## Ждёт решения владельца\n\n1. Pick a colour.\n2. Sign the binary.\n\n## v0.3.0 · Third stage\n\n- [ ] first task\n- [ ] second task\n\n## Бэклог без версии\n\n- [ ] someday, not a task of a stage\n",
)
.unwrap();
std::fs::write(
dir.join("Изменения.md"),
"# Изменения\n\n## v0.2.0 · Second stage — выпущена 2026-09-03\n\n- Something shipped.\n\n## v0.1.0 · First stage — выпущен 2026-09-01\n",
)
.unwrap();
std::fs::write(
dir.join("Решения.md"),
"# Журнал решений\n\n---\n\n## 2026-09-03 · The record is the database\n\nBecause prose cannot be filtered.\n",
)
.unwrap();
}
fn imported_project(data: &Path) -> std::path::PathBuf {
let root = data.join("proj");
repo(&root);
hub(&root.join("hub"));
rigger(data).arg("init").assert().success();
rigger(data).args(["project", "add"]).arg(&root).assert().success();
root
}
#[test]
fn import_reads_stages_tasks_decisions_and_questions() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path())
.args(["import", "proj", "--hub"])
.arg(root.join("hub"))
.assert()
.success()
.stdout(
predicate::str::contains("versions 3 added")
.and(predicate::str::contains("tasks 2 added"))
.and(predicate::str::contains("decisions 1 added"))
.and(predicate::str::contains("questions 2 added")),
);
rigger(data.path()).arg("doctor").assert().success().stdout(
predicate::str::contains("versions: 3")
.and(predicate::str::contains("tasks: 2"))
.and(predicate::str::contains("events: 3")),
);
}
#[test]
fn importing_an_unchanged_hub_again_changes_nothing() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
let hub_dir = root.join("hub");
rigger(data.path()).args(["import", "proj", "--hub"]).arg(&hub_dir).assert().success();
rigger(data.path())
.args(["import", "proj", "--hub"])
.arg(&hub_dir)
.assert()
.success()
.stdout(predicate::str::contains("nothing changed"));
rigger(data.path())
.arg("doctor")
.assert()
.success()
.stdout(predicate::str::contains("versions: 3").and(predicate::str::contains("events: 3")));
}
#[test]
fn a_stage_that_shipped_is_updated_rather_than_duplicated() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
let hub_dir = root.join("hub");
rigger(data.path()).args(["import", "proj", "--hub"]).arg(&hub_dir).assert().success();
std::fs::write(hub_dir.join("План.md"), "# План\n\n## Ждёт решения владельца\n\n- (пусто)\n").unwrap();
std::fs::write(
hub_dir.join("Изменения.md"),
"# Изменения\n\n## v0.3.0 · Third stage — выпущена 2026-09-05\n\n## v0.2.0 · Second stage — выпущена 2026-09-03\n\n## v0.1.0 · First stage — выпущен 2026-09-01\n",
)
.unwrap();
rigger(data.path())
.args(["import", "proj", "--hub"])
.arg(&hub_dir)
.assert()
.success()
.stdout(predicate::str::contains("versions 0 added, 1 updated"));
rigger(data.path())
.arg("doctor")
.assert()
.success()
.stdout(predicate::str::contains("versions: 3"));
}
#[test]
fn a_stage_renumbered_by_its_release_replaces_the_old_row() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
let hub_dir = root.join("hub");
std::fs::write(hub_dir.join("Изменения.md"), "# Изменения\n\n## v1.9 · Очередь — закрыт 2026-09-03\n").unwrap();
rigger(data.path()).args(["import", "proj", "--hub"]).arg(&hub_dir).assert().success();
std::fs::write(
hub_dir.join("Изменения.md"),
"# Изменения\n\n## v1.9 · Очередь — закрыт 2026-09-03, выпущен **v1.9.0**\n",
)
.unwrap();
rigger(data.path()).args(["import", "proj", "--hub"]).arg(&hub_dir).assert().success();
rigger(data.path())
.args(["context", "proj"])
.assert()
.success()
.stdout(predicate::str::contains("Last shipped: v1.9.0"));
rigger(data.path())
.arg("doctor")
.assert()
.success()
.stdout(predicate::str::contains("versions: 2"));
}
#[test]
fn the_newest_release_is_the_highest_number_not_the_last_row() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
let hub_dir = root.join("hub");
std::fs::write(
hub_dir.join("Изменения.md"),
"# Изменения\n\n## v0.10.0 · Tenth — выпущена 2026-09-02\n\n## v0.9.0 · Ninth — выпущена 2026-09-02\n\n## v0.1.0 · First — выпущена 2026-09-02\n",
)
.unwrap();
rigger(data.path()).args(["import", "proj", "--hub"]).arg(&hub_dir).assert().success();
rigger(data.path())
.args(["context", "proj"])
.assert()
.success()
.stdout(predicate::str::contains("Last shipped: v0.10.0"));
}
#[test]
fn import_of_a_missing_hub_reports_the_files_it_wanted() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path())
.args(["import", "proj", "--hub"])
.arg(root.join("no-such-hub"))
.assert()
.success()
.stdout(predicate::str::contains("План.md is missing").and(predicate::str::contains("nothing changed")));
}
#[test]
fn import_of_an_unknown_project_fails() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path())
.args(["import", "ghost", "--hub"])
.arg(root.join("hub"))
.assert()
.failure()
.stderr(predicate::str::contains("no project named 'ghost'"));
}
#[test]
fn context_shows_where_the_project_stands_and_what_is_next() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path()).args(["import", "proj", "--hub"]).arg(root.join("hub")).assert().success();
rigger(data.path()).args(["context", "proj"]).assert().success().stdout(
predicate::str::contains("Last shipped: v0.2.0 on 2026-09-03")
.and(predicate::str::contains("## Current stage: v0.3.0 · Third stage"))
.and(predicate::str::contains("- first task"))
.and(predicate::str::contains("## Waiting for the owner"))
.and(predicate::str::contains("Pick a colour."))
.and(predicate::str::contains("The record is the database")),
);
}
#[test]
fn a_recorded_note_reaches_the_packet_and_a_next_step_stands_alone() {
let data = tempfile::tempdir().unwrap();
imported_project(data.path());
rigger(data.path())
.args(["note", "proj", "The parser must take hubs as they are.", "--kind", "finding"])
.assert()
.success()
.stdout(predicate::str::contains("Recorded a finding for proj"));
rigger(data.path())
.args(["note", "proj", "Ship the importer next.", "--kind", "next"])
.assert()
.success();
rigger(data.path()).args(["wish", "proj", "Show days without a commit."]).assert().success();
rigger(data.path()).args(["context", "proj"]).assert().success().stdout(
predicate::str::contains("finding · The parser must take hubs as they are.")
.and(predicate::str::contains("## Wishes, not yet sorted"))
.and(predicate::str::contains("Show days without a commit."))
.and(predicate::str::contains("## Next step\nShip the importer next."))
.and(predicate::str::contains("next · Ship").not()),
);
}
fn packet_cost(data: &Path, args: &[&str]) -> (usize, String) {
let mut cmd = rigger(data);
cmd.args(["context", "proj", "--explain"]).args(args);
let text = String::from_utf8(cmd.output().unwrap().stdout).unwrap();
let total = text
.lines()
.find(|l| l.starts_with("total"))
.and_then(|l| l.split_whitespace().nth(1))
.and_then(|n| n.parse().ok())
.unwrap_or_else(|| panic!("no total line in:\n{text}"));
(total, text)
}
#[test]
fn the_packet_holds_the_budget_and_reports_what_it_dropped() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path()).args(["import", "proj", "--hub"]).arg(root.join("hub")).assert().success();
let body = "**A decision that took a paragraph.**\n\n".to_string() + &"Reasoning that runs on and on. ".repeat(40);
for i in 0..60 {
rigger(data.path())
.args(["note", "proj", &format!("{body} number {i} of sixty"), "--kind", "decision"])
.assert()
.success();
}
let (total, text) = packet_cost(data.path(), &[]);
assert!(total <= 3000, "packet is {total} tokens, over the budget:\n{text}");
let (total, text) = packet_cost(data.path(), &["--budget", "600"]);
assert!(total <= 600, "packet is {total} tokens, over the 600 asked for:\n{text}");
assert!(
text.contains("older events left out by the budget"),
"the packet did not say what it dropped:\n{text}"
);
}
#[test]
fn the_essentials_survive_an_impossible_budget() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path()).args(["import", "proj", "--hub"]).arg(root.join("hub")).assert().success();
let (_, text) = packet_cost(data.path(), &["--budget", "1"]);
assert!(text.contains("## Current stage: v0.3.0"), "{text}");
assert!(text.contains("## Waiting for the owner"), "{text}");
assert!(text.contains("left out by the budget"), "{text}");
}
#[test]
fn context_as_json_carries_the_same_facts() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path()).args(["import", "proj", "--hub"]).arg(root.join("hub")).assert().success();
rigger(data.path()).args(["context", "proj", "--json"]).assert().success().stdout(
predicate::str::contains("\"project\": \"proj\"")
.and(predicate::str::contains("\"version\": \"v0.3.0\""))
.and(predicate::str::contains("\"events_omitted\": 0")),
);
}
#[test]
fn open_print_shows_the_message_the_assistant_would_receive() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path()).args(["import", "proj", "--hub"]).arg(root.join("hub")).assert().success();
rigger(data.path()).args(["open", "proj", "--print"]).assert().success().stdout(
predicate::str::contains("This is where the project stands")
.and(predicate::str::contains("rigger note"))
.and(predicate::str::contains("## Current stage: v0.3.0 · Third stage")),
);
}
#[test]
fn open_runs_the_assistant_in_the_project_with_the_packet() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
let log = data.path().join("assistant.log");
let (stub, args) = stub_assistant(data.path(), &log, 0);
let mut cmd = rigger(data.path());
cmd.env(open_env(), format!("{stub} {args}"));
cmd.args(["open", "proj"]).assert().success();
let recorded = std::fs::read_to_string(&log).unwrap();
assert!(
recorded.contains("This is where the project stands"),
"the packet did not reach the assistant: {recorded}"
);
let cwd = recorded
.lines()
.find_map(|l| l.trim_start_matches('\u{feff}').strip_prefix("cwd="))
.unwrap_or_else(|| panic!("the assistant did not report where it ran: {recorded}"));
assert!(
cwd.to_lowercase().ends_with("proj"),
"the assistant did not run in the project: ran in {cwd}, expected {}",
root.display()
);
}
#[cfg(windows)]
#[test]
fn open_can_launch_a_batch_file_assistant() {
let data = tempfile::tempdir().unwrap();
imported_project(data.path());
let log = data.path().join("batch.log");
let script = data.path().join("assistant.cmd");
std::fs::write(
&script,
format!(
"@echo off\r\necho cwd=%CD%>\"{}\"\r\nfindstr /n \"^\">>\"{}\"\r\n",
log.display(),
log.display()
),
)
.unwrap();
let mut cmd = rigger(data.path());
cmd.env(open_env(), script.display().to_string());
cmd.args(["open", "proj"]).assert().success();
let recorded = std::fs::read_to_string(&log).unwrap();
assert!(
recorded.contains("This is where the project stands"),
"a .cmd assistant did not receive the packet: {recorded}"
);
}
#[cfg(windows)]
#[test]
fn a_batch_assistant_receives_every_line_of_the_packet() {
let data = tempfile::tempdir().unwrap();
let root = imported_project(data.path());
rigger(data.path()).args(["import", "proj", "--hub"]).arg(root.join("hub")).assert().success();
rigger(data.path())
.args(["note", "proj", "Runs `open <project>` & pipes a|b, then ^ escapes.", "--kind", "finding"])
.assert()
.success();
let log = data.path().join("meta.log");
let script = data.path().join("assistant.cmd");
std::fs::write(&script, format!("@echo off\r\nfindstr /n \"^\">\"{}\"\r\n", log.display())).unwrap();
let mut cmd = rigger(data.path());
cmd.env(open_env(), script.display().to_string());
cmd.args(["open", "proj"]).assert().success();
let recorded = std::fs::read_to_string(&log).unwrap();
assert!(
recorded.contains("`open <project>` & pipes a|b, then ^ escapes."),
"the packet did not arrive whole: {recorded}"
);
assert!(recorded.contains("## Current stage"), "the packet was cut short: {recorded}");
let strays: Vec<_> = std::fs::read_dir(data.path())
.unwrap()
.filter_map(|e| e.ok())
.map(|e| e.file_name().to_string_lossy().into_owned())
.filter(|n| n.contains("project") || n.contains("pipes"))
.collect();
assert!(strays.is_empty(), "cmd redirected into {strays:?}");
}
#[test]
fn open_passes_on_the_assistants_exit_status() {
let data = tempfile::tempdir().unwrap();
imported_project(data.path());
let log = data.path().join("assistant.log");
let (stub, args) = stub_assistant(data.path(), &log, 3);
let mut cmd = rigger(data.path());
cmd.env(open_env(), format!("{stub} {args}"));
cmd.args(["open", "proj"]).assert().code(3);
}
#[test]
fn open_reports_a_missing_assistant_by_name() {
let data = tempfile::tempdir().unwrap();
imported_project(data.path());
let mut cmd = rigger(data.path());
cmd.env(open_env(), "no-such-assistant-binary");
cmd.args(["open", "proj"])
.assert()
.failure()
.stderr(predicate::str::contains("cannot run `no-such-assistant-binary`"));
}
fn open_env() -> &'static str {
"RIGGER_ASSISTANT"
}
fn stub_assistant(dir: &Path, log: &Path, code: i32) -> (String, String) {
let log = log.display().to_string().replace('\\', "/");
if cfg!(windows) {
let script = dir.join("stub.ps1");
std::fs::write(
&script,
format!("\"cwd=$((Get-Location).Path)\" | Out-File -Append -Encoding utf8 '{log}'\n$args | Out-File -Append -Encoding utf8 '{log}'\nexit {code}\n"),
)
.unwrap();
(
"powershell.exe".to_string(),
format!("-NoProfile -ExecutionPolicy Bypass -File {}", script.display()),
)
} else {
let script = dir.join("stub.sh");
std::fs::write(
&script,
format!("#!/bin/sh\necho \"cwd=$(pwd)\" >> '{log}'\necho \"$@\" >> '{log}'\nexit {code}\n"),
)
.unwrap();
#[cfg(unix)]
{
use std::os::unix::fs::PermissionsExt;
std::fs::set_permissions(&script, std::fs::Permissions::from_mode(0o755)).unwrap();
}
("sh".to_string(), script.display().to_string())
}
}
#[test]
fn backup_copies_the_database_beside_itself() {
let data = tempfile::tempdir().unwrap();
rigger(data.path()).arg("init").assert().success();
rigger(data.path())
.arg("backup")
.assert()
.success()
.stdout(predicate::str::contains("Copied to"));
let copies: Vec<_> = std::fs::read_dir(data.path())
.unwrap()
.filter_map(|e| e.ok())
.filter(|e| e.file_name().to_string_lossy().ends_with(".bak"))
.collect();
assert_eq!(copies.len(), 1, "expected one backup, found {copies:?}");
assert!(copies[0].file_name().to_string_lossy().starts_with("rigger.v1-"));
}
#[test]
fn doctor_reports_the_database_before_and_after_init() {
let data = tempfile::tempdir().unwrap();
rigger(data.path())
.arg("doctor")
.assert()
.success()
.stdout(predicate::str::contains("missing - run `rigger init`"));
let root = data.path().join("one");
repo(&root);
rigger(data.path()).arg("init").assert().success();
rigger(data.path()).args(["project", "add"]).arg(&root).assert().success();
rigger(data.path())
.arg("doctor")
.assert()
.success()
.stdout(predicate::str::contains("schema: version 1").and(predicate::str::contains("projects: 1")));
rigger(data.path())
.args(["doctor", "--json"])
.assert()
.success()
.stdout(predicate::str::contains("\"schema_version\": 1"));
}