Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/tui_gateway/test_session_git_metadata_generation.py

@@ -0,0 +1,72 @@
+"""Gateway wiring for generation-scoped Git metadata publication."""
+
+from __future__ import annotations
+
+import tui_gateway.server as server
+
+
+class _ImmediateThread:
+    def __init__(self, *, target, **_kwargs):
+        self._target = target
+
+    def start(self):
+        self._target()
+
+
+def test_cwd_claim_precedes_probe_and_generation_reaches_publish(monkeypatch):
+    events = []
+
+    class DB:
+        def update_session_cwd(self, session_id, cwd):
+            events.append(("claim", session_id, cwd))
+            return 17
+
+        def publish_session_git_metadata(
+            self, session_id, cwd, generation, branch, root
+        ):
+            events.append(
+                ("publish", session_id, cwd, generation, branch, root)
+            )
+            return True
+
+    monkeypatch.setattr(server, "_get_db", lambda: DB())
+    monkeypatch.setattr(server.threading, "Thread", _ImmediateThread)
+    monkeypatch.setattr(
+        server,
+        "_git_branch_for_cwd",
+        lambda cwd: events.append(("probe", cwd)) or "feature",
+    )
+    monkeypatch.setattr(server, "_git_common_repo_root_for_cwd", lambda _cwd: "/repo")
+
+    generation = server._persist_session_cwd_and_schedule_git_meta(
+        {"session_key": "session"}, "/repo/worktree"
+    )
+
+    assert generation == 17
+    assert events == [
+        ("claim", "session", "/repo/worktree"),
+        ("probe", "/repo/worktree"),
+        (
+            "publish",
+            "session",
+            "/repo/worktree",
+            17,
+            "feature",
+            "/repo",
+        ),
+    ]
+
+
+def test_missing_db_claim_never_starts_git_probe(monkeypatch):
+    probed = []
+    monkeypatch.setattr(server, "_get_db", lambda: None)
+    monkeypatch.setattr(
+        server, "_git_branch_for_cwd", lambda cwd: probed.append(cwd)
+    )
+
+    generation = server._persist_session_cwd_and_schedule_git_meta(
+        {"session_key": "session"}, "/repo"
+    )
+
+    assert generation is None
+    assert probed == []