test: add unit tests for core, parser and operations

tests/test_cleanup.py

@@ -0,0 +1,91 @@
+from __future__ import annotations
+
+from pathlib import Path
+from unittest.mock import patch
+
+from sshctl.operations.cleanup import _is_socket_in_use, cleanup_control_sockets
+
+
+def test_cleanup_no_control_dir(ssh_dir: Path) -> None:
+    result = cleanup_control_sockets(ssh_dir)
+    assert result == {"total": 0, "removed": 0, "active": 0}
+
+
+def test_cleanup_empty(ssh_dir: Path) -> None:
+    control = ssh_dir / "control.d"
+    control.mkdir()
+    result = cleanup_control_sockets(ssh_dir)
+    assert result == {"total": 0, "removed": 0, "active": 0}
+
+
+def test_cleanup_removes_stale(ssh_dir: Path) -> None:
+    control = ssh_dir / "control.d"
+    control.mkdir()
+    (control / "abc.sock").write_text("")
+    (control / "def.sock").write_text("")
+
+    with patch("sshctl.operations.cleanup._is_socket_in_use", return_value=False):
+        result = cleanup_control_sockets(ssh_dir)
+
+    assert result["total"] == 2
+    assert result["removed"] == 2
+    assert result["active"] == 0
+    assert not (control / "abc.sock").exists()
+    assert not (control / "def.sock").exists()
+
+
+def test_cleanup_dry_run_keeps_sockets(ssh_dir: Path) -> None:
+    control = ssh_dir / "control.d"
+    control.mkdir()
+    sock = control / "abc.sock"
+    sock.write_text("")
+
+    with patch("sshctl.operations.cleanup._is_socket_in_use", return_value=False):
+        result = cleanup_control_sockets(ssh_dir, dry_run=True)
+
+    assert result["total"] == 1
+    assert result["removed"] == 1
+    assert sock.exists()
+
+
+def test_cleanup_active_not_removed(ssh_dir: Path) -> None:
+    control = ssh_dir / "control.d"
+    control.mkdir()
+    (control / "abc.sock").write_text("")
+    (control / "def.sock").write_text("")
+
+    def fake_in_use(sock: Path) -> bool:
+        return sock.name == "abc.sock"
+
+    with patch("sshctl.operations.cleanup._is_socket_in_use", side_effect=fake_in_use):
+        result = cleanup_control_sockets(ssh_dir)
+
+    assert result["total"] == 2
+    assert result["active"] == 1
+    assert result["removed"] == 1
+    assert (control / "abc.sock").exists()
+    assert not (control / "def.sock").exists()
+
+
+def test_is_socket_in_use_with_ssh() -> None:
+    with patch(
+        "sshctl.operations.cleanup.subprocess.run",
+    ) as mock_run:
+        mock_run.return_value.stdout = "  123 ssh  user  cwd  DIR  /tmp/socket"
+        assert _is_socket_in_use(Path("/tmp/x.sock")) is True
+
+
+def test_is_socket_in_use_no_ssh() -> None:
+    with patch(
+        "sshctl.operations.cleanup.subprocess.run",
+    ) as mock_run:
+        mock_run.return_value.stdout = "  123 vim  user  cwd  DIR  /tmp"
+        assert _is_socket_in_use(Path("/tmp/x.sock")) is False
+
+
+def test_is_socket_in_use_error() -> None:
+    with patch(
+        "sshctl.operations.cleanup.subprocess.run",
+        side_effect=FileNotFoundError,
+    ):
+        assert _is_socket_in_use(Path("/tmp/x.sock")) is False