44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import json
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
SCRIPT = Path(__file__).resolve().parents[2] / "scripts" / "import_controller_bootstrap.py"
|
|
SPEC = importlib.util.spec_from_file_location("controller_bootstrap_import_host", SCRIPT)
|
|
assert SPEC is not None and SPEC.loader is not None
|
|
host_import = importlib.util.module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(host_import)
|
|
|
|
|
|
def test_reads_root_private_bootstrap_without_newline(tmp_path, monkeypatch):
|
|
path = tmp_path / "bootstrap"
|
|
path.write_bytes(b"x" * 64 + b"\n")
|
|
path.chmod(0o600)
|
|
monkeypatch.setattr(host_import.os, "geteuid", lambda: 0, raising=False)
|
|
assert host_import.read_bootstrap(path) == b"x" * 64
|
|
path.write_bytes(b"short\n")
|
|
with pytest.raises(ValueError):
|
|
host_import.read_bootstrap(path)
|
|
|
|
|
|
def test_bootstrap_passes_only_on_stdin(monkeypatch):
|
|
observed = {}
|
|
|
|
def fake_run(argv, *, input, capture_output, check, timeout):
|
|
observed.update({"argv": argv, "input": input, "capture_output": capture_output})
|
|
|
|
class Result:
|
|
returncode = 0
|
|
stdout = json.dumps({"status": "imported", "secretRef": "ENROLLMENT_BOOTSTRAP"}).encode()
|
|
|
|
return Result()
|
|
|
|
monkeypatch.setattr(host_import.subprocess, "run", fake_run)
|
|
assert host_import.import_bootstrap(b"sensitive-test-value") == "imported"
|
|
assert observed["input"] == b"sensitive-test-value"
|
|
assert "sensitive-test-value" not in " ".join(observed["argv"])
|
|
assert observed["capture_output"] is True
|