✅ Gremlins MVP: unit and CLI tests with a fake claude

2026-10-16 23:33:02 +02:00
parent fd75777628
commit d662790f19
6 changed files with 493 additions and 0 deletions
@@ -0,0 +1,57 @@
+import os
+import signal
+import sys
+import tempfile
+import threading
+import time
+import unittest
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+
+from grm import call  # noqa: E402
+
+CHILD = ("import json, sys, time\n"
+         "p = sys.stdin.buffer.read().decode('utf-8')\n"
+         "print(json.dumps({'type': 'system', 'subtype': 'init', 'session_id': p}), flush=True)\n"
+         "if p == 'sleep': time.sleep(30)\n"
+         "print(json.dumps({'type': 'result', 'session_id': p}), flush=True)\n")
+
+
+class RunClaudeTest(unittest.TestCase):
+    def _run(self, d, prompt):
+        pids = []
+        raw = Path(d) / "j.jsonl"
+        out = call.run_claude([sys.executable, "-c", CHILD], prompt, dict(os.environ), d,
+                              raw, pids.append)
+        return out, raw, pids
+
+    def test_tee_and_parse(self):
+        """B1: prompt goes in on stdin; the stream is parsed and teed to the raw file."""
+        with tempfile.TemporaryDirectory() as d:
+            (ss, code, stopped), raw, pids = self._run(d, "héllo")
+            self.assertEqual((code, stopped, len(pids)), (0, False, 1))
+            self.assertEqual(ss.session_id, "héllo")
+            self.assertIsNotNone(ss.result)
+            self.assertEqual(len(raw.read_bytes().splitlines()), 2)
+
+    def test_sigterm_stops_child(self):
+        """B4: SIGTERM stops the child; stopped is set and the partial raw file is kept."""
+        with tempfile.TemporaryDirectory() as d:
+            timer = threading.Timer(1.5, signal.raise_signal, (signal.SIGTERM,))
+            t0 = time.time()
+            timer.start()
+            try:
+                (ss, code, stopped), raw, pids = self._run(d, "sleep")
+            finally:
+                timer.cancel()
+            self.assertTrue(stopped)
+            self.assertIsNotNone(code)        # the child has exited
+            self.assertLess(time.time() - t0, 20)
+            self.assertIsNone(ss.result)
+            self.assertEqual(ss.session_id, "sleep")
+            self.assertIn(b'"init"', raw.read_bytes())
+
+
+if __name__ == "__main__":
+    unittest.main()