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,77 @@
+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