use super::*;
use pretty_assertions::assert_eq;
use tempfile::tempdir;
#[tokio::test]
async fn removes_banned_allow_rules_once() {
const BANNED_PREFIXES: &[&[&str]] = &[
&["cmd.exe", "/k"],
&["git"],
&["pwsh", "-ec"],
&["pwsh", "-f"],
];
let codex_home = tempdir().expect("create codex home");
let policy_path = codex_home.path().join("rules/default.rules");
std::fs::create_dir_all(policy_path.parent().expect("rules directory"))
.expect("create rules directory");
std::fs::write(
&policy_path,
r#"prefix_rule(pattern=["git"], decision="allow")
prefix_rule(pattern=["git"], decision="prompt")
prefix_rule(pattern=["git"], decision="deny")
prefix_rule(pattern=["git", "status"], decision="allow")
prefix_rule(pattern=["CMD.EXE", "/K"], decision="allow")
prefix_rule(pattern=["PWSH", "-EC"], decision="allow")
prefix_rule(pattern=["PwSh", "-F"], decision="allow")
network_rule(host="api.github.com", protocol="https", decision="allow")
"#,
)
.expect("write legacy policy");
prefix_rule_migration(codex_home.path(), &policy_path, BANNED_PREFIXES)
.await
.expect("run sandbox migration");
assert_eq!(
std::fs::read_to_string(&policy_path).expect("read migrated policy"),
r#"prefix_rule(pattern=["git"], decision="prompt")
prefix_rule(pattern=["git"], decision="deny")
prefix_rule(pattern=["git", "status"], decision="allow")
network_rule(host="api.github.com", protocol="https", decision="allow")
"#
);
assert_eq!(
std::fs::read_to_string(codex_home.path().join(MIGRATION_MARKER_FILENAME))
.expect("read migration marker"),
"v1\n"
);
let post_migration_policy = r#"prefix_rule(pattern=["git"], decision="allow")
"#;
std::fs::write(&policy_path, post_migration_policy).expect("write post-migration policy");
prefix_rule_migration(codex_home.path(), &policy_path, BANNED_PREFIXES)
.await
.expect("rerun sandbox migration");
assert_eq!(
std::fs::read_to_string(&policy_path).expect("read post-migration policy"),
post_migration_policy
);
}