1 files changed, 87 insertions, 0 deletions
diff --git a/tests/pane_smoke.py b/tests/pane_smoke.py
new file mode 100644
index 0000000..bd7def6
--- /dev/null
+++ b/tests/pane_smoke.py
@@ -0,0 +1,87 @@
+#!/usr/bin/env python3
+"""Smoke test for idatui.pane's supervisor auto-start machinery (no IDA needed).
+
+    python3 tests/pane_smoke.py
+
+Uses IDATUI_SERVER_CMD to launch a dummy port-binder in a tmux pane instead of
+the real ./spawn.sh, so we can exercise _ensure_server end-to-end (start, detect,
+idempotency, remote guard) without a real ida-pro-mcp server. Must run in tmux.
+"""
+import os
+import socket
+import subprocess
+import sys
+import tempfile
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from idatui import pane  # noqa: E402
+
+PASS = FAIL = 0
+
+
+def check(name, cond, detail=""):
+    global PASS, FAIL
+    if cond:
+        PASS += 1
+        print(f"  ok   {name}")
+    else:
+        FAIL += 1
+        print(f"  FAIL {name}   {detail}")
+
+
+def _free_port() -> int:
+    s = socket.socket()
+    s.bind(("127.0.0.1", 0))
+    p = s.getsockname()[1]
+    s.close()
+    return p
+
+
+def main() -> int:
+    if not os.environ.get("TMUX"):
+        print("error: must run inside tmux", file=sys.stderr)
+        return 2
+
+    # a dummy "supervisor": bind the port and idle, so _server_up sees it.
+    fake = tempfile.NamedTemporaryFile("w", suffix=".py", delete=False)
+    fake.write(
+        "import socket,sys,time\n"
+        "s=socket.socket(); s.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)\n"
+        "s.bind(('127.0.0.1',int(sys.argv[1]))); s.listen(); time.sleep(300)\n")
+    fake.close()
+    port = _free_port()
+    os.environ["IDATUI_SERVER_CMD"] = f"{sys.executable} {fake.name} {port}"
+
+    server_pane = None
+    try:
+        check("port starts down", not pane._server_up("127.0.0.1", port))
+
+        srv = pane._ensure_server("127.0.0.1", port, timeout=15.0)
+        server_pane = srv.get("server_pane")
+        check("ensure_server starts the supervisor and it comes up",
+              srv.get("server_started") and srv.get("server_up") and server_pane,
+              str(srv))
+        check("port is now up", pane._server_up("127.0.0.1", port))
+
+        srv2 = pane._ensure_server("127.0.0.1", port, timeout=5.0)
+        check("ensure_server is idempotent when already up",
+              srv2.get("server_started") is False and srv2.get("server_up") is True,
+              str(srv2))
+
+        srv3 = pane._ensure_server("10.255.255.1", 9, timeout=2.0)
+        check("remote+down server is not auto-started",
+              srv3.get("server_started") is False and "local" in (srv3.get("error") or ""),
+              str(srv3))
+    finally:
+        if server_pane:
+            subprocess.run(["tmux", "kill-pane", "-t", server_pane],
+                           capture_output=True)
+        os.unlink(fake.name)
+        os.environ.pop("IDATUI_SERVER_CMD", None)
+
+    print(f"\n{PASS} passed, {FAIL} failed")
+    return 1 if FAIL else 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())