initial simple tests for AgentOps

Author: jrvb-rl

tests/sdk/conftest.py

@@ -20,6 +20,7 @@
     "snapshot": "snap_123",
     "blueprint": "bp_123",
     "object": "obj_123",
+    "agent": "agent_123",
 }
 
 # Test URL constants
@@ -86,6 +87,14 @@ class MockObjectView:
     name: str = "test-object"
 
 
+@dataclass
+class MockAgentView:
+    """Mock AgentView for testing."""
+
+    id: str = "agent_123"
+    name: str = "test-agent"
+
+
 def create_mock_httpx_client(methods: dict[str, Any] | None = None) -> AsyncMock:
     """
     Create a mock httpx.AsyncClient with proper context manager setup.
@@ -170,6 +179,12 @@ def object_view() -> MockObjectView:
     return MockObjectView()
 
 
+@pytest.fixture
+def agent_view() -> MockAgentView:
+    """Create a mock AgentView."""
+    return MockAgentView()
+
+
 @pytest.fixture
 def mock_httpx_response() -> Mock:
     """Create a mock httpx.Response."""

tests/sdk/test_agent.py

@@ -0,0 +1,43 @@
+"""Comprehensive tests for sync Agent class."""
+
+from __future__ import annotations
+
+from unittest.mock import Mock
+
+from tests.sdk.conftest import MockAgentView
+from runloop_api_client.sdk import Agent
+
+
+class TestAgent:
+    """Tests for Agent class."""
+
+    def test_init(self, mock_client: Mock) -> None:
+        """Test Agent initialization."""
+        agent = Agent(mock_client, "agent_123")
+        assert agent.id == "agent_123"
+
+    def test_repr(self, mock_client: Mock) -> None:
+        """Test Agent string representation."""
+        agent = Agent(mock_client, "agent_123")
+        assert repr(agent) == "<Agent id='agent_123'>"
+
+    def test_get_info(self, mock_client: Mock, agent_view: MockAgentView) -> None:
+        """Test get_info method."""
+        mock_client.agents.retrieve.return_value = agent_view
+
+        agent = Agent(mock_client, "agent_123")
+        result = agent.get_info(
+            extra_headers={"X-Custom": "value"},
+            extra_query={"param": "value"},
+            extra_body={"key": "value"},
+            timeout=30.0,
+        )
+
+        assert result == agent_view
+        mock_client.agents.retrieve.assert_called_once_with(
+            "agent_123",
+            extra_headers={"X-Custom": "value"},
+            extra_query={"param": "value"},
+            extra_body={"key": "value"},
+            timeout=30.0,
+        )

tests/sdk/test_clients.py

@@ -13,15 +13,17 @@
     MockObjectView,
     MockSnapshotView,
     MockBlueprintView,
+    MockAgentView,
     create_mock_httpx_response,
 )
-from runloop_api_client.sdk import Devbox, Snapshot, Blueprint, StorageObject
+from runloop_api_client.sdk import Devbox, Snapshot, Blueprint, StorageObject, Agent
 from runloop_api_client.sdk.sync import (
     DevboxOps,
     RunloopSDK,
     SnapshotOps,
     BlueprintOps,
     StorageObjectOps,
+    AgentOps,
 )
 from runloop_api_client.lib.polling import PollingConfig
 
@@ -472,13 +474,56 @@ def test_upload_from_dir_with_string_path(
         mock_client.objects.complete.assert_called_once()
 
 
+class TestAgentClient:
+    """Tests for AgentClient class."""
+
+    def test_create(self, mock_client: Mock, agent_view: MockAgentView) -> None:
+        """Test create method."""
+        mock_client.agents.create.return_value = agent_view
+
+        client = AgentOps(mock_client)
+        agent = client.create(
+            name="test-agent",
+            metadata={"key": "value"},
+        )
+
+        assert isinstance(agent, Agent)
+        assert agent.id == "agent_123"
+        mock_client.agents.create.assert_called_once()
+
+    def test_from_id(self, mock_client: Mock) -> None:
+        """Test from_id method."""
+        client = AgentOps(mock_client)
+        agent = client.from_id("agent_123")
+
+        assert isinstance(agent, Agent)
+        assert agent.id == "agent_123"
+
+    def test_list(self, mock_client: Mock, agent_view: MockAgentView) -> None:
+        """Test list method."""
+        page = SimpleNamespace(agents=[agent_view])
+        mock_client.agents.list.return_value = page
+
+        client = AgentOps(mock_client)
+        agents = client.list(
+            limit=10,
+            starting_after="agent_000",
+        )
+
+        assert len(agents) == 1
+        assert isinstance(agents[0], Agent)
+        assert agents[0].id == "agent_123"
+        mock_client.agents.list.assert_called_once()
+
+
 class TestRunloopSDK:
     """Tests for RunloopSDK class."""
 
     def test_init(self) -> None:
         """Test RunloopSDK initialization."""
         sdk = RunloopSDK(bearer_token="test-token")
         assert sdk.api is not None
+        assert isinstance(sdk.agent, AgentOps)
         assert isinstance(sdk.devbox, DevboxOps)
         assert isinstance(sdk.snapshot, SnapshotOps)
         assert isinstance(sdk.blueprint, BlueprintOps)