chore: initial commit

Co-authored-by: Lifei Zhou <lifei@squareup.com>
Co-authored-by: Mic Neale <micn@tbd.email>
Co-authored-by: Lily Delalande <ldelalande@squareup.com>
Co-authored-by: Bradley Axen <baxen@squareup.com>
Co-authored-by: Andy Lane <alane@squareup.com>
Co-authored-by: Elena Zherdeva <ezherdeva@squareup.com>
Co-authored-by: Zaki Ali <zaki@squareup.com>
Co-authored-by: Salman Mohammed <smohammed@squareup.com>

tests/test_completer.py

@@ -0,0 +1,50 @@
+from unittest.mock import Mock
+
+import pytest
+from goose.cli.prompt.completer import GoosePromptCompleter
+from goose.command.base import Command
+from prompt_toolkit.completion import Completion
+from prompt_toolkit.document import Document
+
+# Mock Command class
+dummy_command = Mock(spec=Command)
+
+dummy_command.get_completions = Mock(
+    return_value=[
+        Completion(text="completion1"),
+        Completion(text="completion2"),
+    ]
+)
+
+commands_list = {"test_command1": dummy_command, "test_command2": dummy_command}
+
+
+@pytest.fixture
+def completer():
+    return GoosePromptCompleter(commands=commands_list)
+
+
+def test_get_command_completions(completer):
+    document = Document(text="/test_command1:input")
+    completions = list(completer.get_command_completions(document))
+    assert len(completions) == 2
+    assert completions[0].text == "completion1"
+    assert completions[1].text == "completion2"
+
+
+def test_get_command_name_completions(completer):
+    document = Document(text="/test")
+    completions = list(completer.get_command_name_completions(document))
+    print(completions)
+    assert len(completions) == 2
+    assert completions[0].text == "test_command1"
+    assert completions[1].text == "test_command2"
+
+
+def test_get_completions(completer):
+    document = Document(text="/test_command1:input")
+    completions = list(completer.get_completions(document, None))
+    print(completions)
+    assert len(completions) == 2
+    assert completions[0].text == "completion1"
+    assert completions[1].text == "completion2"