feat: add shell-completions subcommand (#76)

tests/test_autocomplete.py

@@ -0,0 +1,34 @@
+import sys
+import unittest.mock as mock
+
+from goose.utils.autocomplete import SUPPORTED_SHELLS, is_autocomplete_installed, setup_autocomplete
+
+
+def test_supported_shells():
+    assert SUPPORTED_SHELLS == ["bash", "zsh", "fish"]
+
+
+def test_install_autocomplete(tmp_path):
+    file = tmp_path / "test_bash_autocomplete"
+    assert is_autocomplete_installed(file) is False
+
+    file.write_text("_GOOSE_COMPLETE")
+    assert is_autocomplete_installed(file) is True
+
+
+@mock.patch("sys.stdout")
+def test_setup_bash(mocker: mock.MagicMock):
+    setup_autocomplete("bash", install=False)
+    sys.stdout.write.assert_called_with('eval "$(_GOOSE_COMPLETE=bash_source goose)"\n')
+
+
+@mock.patch("sys.stdout")
+def test_setup_zsh(mocker: mock.MagicMock):
+    setup_autocomplete("zsh", install=False)
+    sys.stdout.write.assert_called_with('eval "$(_GOOSE_COMPLETE=zsh_source goose)"\n')
+
+
+@mock.patch("sys.stdout")
+def test_setup_fish(mocker: mock.MagicMock):
+    setup_autocomplete("fish", install=False)
+    sys.stdout.write.assert_called_with("_GOOSE_COMPLETE=fish_source goose | source\n")

tests/test_cli_main.py

@@ -0,0 +1,21 @@
+from click.testing import CliRunner
+from goose.cli.main import get_current_shell, shell_completions
+
+
+def test_get_current_shell(mocker):
+    mocker.patch("os.getenv", return_value="/bin/bash")
+    assert get_current_shell() == "bash"
+
+
+def test_shell_completions_install_invalid_combination():
+    runner = CliRunner()
+    result = runner.invoke(shell_completions, ["--install", "--generate", "bash"])
+    assert result.exit_code != 0
+    assert "Only one of --install or --generate can be specified" in result.output
+
+
+def test_shell_completions_install_no_option():
+    runner = CliRunner()
+    result = runner.invoke(shell_completions, ["bash"])
+    assert result.exit_code != 0
+    assert "One of --install or --generate must be specified" in result.output