!squash wip TestServer

Author: tony

src/libtmux/test.py

@@ -9,6 +9,7 @@
 import random
 import time
 import typing as t
+from functools import partial
 
 from .exc import WaitTimeout
 
@@ -349,3 +350,44 @@ def __exit__(
             self._environ[envvar] = value
         for unset in self._unset:
             del self._environ[unset]
+
+
+def _temp_server_finalizer(server: Server) -> None:
+    """Finalizer function for TempTmuxServer that kills the server if it exists."""
+    if server.is_alive():
+        server.kill()
+
+
+def TempTmuxServer(socket_name: str | None = None, **kwargs: t.Any) -> Server:
+    """Create a temporary tmux server that cleans up after itself.
+
+    This is similar to the server pytest fixture, but can be used outside of pytest.
+    The server will be killed when the object is garbage collected.
+
+    Parameters
+    ----------
+    socket_name : str, optional
+        Name for the tmux socket. If not provided, a random one will be generated.
+    **kwargs : dict
+        Additional arguments to pass to Server constructor
+
+    Returns
+    -------
+    Server
+        A Server instance that will clean up after itself
+
+    Examples
+    --------
+    >>> server = TempTmuxServer()
+    >>> server.new_session()
+    Session($... ...)
+    >>> server.is_alive()
+    True
+    >>> del server  # Server is automatically killed
+    """
+    if socket_name is None:
+        socket_name = f"libtmux_test{next(namer)}"
+
+    server = Server(socket_name=socket_name, **kwargs)
+    server.__del__ = partial(_temp_server_finalizer, server)  # type: ignore
+    return server