Fix tests
This commit is contained in:
parent
737316d507
commit
4146aa997b
4 changed files with 22 additions and 1 deletions
6
tests.sh
6
tests.sh
|
|
@ -2,6 +2,11 @@
|
||||||
|
|
||||||
set -Eeuo pipefail
|
set -Eeuo pipefail
|
||||||
|
|
||||||
|
# These tests intentionally run as root and exercise Enroll's root-output
|
||||||
|
# safety checks. Keep test-created directories private even on Forgejo/Docker
|
||||||
|
# images that default to a permissive umask such as 0002.
|
||||||
|
umask 077
|
||||||
|
|
||||||
PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
TMP_PARENT="${TMPDIR:-/tmp}"
|
TMP_PARENT="${TMPDIR:-/tmp}"
|
||||||
KEEP_WORKDIR=0
|
KEEP_WORKDIR=0
|
||||||
|
|
@ -9,6 +14,7 @@ if [[ -n "${ENROLL_TEST_WORKDIR:-}" ]]; then
|
||||||
WORK_DIR="${ENROLL_TEST_WORKDIR}"
|
WORK_DIR="${ENROLL_TEST_WORKDIR}"
|
||||||
KEEP_WORKDIR=1
|
KEEP_WORKDIR=1
|
||||||
mkdir -p "${WORK_DIR}"
|
mkdir -p "${WORK_DIR}"
|
||||||
|
chmod 700 "${WORK_DIR}" || true
|
||||||
else
|
else
|
||||||
WORK_DIR="$(mktemp -d "${TMP_PARENT%/}/enroll-tests.XXXXXX")"
|
WORK_DIR="$(mktemp -d "${TMP_PARENT%/}/enroll-tests.XXXXXX")"
|
||||||
fi
|
fi
|
||||||
|
|
|
||||||
|
|
@ -645,7 +645,9 @@ def test_cli_harvest_remote_sops_encrypts_and_prints_path(
|
||||||
assert calls.get("encrypt")
|
assert calls.get("encrypt")
|
||||||
|
|
||||||
|
|
||||||
def test_cli_harvest_remote_password_required_exits_cleanly(monkeypatch):
|
def test_cli_harvest_remote_password_required_exits_cleanly(
|
||||||
|
tmp_path: Path, monkeypatch
|
||||||
|
):
|
||||||
def boom(**kwargs):
|
def boom(**kwargs):
|
||||||
raise RemoteSudoPasswordRequired("pw required")
|
raise RemoteSudoPasswordRequired("pw required")
|
||||||
|
|
||||||
|
|
@ -660,6 +662,8 @@ def test_cli_harvest_remote_password_required_exits_cleanly(monkeypatch):
|
||||||
"example.com",
|
"example.com",
|
||||||
"--remote-user",
|
"--remote-user",
|
||||||
"root",
|
"root",
|
||||||
|
"--out",
|
||||||
|
str(tmp_path / "remote-harvest"),
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
with pytest.raises(SystemExit) as e:
|
with pytest.raises(SystemExit) as e:
|
||||||
|
|
|
||||||
|
|
@ -969,6 +969,10 @@ def test_manifest_fqdn_existing_output_rejects_symlink_component(tmp_path: Path)
|
||||||
_write_state(bundle, state)
|
_write_state(bundle, state)
|
||||||
|
|
||||||
(out / "inventory").mkdir(parents=True)
|
(out / "inventory").mkdir(parents=True)
|
||||||
|
# The output path itself must be root-safe so this test reaches the
|
||||||
|
# intended nested-symlink guard even when CI/root uses a permissive umask.
|
||||||
|
out.chmod(0o700)
|
||||||
|
(out / "inventory").chmod(0o700)
|
||||||
escape.mkdir()
|
escape.mkdir()
|
||||||
(out / "inventory" / "host_vars").symlink_to(escape, target_is_directory=True)
|
(out / "inventory" / "host_vars").symlink_to(escape, target_is_directory=True)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -36,7 +36,11 @@ def test_prepare_manifest_output_dir_allows_existing_clean_tree_in_site_mode(
|
||||||
):
|
):
|
||||||
out = tmp_path / "site"
|
out = tmp_path / "site"
|
||||||
out.mkdir()
|
out.mkdir()
|
||||||
|
# Match Enroll's root-run output safety expectations regardless of the
|
||||||
|
# ambient CI/container umask.
|
||||||
|
out.chmod(0o700)
|
||||||
(out / ".git").mkdir()
|
(out / ".git").mkdir()
|
||||||
|
(out / ".git").chmod(0o700)
|
||||||
(out / ".git" / "ignored-link").symlink_to(tmp_path, target_is_directory=True)
|
(out / ".git" / "ignored-link").symlink_to(tmp_path, target_is_directory=True)
|
||||||
|
|
||||||
assert prepare_manifest_output_dir(out, allow_existing=True) == out
|
assert prepare_manifest_output_dir(out, allow_existing=True) == out
|
||||||
|
|
@ -45,6 +49,9 @@ def test_prepare_manifest_output_dir_allows_existing_clean_tree_in_site_mode(
|
||||||
def test_prepare_manifest_output_dir_rejects_existing_tree_symlink(tmp_path: Path):
|
def test_prepare_manifest_output_dir_rejects_existing_tree_symlink(tmp_path: Path):
|
||||||
out = tmp_path / "site"
|
out = tmp_path / "site"
|
||||||
out.mkdir()
|
out.mkdir()
|
||||||
|
# Keep the root-safety check from masking the specific symlink assertion on
|
||||||
|
# Forgejo/Docker hosts that run with umask 0002.
|
||||||
|
out.chmod(0o700)
|
||||||
(out / "bad-link").symlink_to(tmp_path, target_is_directory=True)
|
(out / "bad-link").symlink_to(tmp_path, target_is_directory=True)
|
||||||
|
|
||||||
with pytest.raises(ManifestOutputError, match="contains a symlink"):
|
with pytest.raises(ManifestOutputError, match="contains a symlink"):
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue