152 lines
4.8 KiB
Python
152 lines
4.8 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
from enroll.diff import compare_harvests
|
|
|
|
|
|
def _write_bundle(root: Path, state: dict, artifacts: dict[str, bytes]) -> None:
|
|
root.mkdir(parents=True, exist_ok=True)
|
|
(root / "state.json").write_text(json.dumps(state, indent=2), encoding="utf-8")
|
|
for rel, data in artifacts.items():
|
|
p = root / rel
|
|
p.parent.mkdir(parents=True, exist_ok=True)
|
|
p.write_bytes(data)
|
|
|
|
|
|
def test_diff_includes_usr_local_custom_files(tmp_path: Path):
|
|
old = tmp_path / "old"
|
|
new = tmp_path / "new"
|
|
|
|
old_state = {
|
|
"schema_version": 3,
|
|
"host": {"hostname": "h1", "os": "debian", "pkg_backend": "dpkg"},
|
|
"inventory": {
|
|
"packages": {
|
|
"curl": {
|
|
"version": "1.0",
|
|
"arches": [],
|
|
"installations": [{"version": "1.0", "arch": "amd64"}],
|
|
"observed_via": [{"kind": "user_installed"}],
|
|
"roles": [],
|
|
}
|
|
}
|
|
},
|
|
"roles": {
|
|
"users": {
|
|
"role_name": "users",
|
|
"users": [],
|
|
"managed_files": [],
|
|
"excluded": [],
|
|
"notes": [],
|
|
},
|
|
"services": [],
|
|
"packages": [],
|
|
"apt_config": {
|
|
"role_name": "apt_config",
|
|
"managed_files": [],
|
|
"excluded": [],
|
|
"notes": [],
|
|
},
|
|
"etc_custom": {
|
|
"role_name": "etc_custom",
|
|
"managed_files": [],
|
|
"excluded": [],
|
|
"notes": [],
|
|
},
|
|
"usr_local_custom": {
|
|
"role_name": "usr_local_custom",
|
|
"managed_files": [
|
|
{
|
|
"path": "/usr/local/etc/myapp.conf",
|
|
"src_rel": "usr/local/etc/myapp.conf",
|
|
"owner": "root",
|
|
"group": "root",
|
|
"mode": "0644",
|
|
"reason": "usr_local_etc_custom",
|
|
}
|
|
],
|
|
"excluded": [],
|
|
"notes": [],
|
|
},
|
|
"extra_paths": {
|
|
"role_name": "extra_paths",
|
|
"include_patterns": [],
|
|
"exclude_patterns": [],
|
|
"managed_files": [],
|
|
"excluded": [],
|
|
"notes": [],
|
|
},
|
|
},
|
|
}
|
|
|
|
new_state = {
|
|
**old_state,
|
|
"inventory": {
|
|
"packages": {
|
|
**old_state["inventory"]["packages"],
|
|
"htop": {
|
|
"version": "3.0",
|
|
"arches": [],
|
|
"installations": [{"version": "3.0", "arch": "amd64"}],
|
|
"observed_via": [{"kind": "user_installed"}],
|
|
"roles": [],
|
|
},
|
|
}
|
|
},
|
|
"roles": {
|
|
**old_state["roles"],
|
|
"usr_local_custom": {
|
|
"role_name": "usr_local_custom",
|
|
"managed_files": [
|
|
{
|
|
"path": "/usr/local/etc/myapp.conf",
|
|
"src_rel": "usr/local/etc/myapp.conf",
|
|
"owner": "root",
|
|
"group": "root",
|
|
"mode": "0644",
|
|
"reason": "usr_local_etc_custom",
|
|
},
|
|
{
|
|
"path": "/usr/local/bin/myscript",
|
|
"src_rel": "usr/local/bin/myscript",
|
|
"owner": "root",
|
|
"group": "root",
|
|
"mode": "0755",
|
|
"reason": "usr_local_bin_script",
|
|
},
|
|
],
|
|
"excluded": [],
|
|
"notes": [],
|
|
},
|
|
},
|
|
}
|
|
|
|
_write_bundle(
|
|
old,
|
|
old_state,
|
|
{
|
|
"artifacts/usr_local_custom/usr/local/etc/myapp.conf": b"myapp=1\n",
|
|
},
|
|
)
|
|
_write_bundle(
|
|
new,
|
|
new_state,
|
|
{
|
|
"artifacts/usr_local_custom/usr/local/etc/myapp.conf": b"myapp=2\n",
|
|
"artifacts/usr_local_custom/usr/local/bin/myscript": b"#!/bin/sh\necho hi\n",
|
|
},
|
|
)
|
|
|
|
report, has_changes = compare_harvests(str(old), str(new))
|
|
assert has_changes is True
|
|
|
|
# Packages: htop was added.
|
|
assert report["packages"]["added"] == ["htop"]
|
|
|
|
# Files: /usr/local/etc/myapp.conf should be detected as changed (content sha differs).
|
|
changed_paths = {c["path"] for c in report["files"]["changed"]}
|
|
assert "/usr/local/etc/myapp.conf" in changed_paths
|
|
|
|
# Files: new script was added.
|
|
added_paths = {a["path"] for a in report["files"]["added"]}
|
|
assert "/usr/local/bin/myscript" in added_paths
|