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

tests/test_pty_keepalive_ws.py

@@ -0,0 +1,122 @@
+import json
+
+import pytest
+
+from hermes_cli import web_server
+
+
+class FakeBridge:
+    def __init__(self):
+        self.alive = True
+        self.written = bytearray()
+
+    def read(self, timeout):
+        return b""        # idle forever
+
+    def write(self, data):
+        self.written.extend(data)
+
+    def resize(self, cols, rows):
+        pass
+
+    def close(self):
+        self.alive = False
+
+
+@pytest.fixture
+def pty_keepalive_harness(monkeypatch):
+    class Spawned(list):
+        pass
+
+    spawned = Spawned()
+    spawned.bridges = []
+
+    def fake_spawn(argv, cwd=None, env=None):
+        b = FakeBridge()
+        spawned.append(argv)
+        spawned.bridges.append(b)
+        return b
+
+    monkeypatch.setattr(web_server.PtyBridge, "spawn", staticmethod(fake_spawn))
+    monkeypatch.setattr(web_server, "_ws_auth_reason", lambda ws: (None, "test"))
+    monkeypatch.setattr(web_server, "_ws_host_origin_reason", lambda ws: None)
+    monkeypatch.setattr(web_server, "_ws_client_reason", lambda ws: None)
+
+    async def fake_argv(**kw):
+        resume = "child" if kw.get("resume") == "parent" else kw.get("resume")
+        env = {"HERMES_TUI_RESUME": resume} if resume else {}
+        return (["x", resume or "fresh"], "/tmp", env)
+
+    monkeypatch.setattr(web_server, "_resolve_chat_argv_async", fake_argv)
+
+    try:
+        yield spawned
+    finally:
+        web_server.PTY_REGISTRY._sessions.clear()
+
+
+@pytest.mark.asyncio
+async def test_attach_token_reuses_same_session(pty_keepalive_harness):
+    """Two connects with the same ?attach= token hit one spawned bridge."""
+    from starlette.testclient import TestClient
+
+    client = TestClient(web_server.app)
+    with client.websocket_connect("/api/pty?attach=TOK1") as ws1:
+        ws1.send_bytes(b"hi")
+    with client.websocket_connect("/api/pty?attach=TOK1") as ws2:
+        ws2.send_bytes(b"again")
+    assert len(pty_keepalive_harness) == 1                # reattached, did not respawn
+    assert bytes(pty_keepalive_harness.bridges[0].written) == b"hi\x0cagain"
+
+
+@pytest.mark.asyncio
+async def test_attach_token_reuses_same_resume(pty_keepalive_harness):
+    from starlette.testclient import TestClient
+
+    client = TestClient(web_server.app)
+    with client.websocket_connect("/api/pty?attach=TOK1&resume=same") as ws1:
+        ws1.send_bytes(b"hi")
+    with client.websocket_connect("/api/pty?attach=TOK1&resume=same") as ws2:
+        ws2.send_bytes(b"again")
+    assert pty_keepalive_harness == [["x", "same"]]
+
+
+
+
+@pytest.mark.asyncio
+async def test_attach_token_reuses_canonical_resume(pty_keepalive_harness):
+    from starlette.testclient import TestClient
+
+    client = TestClient(web_server.app)
+    with client.websocket_connect("/api/pty?attach=TOK1&resume=parent") as ws1:
+        ws1.send_bytes(b"hi")
+    with client.websocket_connect("/api/pty?attach=TOK1&resume=child") as ws2:
+        ws2.send_bytes(b"again")
+    assert pty_keepalive_harness == [["x", "child"]]
+
+
+
+
+@pytest.mark.asyncio
+async def test_attach_token_reuses_default_chat_after_active_session_fallback(
+    pty_keepalive_harness, tmp_path, monkeypatch
+):
+    from starlette.testclient import TestClient
+
+    active_session_file = tmp_path / "active-session.json"
+    monkeypatch.setattr(
+        web_server,
+        "_active_session_file_for_channel",
+        lambda app, channel: active_session_file,
+    )
+
+    client = TestClient(web_server.app)
+    with client.websocket_connect("/api/pty?attach=TOK1&channel=CHAT") as ws1:
+        ws1.send_bytes(b"hi")
+
+    active_session_file.write_text(json.dumps({"session_id": "existing"}))
+
+    with client.websocket_connect("/api/pty?attach=TOK1&channel=CHAT") as ws2:
+        ws2.send_bytes(b"again")
+
+    assert pty_keepalive_harness == [["x", "fresh"]]