mirror of
https://github.com/aljazceru/goose.git
synced 2025-12-27 02:54:23 +01:00
dd126afa6c
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>
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
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"
|