z-bar-qt/cli/tests/test_shell.py

from __future__ import annotations

from subprocess import CompletedProcess
from unittest.mock import call, patch

from typer.testing import CliRunner
from zshell.subcommands.shell import app

runner = CliRunner()


def invoke(*args: str):
	result = runner.invoke(app, args)
	if result.exit_code != 0:
		raise RuntimeError(result.output)
	return result


class TestKill:
	@patch("zshell.subcommands.shell.subprocess.run")
	def test_kill_runs_qs_kill_success(self, mock_run):
		mock_run.return_value = CompletedProcess([], 0, b"", b"Killed abc\n")
		invoke("kill")
		mock_run.assert_called_once_with(
			["qs", "-c", "zshell", "kill"], capture_output=True
		)

	@patch("zshell.subcommands.shell.subprocess.run")
	def test_kill_no_instance_errors(self, mock_run):
		mock_run.return_value = CompletedProcess(
			[], 255, b"", b"No running instances\n"
		)
		result = runner.invoke(app, ["kill"])
		assert result.exit_code != 0
		assert "No running instance to kill" in result.output


class TestStart:
	@patch("zshell.subcommands.shell.subprocess.run")
	def test_start_default_daemon(self, mock_run):
		mock_run.return_value = CompletedProcess(
			[], 0, b"", b"Launching config\n"
		)
		invoke("start")
		mock_run.assert_called_once_with(
			["qs", "-c", "zshell", "-n", "-d"], capture_output=True
		)

	@patch("zshell.subcommands.shell.subprocess.run")
	def test_start_no_daemon(self, mock_run):
		mock_run.return_value = CompletedProcess(
			[], 0, b"", b"Launching config\n"
		)
		invoke("start", "--no-daemon")
		mock_run.assert_called_once_with(
			["qs", "-c", "zshell", "-n"], capture_output=True
		)

	@patch("zshell.subcommands.shell.subprocess.run")
	def test_start_already_running_errors(self, mock_run):
		mock_run.return_value = CompletedProcess(
			[],
			0,
			b"An instance of this configuration is already running.\n",
			b"",
		)
		result = runner.invoke(app, ["start"])
		assert result.exit_code != 0
		assert "already running" in result.output

	@patch("zshell.subcommands.shell.subprocess.run")
	def test_start_other_failure_errors(self, mock_run):
		mock_run.return_value = CompletedProcess([], 1, b"", b"Config error\n")
		result = runner.invoke(app, ["start"])
		assert result.exit_code != 0
		assert "Config error" in result.output


class TestShow:
	@patch("zshell.subcommands.shell.subprocess.run")
	def test_show_runs_ipc_show(self, mock_run):
		mock_run.return_value = CompletedProcess(
			[], 0, b"target visibilities\n", b""
		)
		result = invoke("show")
		assert "target visibilities" in result.output
		mock_run.assert_called_once_with(
			["qs", "-c", "zshell", "ipc", "show"], capture_output=True
		)


class TestLog:
	@patch("zshell.subcommands.shell.subprocess.run")
	def test_log_runs_qs_log(self, mock_run):
		mock_run.return_value = CompletedProcess([], 0, b"log output\n", b"")
		invoke("log")
		mock_run.assert_called_once_with(
			["qs", "-c", "zshell", "log"], capture_output=True
		)


class TestLock:
	@patch("zshell.subcommands.shell.subprocess.run")
	def test_lock_runs_ipc_call_lock(self, mock_run):
		mock_run.return_value = CompletedProcess([], 0, b"", b"")
		invoke("lock")
		mock_run.assert_called_once_with(
			["qs", "-c", "zshell", "ipc", "call", "lock", "lock"],
			capture_output=True,
		)


class TestCall:
	@patch("zshell.subcommands.shell.subprocess.run")
	def test_call_no_args(self, mock_run):
		mock_run.return_value = CompletedProcess([], 0, b"", b"")
		invoke("call", "target", "method")
		mock_run.assert_called_once_with(
			["qs", "-c", "zshell", "ipc", "call", "target", "method"],
			capture_output=True,
		)

	@patch("zshell.subcommands.shell.subprocess.run")
	def test_call_with_args(self, mock_run):
		mock_run.return_value = CompletedProcess([], 0, b"", b"")
		invoke("call", "target", "method", "arg1", "arg2")
		mock_run.assert_called_once_with(
			[
				"qs",
				"-c",
				"zshell",
				"ipc",
				"call",
				"target",
				"method",
				"arg1",
				"arg2",
			],
			capture_output=True,
		)


class TestRestart:
	@patch("zshell.subcommands.shell.start_instance")
	@patch("zshell.subcommands.shell.subprocess.run")
	def test_restart_kills_then_starts(self, mock_run, mock_start):
		mock_run.side_effect = [
			CompletedProcess(
				[], 0, b"", b"Killed abc\n"
			),  # first kill (captured)
			CompletedProcess([], 255, b"", b""),  # poll → no instance
		]
		invoke("restart")
		assert mock_run.call_args_list == [
			call(["qs", "-c", "zshell", "kill"], capture_output=True),
			call(["qs", "-c", "zshell", "kill"], capture_output=True),
		]
		mock_start.assert_called_once_with(no_daemon=False)

	@patch("zshell.subcommands.shell.start_instance")
	@patch("zshell.subcommands.shell.subprocess.run")
	def test_restart_no_daemon(self, mock_run, mock_start):
		mock_run.side_effect = [
			CompletedProcess([], 0, b"", b"Killed abc\n"),
			CompletedProcess([], 255, b"", b""),
		]
		invoke("restart", "--no-daemon")
		assert mock_run.call_args_list == [
			call(["qs", "-c", "zshell", "kill"], capture_output=True),
			call(["qs", "-c", "zshell", "kill"], capture_output=True),
		]
		mock_start.assert_called_once_with(no_daemon=True)