Move unit tests to dedicated folder (tests/unit)

Signed-off-by: Teo <[email protected]>
AgentOps-AI · Jan 12, 2025 · 4261f51 · 4261f51
1 parent e2a87e7
commit 4261f51
Show file tree

Hide file tree

Showing 16 changed files with 36 additions and 18 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
diff --git a/tests/unit/__init__.py b/tests/unit/__init__.py
diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py
@@ -0,0 +1,36 @@
+import contextlib
+from typing import Iterator
+
+import pytest
+import requests_mock
+from pytest import Config, Session
+
+import agentops
+from agentops.singleton import clear_singletons
+from tests.fixtures.event import llm_event_spy
+from tests.fixtures.vcr import vcr_config
+
+# Common JWT tokens used across tests
+JWTS = ["some_jwt", "some_jwt2", "some_jwt3"]
+
+
+@pytest.fixture(autouse=True)
+def setup_teardown():
+    """
+    Ensures that all agentops sessions are closed and singletons are cleared in-between tests
+    """
+    clear_singletons()
+    yield
+    agentops.end_all_sessions()  # teardown part
+
+
+@pytest.fixture(scope="session")
+def api_key() -> str:
+    """Standard API key for testing"""
+    return "11111111-1111-4111-8111-111111111111"
+
+
+@pytest.fixture(scope="session")
+def base_url() -> str:
+    """Base API URL"""
+    return "https://api.agentops.ai"
diff --git a/tests/test_agent.py → tests/unit/test_agent.py b/tests/test_agent.py → tests/unit/test_agent.py
diff --git a/tests/test_canary.py → tests/unit/test_canary.py b/tests/test_canary.py → tests/unit/test_canary.py
diff --git a/tests/test_decorators.py → tests/unit/test_decorators.py b/tests/test_decorators.py → tests/unit/test_decorators.py
diff --git a/tests/test_events.py → tests/unit/test_events.py b/tests/test_events.py → tests/unit/test_events.py
diff --git a/tests/test_host_env.py → tests/unit/test_host_env.py b/tests/test_host_env.py → tests/unit/test_host_env.py
diff --git a/tests/test_patcher.py → tests/unit/test_patcher.py b/tests/test_patcher.py → tests/unit/test_patcher.py
diff --git a/tests/test_pre_init.py → tests/unit/test_pre_init.py b/tests/test_pre_init.py → tests/unit/test_pre_init.py
diff --git a/tests/test_record_action.py → tests/unit/test_record_action.py b/tests/test_record_action.py → tests/unit/test_record_action.py
diff --git a/tests/test_record_tool.py → tests/unit/test_record_tool.py b/tests/test_record_tool.py → tests/unit/test_record_tool.py
diff --git a/tests/test_session.py → tests/unit/test_session.py b/tests/test_session.py → tests/unit/test_session.py
diff --git a/tests/test_singleton.py → tests/unit/test_singleton.py b/tests/test_singleton.py → tests/unit/test_singleton.py
diff --git a/tests/test_teardown.py → tests/unit/test_teardown.py b/tests/test_teardown.py → tests/unit/test_teardown.py
diff --git a/tests/test_time_travel.py → tests/unit/test_time_travel.py b/tests/test_time_travel.py → tests/unit/test_time_travel.py