mirror of
https://github.com/block/goose.git
synced 2026-07-17 12:56:20 +02:00
Luke Alvoeiro
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>
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from goose.cli.prompt.goose_prompt_session import GoosePromptSession
|
|
from goose.cli.prompt.user_input import PromptAction, UserInput
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_prompt_session():
|
|
with patch("prompt_toolkit.PromptSession") as mock_prompt_session:
|
|
yield mock_prompt_session
|
|
|
|
|
|
def test_get_save_session_name(mock_prompt_session):
|
|
mock_prompt_session.prompt.return_value = "my_session"
|
|
goose_prompt_session = GoosePromptSession(mock_prompt_session)
|
|
|
|
assert goose_prompt_session.get_save_session_name() == "my_session"
|
|
|
|
|
|
def test_get_user_input_to_continue(mock_prompt_session):
|
|
mock_prompt_session.prompt.return_value = "input_value"
|
|
goose_prompt_session = GoosePromptSession(mock_prompt_session)
|
|
|
|
user_input = goose_prompt_session.get_user_input()
|
|
|
|
assert user_input == UserInput(PromptAction.CONTINUE, "input_value")
|
|
|
|
|
|
@pytest.mark.parametrize("exit_input", ["exit", ":q"])
|
|
def test_get_user_input_to_exit(exit_input, mock_prompt_session):
|
|
mock_prompt_session.prompt.return_value = exit_input
|
|
goose_prompt_session = GoosePromptSession(mock_prompt_session)
|
|
|
|
user_input = goose_prompt_session.get_user_input()
|
|
|
|
assert user_input == UserInput(PromptAction.EXIT)
|
|
|
|
|
|
@pytest.mark.parametrize("error", [EOFError, KeyboardInterrupt])
|
|
def test_get_user_input_to_exit_when_error_occurs(error, mock_prompt_session):
|
|
mock_prompt_session.prompt.side_effect = error
|
|
goose_prompt_session = GoosePromptSession(mock_prompt_session)
|
|
|
|
user_input = goose_prompt_session.get_user_input()
|
|
|
|
assert user_input == UserInput(PromptAction.EXIT)
|