add some tests; not working yet

Author: jrvb-rl

tests/sdk/test_async_agent.py

@@ -0,0 +1,46 @@
+"""Comprehensive tests for async AsyncAgent class."""
+
+from __future__ import annotations
+
+from unittest.mock import AsyncMock
+
+import pytest
+
+from tests.sdk.conftest import MockAgentView
+from runloop_api_client.sdk import AsyncAgent
+
+
+class TestAsyncAgent:
+    """Tests for AsyncAgent class."""
+
+    def test_init(self, mock_async_client: AsyncMock) -> None:
+        """Test AsyncAgent initialization."""
+        agent = AsyncAgent(mock_async_client, "agent_123")
+        assert agent.id == "agent_123"
+
+    def test_repr(self, mock_async_client: AsyncMock) -> None:
+        """Test AsyncAgent string representation."""
+        agent = AsyncAgent(mock_async_client, "agent_123")
+        assert repr(agent) == "<AsyncAgent id='agent_123'>"
+
+    @pytest.mark.asyncio
+    async def test_get_info(self, mock_async_client: AsyncMock, agent_view: MockAgentView) -> None:
+        """Test get_info method."""
+        mock_async_client.agents.retrieve = AsyncMock(return_value=agent_view)
+
+        agent = AsyncAgent(mock_async_client, "agent_123")
+        result = await agent.get_info(
+            extra_headers={"X-Custom": "value"},
+            extra_query={"param": "value"},
+            extra_body={"key": "value"},
+            timeout=30.0,
+        )
+
+        assert result == agent_view
+        mock_async_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_async_clients.py

@@ -15,15 +15,17 @@
     MockObjectView,
     MockSnapshotView,
     MockBlueprintView,
+    MockAgentView,
     create_mock_httpx_response,
 )
-from runloop_api_client.sdk import AsyncDevbox, AsyncSnapshot, AsyncBlueprint, AsyncStorageObject
+from runloop_api_client.sdk import AsyncDevbox, AsyncSnapshot, AsyncBlueprint, AsyncStorageObject, AsyncAgent
 from runloop_api_client.sdk.async_ import (
     AsyncDevboxOps,
     AsyncRunloopSDK,
     AsyncSnapshotOps,
     AsyncBlueprintOps,
     AsyncStorageObjectOps,
+    AsyncAgentOps,
 )
 from runloop_api_client.lib.polling import PollingConfig
 
@@ -515,13 +517,126 @@ async def test_upload_from_dir_with_string_path(
         mock_async_client.objects.complete.assert_awaited_once()
 
 
+class TestAsyncAgentClient:
+    """Tests for AsyncAgentClient class."""
+
+    @pytest.mark.asyncio
+    async def test_create(self, mock_async_client: AsyncMock, agent_view: MockAgentView) -> None:
+        """Test create method."""
+        mock_async_client.agents.create = AsyncMock(return_value=agent_view)
+
+        client = AsyncAgentOps(mock_async_client)
+        agent = await client.create(
+            name="test-agent",
+            metadata={"key": "value"},
+        )
+
+        assert isinstance(agent, AsyncAgent)
+        assert agent.id == "agent_123"
+        mock_async_client.agents.create.assert_called_once()
+
+    def test_from_id(self, mock_async_client: AsyncMock) -> None:
+        """Test from_id method."""
+        client = AsyncAgentOps(mock_async_client)
+        agent = client.from_id("agent_123")
+
+        assert isinstance(agent, AsyncAgent)
+        assert agent.id == "agent_123"
+
+    @pytest.mark.asyncio
+    async def test_list(self, mock_async_client: AsyncMock) -> None:
+        """Test list method."""
+        # Create three agent views with different data
+        agent_view_1 = MockAgentView(
+            id="agent_001",
+            name="first-agent",
+            create_time_ms=1234567890000,
+            is_public=False,
+            source=None,
+        )
+        agent_view_2 = MockAgentView(
+            id="agent_002",
+            name="second-agent",
+            create_time_ms=1234567891000,
+            is_public=True,
+            source={"type": "git", "git": {"repository": "https://github.com/example/repo"}},
+        )
+        agent_view_3 = MockAgentView(
+            id="agent_003",
+            name="third-agent",
+            create_time_ms=1234567892000,
+            is_public=False,
+            source={"type": "npm", "npm": {"package_name": "example-package"}},
+        )
+
+        page = SimpleNamespace(agents=[agent_view_1, agent_view_2, agent_view_3])
+        mock_async_client.agents.list = AsyncMock(return_value=page)
+
+        # Mock retrieve to return the corresponding agent_view when called
+        async def mock_retrieve(agent_id, **kwargs):
+            if agent_id == "agent_001":
+                return agent_view_1
+            elif agent_id == "agent_002":
+                return agent_view_2
+            elif agent_id == "agent_003":
+                return agent_view_3
+            return None
+
+        mock_async_client.agents.retrieve = AsyncMock(side_effect=mock_retrieve)
+
+        client = AsyncAgentOps(mock_async_client)
+        agents = await client.list(
+            limit=10,
+            starting_after="agent_000",
+        )
+
+        # Verify we got three agents
+        assert len(agents) == 3
+        assert all(isinstance(agent, AsyncAgent) for agent in agents)
+
+        # Verify the agent IDs
+        assert agents[0].id == "agent_001"
+        assert agents[1].id == "agent_002"
+        assert agents[2].id == "agent_003"
+
+        # Test that get_info() retrieves the AgentView for the first agent
+        info = await agents[0].get_info()
+        assert info.id == "agent_001"
+        assert info.name == "first-agent"
+        assert info.create_time_ms == 1234567890000
+        assert info.is_public is False
+        assert info.source is None
+
+        # Test that get_info() retrieves the AgentView for the second agent
+        info = await agents[1].get_info()
+        assert info.id == "agent_002"
+        assert info.name == "second-agent"
+        assert info.create_time_ms == 1234567891000
+        assert info.is_public is True
+        assert info.source == {"type": "git", "git": {"repository": "https://github.com/example/repo"}}
+
+        # Test that get_info() retrieves the AgentView for the third agent
+        info = await agents[2].get_info()
+        assert info.id == "agent_003"
+        assert info.name == "third-agent"
+        assert info.create_time_ms == 1234567892000
+        assert info.is_public is False
+        assert info.source == {"type": "npm", "npm": {"package_name": "example-package"}}
+
+        # Verify that agents.retrieve was called three times (once for each get_info)
+        assert mock_async_client.agents.retrieve.call_count == 3
+
+        mock_async_client.agents.list.assert_called_once()
+
+
 class TestAsyncRunloopSDK:
     """Tests for AsyncRunloopSDK class."""
 
     def test_init(self) -> None:
         """Test AsyncRunloopSDK initialization."""
         sdk = AsyncRunloopSDK(bearer_token="test-token")
         assert sdk.api is not None
+        assert isinstance(sdk.agent, AsyncAgentOps)
         assert isinstance(sdk.devbox, AsyncDevboxOps)
         assert isinstance(sdk.snapshot, AsyncSnapshotOps)
         assert isinstance(sdk.blueprint, AsyncBlueprintOps)

tests/smoketests/sdk/README.md

@@ -1,12 +1,12 @@
 # SDK End-to-End Smoke Tests
 
-Comprehensive end-to-end tests for the object-oriented Python SDK (`runloop_api_client.sdk`). These tests run against the real Runloop API to validate critical workflows including devboxes, blueprints, snapshots, and storage objects.
+Comprehensive end-to-end tests for the object-oriented Python SDK (`runloop_api_client.sdk`). These tests run against the real Runloop API to validate critical workflows including agents, devboxes, blueprints, snapshots, and storage objects.
 
 ## Overview
 
 The Python SDK provides both synchronous and asynchronous interfaces:
-- **Synchronous SDK**: `RunloopSDK` with `Devbox`, `Blueprint`, `Snapshot`, `StorageObject`
-- **Asynchronous SDK**: `AsyncRunloopSDK` with `AsyncDevbox`, `AsyncBlueprint`, `AsyncSnapshot`, `AsyncStorageObject`
+- **Synchronous SDK**: `RunloopSDK` with `Agent`, `Devbox`, `Blueprint`, `Snapshot`, `StorageObject`
+- **Asynchronous SDK**: `AsyncRunloopSDK` with `AsyncAgent`, `AsyncDevbox`, `AsyncBlueprint`, `AsyncSnapshot`, `AsyncStorageObject`
 
 These tests ensure both interfaces work correctly in real-world scenarios.
 
@@ -15,6 +15,17 @@ These tests ensure both interfaces work correctly in real-world scenarios.
 ### Infrastructure
 - `conftest.py` - Pytest fixtures for SDK client instances
 
+### Agent Tests
+- `test_agent.py` - Synchronous agent operations
+- `test_async_agent.py` - Asynchronous agent operations
+
+**Test Coverage:**
+- Agent lifecycle (create, get_info)
+- Agent listing and retrieval
+- Agent creation with metadata
+- Agent creation with different source types (npm, git)
+- Public/private agent configuration
+
 ### Devbox Tests
 - `test_devbox.py` - Synchronous devbox operations
 - `test_async_devbox.py` - Asynchronous devbox operations
@@ -74,23 +85,30 @@ export RUNLOOP_API_KEY=your_api_key_here
 
 ### Run All SDK Smoke Tests
 ```bash
-RUN_SMOKETESTS=1 uv run pytest -q -vv -m smoketest tests/smoketests/sdk/
+RUN_SMOKETESTS=1 uv run pytest -q -vv -n 20 -m smoketest tests/smoketests/sdk/
 ```
 
 ### Run Specific Test File
 ```bash
-RUN_SMOKETESTS=1 uv run pytest -q -vv -m smoketest tests/smoketests/sdk/test_devbox.py
+RUN_SMOKETESTS=1 uv run pytest -q -vv -n 20 -m smoketest tests/smoketests/sdk/test_devbox.py
 ```
 
 ### Run Specific Test
 ```bash
-RUN_SMOKETESTS=1 uv run pytest -q -vv -m smoketest -k "test_devbox_lifecycle" tests/smoketests/sdk/
+# match a test from anywhere in the directory
+RUN_SMOKETESTS=1 uv run pytest -q -vv-m smoketest -k "test_devbox_lifecycle" tests/smoketests/sdk/
+
+# specify the exact test
+RUN_SMOKETESTS=1 uv run pytest -q -vv -m smoketest tests/smoketests/sdk/test_devbox.py::TestDevboxLifecycle::test_devbox_create
+
+# match a test from one file
+RUN_SMOKETESTS=1 uv run pytest -q -vv -m smoketest -k "test_devbox_create" tests/smoketests/sdk/test_devbox.py
 ```
 
 ### Run Only Sync or Async Tests
 ```bash
 # Sync tests only (files without 'async' prefix)
-RUN_SMOKETESTS=1 uv run pytest -q -vv -m smoketest tests/smoketests/sdk/test_devbox.py tests/smoketests/sdk/test_blueprint.py tests/smoketests/sdk/test_snapshot.py tests/smoketests/sdk/test_storage_object.py
+RUN_SMOKETESTS=1 uv run pytest -q -vv -m smoketest $(ls tests/smoketests/sdk/*.py | grep -v async)
 
 # Async tests only (files with 'async' prefix)
 RUN_SMOKETESTS=1 uv run pytest -q -vv -m smoketest tests/smoketests/sdk/test_async_*.py