Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
51 changes: 33 additions & 18 deletions src/daemon.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8125,46 +8125,61 @@ mod tests {
}

#[test]
#[serial]
fn checkpoint_requests_use_long_timeout_in_ci_or_test_env() {
let _unset_ci = EnvVarGuard::unset("CI");
let _unset_legacy_test = EnvVarGuard::unset("GITAI_TEST_DB_PATH");
let _test_db = EnvVarGuard::set("GIT_AI_TEST_DB_PATH", "/tmp/git-ai-test.db");

assert_eq!(
control_request_response_timeout(&queued_checkpoint_request()),
checkpoint_control_response_timeout(&queued_checkpoint_request(), true),
DAEMON_CHECKPOINT_RESPONSE_TIMEOUT
);
assert_eq!(
control_request_response_timeout(&waited_checkpoint_request()),
checkpoint_control_response_timeout(&waited_checkpoint_request(), true),
DAEMON_CHECKPOINT_RESPONSE_TIMEOUT
);
}

#[test]
#[serial]
fn queued_checkpoint_requests_use_short_timeout_in_product_env() {
let _unset_ci = EnvVarGuard::unset("CI");
let _unset_test = EnvVarGuard::unset("GIT_AI_TEST_DB_PATH");
let _unset_legacy_test = EnvVarGuard::unset("GITAI_TEST_DB_PATH");

assert_eq!(
control_request_response_timeout(&queued_checkpoint_request()),
checkpoint_control_response_timeout(&queued_checkpoint_request(), false),
DAEMON_CONTROL_RESPONSE_TIMEOUT
);
}

#[test]
#[serial]
fn waited_checkpoint_requests_use_long_timeout_in_product_env() {
assert_eq!(
checkpoint_control_response_timeout(&waited_checkpoint_request(), false),
DAEMON_CHECKPOINT_RESPONSE_TIMEOUT
);
}

#[test]
#[serial]
fn checkpoint_control_timeout_uses_ci_env_var() {
let _unset_test = EnvVarGuard::unset("GIT_AI_TEST_DB_PATH");
let _unset_legacy_test = EnvVarGuard::unset("GITAI_TEST_DB_PATH");
let _set_ci = EnvVarGuard::set("CI", "true");

assert!(checkpoint_control_timeout_uses_ci_or_test_budget());
}

#[test]
#[serial]
fn checkpoint_control_timeout_uses_test_db_env_var() {
let _unset_ci = EnvVarGuard::unset("CI");
let _unset_legacy_test = EnvVarGuard::unset("GITAI_TEST_DB_PATH");
let _set_test = EnvVarGuard::set("GIT_AI_TEST_DB_PATH", "/tmp/git-ai-test.db");

assert!(checkpoint_control_timeout_uses_ci_or_test_budget());
}

#[test]
#[serial]
fn checkpoint_control_timeout_false_when_no_ci_or_test_vars() {
let _unset_ci = EnvVarGuard::unset("CI");
let _unset_test = EnvVarGuard::unset("GIT_AI_TEST_DB_PATH");
let _unset_legacy_test = EnvVarGuard::unset("GITAI_TEST_DB_PATH");

assert_eq!(
control_request_response_timeout(&waited_checkpoint_request()),
DAEMON_CHECKPOINT_RESPONSE_TIMEOUT
);
assert!(!checkpoint_control_timeout_uses_ci_or_test_budget());
}

#[test]
Expand Down
7 changes: 7 additions & 0 deletions tests/integration/repos/test_repo.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2324,6 +2324,13 @@ impl TestRepo {
.current_dir(&absolute_working_dir);
self.configure_git_ai_env(&mut command);

// Do NOT delegate checkpoints to the daemon drain here. The caller's
// CWD may differ from self's repo root (cross-repo / subrepo tests),
// so the completion log family key is unpredictable and any async wait
// would target the wrong family. Running synchronously means the
// working log is guaranteed to be written before we return.
command.env_remove("GIT_AI_DAEMON_CHECKPOINT_DELEGATE");

if let Some(patch) = &self.config_patch
&& let Ok(patch_json) = serde_json::to_string(patch)
{
Expand Down
Loading