helpers.py

tests/helpers.py

1. """Shared test helpers: load a gate module fresh with its env, and copy the
2. sample vaults into a temp dir so tests can edit, delete and symlink freely."""
3. import contextlib
4. import importlib.util
5. import io
6. import itertools
7. import os
8. import shutil
9. import tempfile
10. from pathlib import Path
12. REPO = Path(__file__).resolve().parent.parent
13. SAMPLE = REPO / "sample-vaults"
14. _counter = itertools.count()
17. def load(script: str, env: dict):
18. """Import REPO/<script> as a fresh module with `env` applied. The gates read
19. their configuration from the environment at import time."""
20. old = {k: os.environ.get(k) for k in env}
21. os.environ.update(env)
22. try:
23. spec = importlib.util.spec_from_file_location(
24. f"_bm_{Path(script).stem}_{next(_counter)}", REPO / script)
25. mod = importlib.util.module_from_spec(spec)
26. spec.loader.exec_module(mod)
27. return mod
28. finally:
29. for k, v in old.items():
30. if v is None:
31. os.environ.pop(k, None)
32. else:
33. os.environ[k] = v
36. def run(mod) -> tuple[int, str]:
37. """Run mod.main(), capturing stdout."""
38. buf = io.StringIO()
39. with contextlib.redirect_stdout(buf):
40. code = mod.main()
41. return code, buf.getvalue()
44. class TempVaults:
45. """A throwaway copy of the sample vaults plus a staging root."""
47. def __init__(self):
48. self.dir = Path(tempfile.mkdtemp(prefix="bm-test-"))
49. self.vaults = self.dir / "vaults"
50. shutil.copytree(SAMPLE, self.vaults)
51. # A .git folder inside a nested vault cannot be committed to this repo,
52. # so the fixture creates it here.
53. git = self.vaults / "workbench" / "Efforts" / "garden-studio" / ".git"
54. git.mkdir(parents=True)
55. (git / "notes.md").write_text("# git internals\n\nWALLED at any depth.\n")
56. self.workbench = self.vaults / "workbench"
57. self.identity = self.vaults / "identity"
58. self.staging_root = self.dir / "cowork-staging"
60. @property
61. def env(self) -> dict:
62. return {"VAULT_ROOTS": f"{self.workbench}:{self.identity}",
63. "STAGING_DIR": str(self.staging_root), "EXTRA_DENY": ""}
65. @property
66. def staged(self) -> list[str]:
67. d = self.staging_root / "staging"
68. return sorted(p.name for p in d.iterdir()) if d.exists() else []
70. def gate(self):
71. return load("gate.py", self.env)
73. def cleanup(self):
74. shutil.rmtree(self.dir, ignore_errors=True)