mirror of
https://github.com/aljazceru/goose.git
synced 2025-12-25 18:14: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>
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
from exchange import Message
|
|
from goose.utils.session_file import list_sorted_session_files, read_from_file, session_file_exists, write_to_file
|
|
|
|
|
|
@pytest.fixture
|
|
def file_path(tmp_path):
|
|
return tmp_path / "test_file.jsonl"
|
|
|
|
|
|
def test_read_write_to_file(file_path):
|
|
messages = [
|
|
Message.user("prompt1"),
|
|
Message.user("prompt2"),
|
|
]
|
|
write_to_file(file_path, messages)
|
|
assert file_path.exists()
|
|
|
|
assert read_from_file(file_path) == messages
|
|
|
|
|
|
def test_read_from_file_non_existing_file(tmp_path):
|
|
with pytest.raises(FileNotFoundError):
|
|
read_from_file(tmp_path / "no_existing.json")
|
|
|
|
|
|
def test_read_from_file_non_jsonl_file(file_path):
|
|
file_path.write_text("Hello World")
|
|
with pytest.raises(RuntimeError):
|
|
read_from_file(file_path)
|
|
|
|
|
|
def test_list_sorted_session_files(tmp_path):
|
|
session_files_directory = tmp_path / "session_files_dir"
|
|
session_files_directory.mkdir()
|
|
file_names = ["file1", "file2", "file3"]
|
|
created_session_files = [create_session_file(session_files_directory, file_name) for file_name in file_names]
|
|
|
|
sorted_files = list_sorted_session_files(session_files_directory)
|
|
assert sorted_files == {
|
|
"file3": created_session_files[2],
|
|
"file2": created_session_files[1],
|
|
"file1": created_session_files[0],
|
|
}
|
|
|
|
|
|
def test_list_sorted_session_without_session_files(tmp_path):
|
|
session_files_directory = tmp_path / "session_files_dir"
|
|
|
|
sorted_files = list_sorted_session_files(session_files_directory)
|
|
assert sorted_files == {}
|
|
|
|
|
|
def test_session_file_exists_return_false_when_directory_does_not_exist(tmp_path):
|
|
session_files_directory = tmp_path / "session_files_dir"
|
|
assert not session_file_exists(session_files_directory)
|
|
|
|
|
|
def test_session_file_exists_return_false_when_no_session_file_exists(tmp_path):
|
|
session_files_directory = tmp_path / "session_files_dir"
|
|
session_files_directory.mkdir()
|
|
assert not session_file_exists(session_files_directory)
|
|
|
|
|
|
def test_session_file_exists_return_true_when_session_file_exists(tmp_path):
|
|
session_files_directory = tmp_path / "session_files_dir"
|
|
session_files_directory.mkdir()
|
|
create_session_file(session_files_directory, "session1")
|
|
assert session_file_exists(session_files_directory)
|
|
|
|
|
|
def create_session_file(file_path, file_name) -> Path:
|
|
file = file_path / f"{file_name}.jsonl"
|
|
file.touch()
|
|
return file
|