53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
from zerto_rewind_mcp.catalog import CatalogEntry, MutatingCatalog, entry_from_dict
|
|
|
|
|
|
def test_add_and_get(tmp_path: Path):
|
|
path = tmp_path / "config.json"
|
|
path.write_text(
|
|
json.dumps({"zerto_url": "https://zvm", "mutating_tools": []}),
|
|
encoding="utf-8",
|
|
)
|
|
cat = MutatingCatalog(path=path)
|
|
cat.add(CatalogEntry(server="ssh", tool="exec", vm_arg="host", notes="guest shell"))
|
|
found = cat.get("SSH", "Exec")
|
|
assert found is not None
|
|
assert found.vm_arg == "host"
|
|
saved = json.loads(path.read_text(encoding="utf-8"))
|
|
assert saved["zerto_url"] == "https://zvm"
|
|
assert saved["mutating_tools"][0]["tool"] == "exec"
|
|
|
|
|
|
def test_unlisted_is_missing():
|
|
cat = MutatingCatalog()
|
|
assert cat.get("foo", "bar") is None
|
|
|
|
|
|
def test_entry_requires_fields():
|
|
try:
|
|
entry_from_dict({"server": "ssh"})
|
|
raise AssertionError("expected ValueError")
|
|
except ValueError:
|
|
pass
|
|
|
|
|
|
def test_example_config_covers_windows_and_linux():
|
|
"""The starter catalog must not be Linux-only.
|
|
|
|
The catalog is opt-in: an unlisted tool passes through unguarded. A
|
|
Windows-only shop taking the shipped defaults would therefore mutate
|
|
protected guests with no checkpoint at all.
|
|
"""
|
|
example = Path(__file__).resolve().parent.parent / "config.example.json"
|
|
data = json.loads(example.read_text(encoding="utf-8"))
|
|
cat = MutatingCatalog.from_config(data)
|
|
assert cat.get("ssh", "exec") is not None
|
|
assert cat.get("winrm", "run_command") is not None
|
|
assert cat.get("winrm", "run_ps") is not None
|
|
assert cat.get("powershell", "invoke_command") is not None
|
|
# every entry must name the arg holding the VM, or the guard cannot resolve one
|
|
for entry in cat.list():
|
|
assert entry.vm_arg, f"{entry.server}/{entry.tool} has no vm_arg"
|