mattrguerra.comHouston area
Matt Guerra
  1. 01AI & Automation
  2. 02Photography
  3. 03Design
  4. 04About
  5. 05Notes
  6. 06Contact
Back to the systems

mattrguerra/better-memory-pipelinemainPrivate on GitHub, published here in full

test_gate.py

Python7.6 KB174 lines

tests/test_gate.py

  1. """gate.py: the walls, the self-check, change detection and the scaffolding rules,
  2. run against a copy of the invented sample vaults."""
  3. import json
  4. import os
  5. import re
  6. import unittest
  8. from helpers import TempVaults, load, run
  10. WALLED_MARKERS = ("WALLED", "sk-FAKE")
  13. class GateTest(unittest.TestCase):
  14. def setUp(self):
  15. self.t = TempVaults()
  17. def tearDown(self):
  18. self.t.cleanup()
  20. def stage(self):
  21. return run(self.t.gate())
  23. def counts(self, out):
  24. return {k: int(v) for k, v in re.findall(r"(\w[\w-]*)=(\d+)", out)}
  26. # ---- the walls ---------------------------------------------------------
  27. def test_no_walled_file_reaches_staging(self):
  28. self.stage()
  29. staged = self.t.staged
  30. for name in staged:
  31. low = name.lower()
  32. for wall in ("_private", "efforts__finance", "efforts__journal", ".obsidian",
  33. ".trash", ".git", ".basic-memory"):
  34. self.assertNotIn(wall, low, name)
  35. text = (self.t.staging_root / "staging" / name).read_text()
  36. for marker in WALLED_MARKERS:
  37. self.assertNotIn(marker, text, name)
  39. def test_only_note_suffixes_stage(self):
  40. self.stage()
  41. self.assertIn("workbench__inbox.txt", self.t.staged)
  42. self.assertFalse([n for n in self.t.staged if n.endswith(".png") or n.endswith(".json")])
  44. def test_self_check_refuses_when_a_wall_is_weakened(self):
  45. gate = self.t.gate()
  46. gate.DENY_PREFIXES = [p for p in gate.DENY_PREFIXES if p != "efforts/finance"]
  47. with self.assertRaises(SystemExit) as cm:
  48. run(gate)
  49. self.assertIn("DENYLIST FAILED", str(cm.exception))
  50. self.assertEqual(self.t.staged, []) # refused before touching staging
  52. def test_traversal_is_denied(self):
  53. gate = self.t.gate()
  54. for rel in ("../identity/Rowan.md", "Efforts/../../identity/Rowan.md", "/../x.md"):
  55. with self.assertRaises(gate.PathDenied):
  56. gate._resolve_safe(self.t.workbench.resolve(), rel)
  58. def test_symlink_out_of_the_vault_is_denied(self):
  59. outside = self.t.dir / "outside.md"
  60. outside.write_text("# Outside\n\nWALLED: a symlink target outside the vault.\n")
  61. os.symlink(outside, self.t.workbench / "shortcut.md")
  62. self.stage()
  63. self.assertNotIn("workbench__shortcut.md", self.t.staged)
  65. def test_extra_deny_from_env(self):
  66. env = dict(self.t.env, EXTRA_DENY="research, atlas/glaze-library/")
  67. run(load("gate.py", env))
  68. self.assertFalse([n for n in self.t.staged if "research" in n or "glaze-library" in n])
  70. # ---- staging and change detection -------------------------------------
  71. def test_flattened_names_encode_the_path(self):
  72. self.stage()
  73. self.assertIn("workbench__Efforts__garden-studio__notes__raised-beds.md", self.t.staged)
  74. self.assertIn("identity__Tess Okonjo.md", self.t.staged)
  76. def test_staging_is_wiped_every_run(self):
  77. self.stage()
  78. ghost = self.t.staging_root / "staging" / "ghost-from-yesterday.md"
  79. ghost.write_text("left behind")
  80. self.stage()
  81. self.assertFalse(ghost.exists())
  83. def test_rerun_stages_nothing(self):
  84. _, first = self.stage()
  85. _, second = self.stage()
  86. self.assertGreater(self.counts(first)["staged"], 0)
  87. self.assertEqual(self.counts(second)["staged"], 0)
  88. self.assertEqual(self.t.staged, [])
  90. def test_editing_one_note_stages_exactly_that_note(self):
  91. self.stage()
  92. note = self.t.identity / "Studio Glaze Library.md"
  93. note.write_text(note.read_text() + "- [status] 15 of about 40 glazes documented\n")
  94. _, out = self.stage()
  95. self.assertEqual(self.counts(out)["staged"], 1)
  96. self.assertEqual(self.t.staged, ["identity__Studio Glaze Library.md"])
  98. def test_deleted_note_leaves_the_manifest(self):
  99. self.stage()
  100. (self.t.workbench / "research" / "lake-history.md").unlink()
  101. self.stage()
  102. manifest = json.loads((self.t.staging_root / ".manifest.json").read_text())
  103. self.assertNotIn("workbench/research/lake-history.md", manifest)
  105. def test_manifest_hashes_post_gate_content(self):
  106. gate = self.t.gate()
  107. run(gate)
  108. manifest = json.loads((self.t.staging_root / ".manifest.json").read_text())
  109. me = (self.t.workbench / "me.md").read_text(encoding="utf-8")
  110. self.assertEqual(manifest["workbench/me.md"], gate._hash(gate._gate_content(me)))
  112. # ---- scaffolding ------------------------------------------------------
  113. def test_scaffolding_is_skipped(self):
  114. self.stage()
  115. for skipped in ("workbench__CLAUDE.md", "workbench__HEARTBEAT.md", "workbench__MEMORY.md",
  116. "workbench__00-Index.md", "workbench__2026-09-20.md",
  117. "workbench__kiln-repair-done.md", "workbench___templates__daily.md",
  118. "workbench___scratch__idea.md",
  119. "workbench__Efforts__garden-studio__02-run-logs__2026-09-14.md",
  120. "workbench__Efforts__pottery-class__01-sessions__week-1.md"):
  121. self.assertNotIn(skipped, self.t.staged)
  123. def test_one_explainer_per_vault_in_priority_order(self):
  124. self.stage()
  125. s = self.t.staged
  126. self.assertIn("workbench__README.md", s) # README beats all
  127. self.assertIn("workbench__Efforts__garden-studio__README.md", s)
  128. self.assertNotIn("workbench__Efforts__garden-studio__CLAUDE.md", s)
  129. self.assertIn("workbench__Efforts__bike-restoration__00-Index.md", s) # no README
  130. self.assertNotIn("workbench__Efforts__bike-restoration__CLAUDE.md", s)
  131. self.assertIn("workbench__Efforts__pottery-class__CLAUDE.md", s) # CLAUDE is all it has
  132. self.assertIn("workbench__Atlas__glaze-library__cone-6-celadon.md", s) # no explainer: fine
  134. def test_keep_path_exception(self):
  135. self.stage()
  136. self.assertIn("workbench___meta__decisions.md", self.t.staged)
  137. self.assertIn("workbench___meta__state.md", self.t.staged)
  139. # ---- the section gate --------------------------------------------------
  140. def test_disabled_section_gate_stages_notes_whole(self):
  141. gate = self.t.gate()
  142. text = (self.t.identity / "Rowan.md").read_text()
  143. self.assertEqual(gate._gate_content(text), text)
  145. def test_empty_pattern_lists_match_nothing(self):
  146. gate = self.t.gate()
  147. self.assertIsNone(gate._SENSITIVE_RE.search("anything at all [[Rowan]]"))
  148. self.assertIsNone(gate._EXCLUDED_LINK_RE.search("[[Rowan]] [[Tess Okonjo]]"))
  150. def test_enabled_section_gate_stages_framing_only(self):
  151. gate = self.t.gate()
  152. gate._SENSITIVE_RE = re.compile(r"mentors", re.IGNORECASE)
  153. text = (self.t.identity / "Rowan.md").read_text()
  154. framed = gate._gate_content(text)
  155. self.assertIn("The owner of both vaults", framed)
  156. self.assertNotIn("## Observations", framed)
  157. self.assertNotIn("mentors", framed)
  159. def test_excluded_links_are_redacted_in_framing(self):
  160. gate = self.t.gate()
  161. gate._EXCLUDED_LINK_RE = re.compile(r"\[\[\s*(Tess Okonjo)", re.IGNORECASE)
  162. framed = gate._gate_content("# Note\n\nSee [[Tess Okonjo]] for details.\n\n## Relations\n- x -> y\n")
  163. self.assertIn("[[redacted", framed)
  164. self.assertNotIn("Tess", framed)
  166. def test_full_sample_run_counts(self):
  167. _, out = self.stage()
  168. c = self.counts(out)
  169. self.assertEqual(c["staged"], 21)
  170. self.assertEqual(c["total-staged"], 21)
  173. if __name__ == "__main__":
  174. unittest.main()