tests for checkpointer

Author: nemozak1

src/client/__init__.py

@@ -1,3 +0,0 @@
-from client.agent_client import AgentClient
-
-__all__ = ["AgentClient"]

test/checkpointer/conftest.py

@@ -0,0 +1,12 @@
+"""Checkpoint-specific test configuration that doesn't require service imports."""
+import sys
+from pathlib import Path
+import pytest
+
+# Add project root to path
+sys.path.insert(0, str(Path(__file__).parent.parent.parent))
+
+@pytest.fixture(scope="module")
+def anyio_backend():
+    """Use asyncio only for these tests."""
+    return "asyncio"

test/checkpointer/test_obp_checkpoint_saver.py

@@ -0,0 +1,45 @@
+import pytest
+from unittest.mock import AsyncMock, MagicMock
+from src.checkpointer.obp_checkpoint_saver import OBPCheckpointSaver
+
+@pytest.mark.anyio
+async def test_check_existing_setup_returns_true_when_entities_exist():
+    # Mock the OBPClient
+    mock_client = MagicMock()
+    mock_response = MagicMock()
+    mock_response.json.return_value = {
+        "dynamic_entities": [
+            {"OpeyCheckpoint": {}},
+            {"OpeyCheckpointWrite": {}}
+        ]
+    }
+    mock_client.get = AsyncMock(return_value=mock_response)
+    
+    # Create saver and test
+    saver = OBPCheckpointSaver(client=mock_client)
+    result = await saver._check_existing_setup()
+    
+    assert result is True
+    mock_client.get.assert_called_once_with("/obp/v6.0.0/management/system-dynamic-entities")
+
+@pytest.mark.anyio
+async def test_check_existing_setup_returns_false_when_entities_missing():
+    mock_client = MagicMock()
+    mock_response = MagicMock()
+    mock_response.json.return_value = {"dynamic_entities": []}
+    mock_client.get = AsyncMock(return_value=mock_response)
+    
+    saver = OBPCheckpointSaver(client=mock_client)
+    result = await saver._check_existing_setup()
+    
+    assert result is False
+
+@pytest.mark.anyio
+async def test_check_existing_setup_raises_on_api_error():
+    mock_client = MagicMock()
+    mock_client.get = AsyncMock(side_effect=Exception("API Error"))
+    
+    saver = OBPCheckpointSaver(client=mock_client)
+    
+    with pytest.raises(RuntimeError, match="Error checking existing OBP setup"):
+        await saver._check_existing_setup()

test/conftest.py

@@ -8,9 +8,14 @@
 from unittest.mock import patch, MagicMock
 
 import pytest
-from service.service import app
-from httpx import AsyncClient, ASGITransport
-from asgi_lifespan import LifespanManager
+
+# Conditionally import service to avoid loading issues in unit tests
+try:
+    from service.service import app
+    from httpx import AsyncClient, ASGITransport
+    from asgi_lifespan import LifespanManager
+except Exception:
+    app = None
 
 # @pytest_asyncio.fixture(scope="session", loop_scope="session")
 # def anyio_backend():