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>
2024-08-23 16:39:04 -07:00
commit dd126afa6c
68 changed files with 4498 additions and 0 deletions
@@ -0,0 +1,80 @@
+from datetime import datetime
+from time import time
+from unittest.mock import MagicMock, patch
+
+import pytest
+from click.testing import CliRunner
+from exchange import Message
+from goose.cli.main import goose_cli
+
+
+@pytest.fixture
+def mock_print():
+    with patch("goose.cli.main.print") as mock_print:
+        yield mock_print
+
+
+@pytest.fixture
+def mock_session_files_path(tmp_path):
+    with patch("goose.cli.main.SESSIONS_PATH", tmp_path) as session_files_path:
+        yield session_files_path
+
+
+@pytest.fixture
+def mock_session():
+    with patch("goose.cli.main.Session") as mock_session_class:
+        mock_session_instance = MagicMock()
+        mock_session_class.return_value = mock_session_instance
+        yield mock_session_class, mock_session_instance
+
+
+def test_session_resume_command_with_session_name(mock_session):
+    mock_session_class, mock_session_instance = mock_session
+    runner = CliRunner()
+    runner.invoke(goose_cli, ["session", "resume", "session1", "--profile", "default"])
+    mock_session_class.assert_called_once_with(name="session1", profile="default")
+    mock_session_instance.run.assert_called_once()
+
+
+def test_session_resume_command_without_session_name_without_session_files(
+    mock_print, mock_session_files_path, mock_session
+):
+    _, mock_session_instance = mock_session
+    runner = CliRunner()
+    runner.invoke(goose_cli, ["session", "resume"])
+    mock_print.assert_called_with("No sessions found.")
+    mock_session_instance.run.assert_not_called()
+
+
+def test_session_resume_command_without_session_name_use_latest_session(
+    mock_print, mock_session_files_path, mock_session, create_session_file
+):
+    mock_session_class, mock_session_instance = mock_session
+    for index, session_name in enumerate(["first", "second"]):
+        create_session_file([Message.user("Hello1")], mock_session_files_path / f"{session_name}.jsonl", time() + index)
+    runner = CliRunner()
+    runner.invoke(goose_cli, ["session", "resume", "--profile", "default"])
+
+    second_file_path = mock_session_files_path / "second.jsonl"
+    mock_print.assert_called_once_with(f"Resuming most recent session: second from {second_file_path}")
+    mock_session_class.assert_called_once_with(name="second", profile="default")
+    mock_session_instance.run.assert_called_once()
+
+
+def test_session_list_command(mock_print, mock_session_files_path, create_session_file):
+    create_session_file([Message.user("Hello")], mock_session_files_path / "abc.jsonl")
+    runner = CliRunner()
+    runner.invoke(goose_cli, ["session", "list"])
+    file_time = datetime.fromtimestamp(mock_session_files_path.stat().st_mtime).strftime("%Y-%m-%d %H:%M:%S")
+    mock_print.assert_called_with(f"{file_time}    abc")
+
+
+def test_session_clear_command(mock_session_files_path, create_session_file):
+    for index, session_name in enumerate(["first", "second"]):
+        create_session_file([Message.user("Hello1")], mock_session_files_path / f"{session_name}.jsonl", time() + index)
+    runner = CliRunner()
+    runner.invoke(goose_cli, ["session", "clear", "--keep", "1"])
+
+    session_files = list(mock_session_files_path.glob("*.jsonl"))
+    assert len(session_files) == 1
+    assert session_files[0].stem == "second"