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,59 @@
+import json
+import os
+from time import time
+from unittest.mock import Mock, patch
+
+import pytest
+from exchange import Exchange
+from goose.profile import Profile
+
+
+@pytest.fixture
+def profile_factory():
+    def _create_profile(attributes={}):
+        profile_attrs = {
+            "provider": "mock_provider",
+            "processor": "mock_processor",
+            "accelerator": "mock_accelerator",
+            "moderator": "mock_moderator",
+            "toolkits": [],
+        }
+        profile_attrs.update(attributes)
+        return Profile(**profile_attrs)
+
+    return _create_profile
+
+
+@pytest.fixture
+def exchange_factory():
+    def _create_exchange(attributes={}):
+        exchange_attrs = {
+            "provider": "mock_provider",
+            "system": "mock_system",
+            "tools": [],
+            "moderator": Mock(),
+            "model": "mock_model",
+        }
+        exchange_attrs.update(attributes)
+        return Exchange(**exchange_attrs)
+
+    return _create_exchange
+
+
+@pytest.fixture
+def mock_sessions_path(tmp_path):
+    with patch("goose.cli.config.SESSIONS_PATH", tmp_path) as mock_path:
+        yield mock_path
+
+
+@pytest.fixture
+def create_session_file():
+    def _create_session_file(messages, session_file_path, mtime=time()):
+        with open(session_file_path, "w") as session_file:
+            for m in messages:
+                json.dump(m.to_dict(), session_file)
+                session_file.write("\n")
+        session_file.close()
+        os.utime(session_file_path, (mtime, mtime))
+
+    return _create_session_file