add unit tests for ScenarioOps and AsyncScenarioOps

Author: sid-rl

tests/sdk/conftest.py

@@ -273,9 +273,8 @@ def mock_async_stream() -> AsyncMock:
     """
 
     async def async_iter():
-        # Empty async iterator
-        if False:
-            yield
+        return
+        yield  # Make this a generator
 
     stream = AsyncMock()
     stream.__aiter__ = Mock(return_value=async_iter())

tests/sdk/test_async_ops.py

@@ -15,6 +15,7 @@
     MockDevboxView,
     MockObjectView,
     MockScorerView,
+    MockScenarioView,
     MockSnapshotView,
     MockBlueprintView,
     create_mock_httpx_response,
@@ -1067,6 +1068,82 @@ async def test_create_from_object_with_agent_setup(
         )
 
 
+class TestAsyncScenarioOps:
+    """Tests for AsyncScenarioOps class."""
+
+    def test_from_id(self, mock_async_client: AsyncMock) -> None:
+        """Test from_id method."""
+        from runloop_api_client.sdk import AsyncScenario
+        from runloop_api_client.sdk.async_ import AsyncScenarioOps
+
+        ops = AsyncScenarioOps(mock_async_client)
+        scenario = ops.from_id("scn_123")
+
+        assert isinstance(scenario, AsyncScenario)
+        assert scenario.id == "scn_123"
+
+    @pytest.mark.asyncio
+    async def test_list_empty(self, mock_async_client: AsyncMock) -> None:
+        """Test list method with empty results."""
+        from runloop_api_client.sdk.async_ import AsyncScenarioOps
+
+        async def async_iter():
+            return
+            yield  # Make this a generator
+
+        mock_async_client.scenarios.list = AsyncMock(return_value=async_iter())
+
+        ops = AsyncScenarioOps(mock_async_client)
+        scenarios = await ops.list(limit=10)
+
+        assert len(scenarios) == 0
+        mock_async_client.scenarios.list.assert_awaited_once()
+
+    @pytest.mark.asyncio
+    async def test_list_single(self, mock_async_client: AsyncMock, scenario_view: MockScenarioView) -> None:
+        """Test list method with single result."""
+        from runloop_api_client.sdk import AsyncScenario
+        from runloop_api_client.sdk.async_ import AsyncScenarioOps
+
+        async def async_iter():
+            yield scenario_view
+
+        mock_async_client.scenarios.list = AsyncMock(return_value=async_iter())
+
+        ops = AsyncScenarioOps(mock_async_client)
+        scenarios = await ops.list(limit=10)
+
+        assert len(scenarios) == 1
+        assert isinstance(scenarios[0], AsyncScenario)
+        assert scenarios[0].id == "scn_123"
+        mock_async_client.scenarios.list.assert_awaited_once()
+
+    @pytest.mark.asyncio
+    async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
+        """Test list method with multiple results."""
+        from runloop_api_client.sdk import AsyncScenario
+        from runloop_api_client.sdk.async_ import AsyncScenarioOps
+
+        scenario_view1 = MockScenarioView(id="scn_001", name="scenario-1")
+        scenario_view2 = MockScenarioView(id="scn_002", name="scenario-2")
+
+        async def async_iter():
+            yield scenario_view1
+            yield scenario_view2
+
+        mock_async_client.scenarios.list = AsyncMock(return_value=async_iter())
+
+        ops = AsyncScenarioOps(mock_async_client)
+        scenarios = await ops.list(limit=10)
+
+        assert len(scenarios) == 2
+        assert isinstance(scenarios[0], AsyncScenario)
+        assert isinstance(scenarios[1], AsyncScenario)
+        assert scenarios[0].id == "scn_001"
+        assert scenarios[1].id == "scn_002"
+        mock_async_client.scenarios.list.assert_awaited_once()
+
+
 class TestAsyncRunloopSDK:
     """Tests for AsyncRunloopSDK class."""
 

tests/sdk/test_ops.py

@@ -13,6 +13,7 @@
     MockDevboxView,
     MockObjectView,
     MockScorerView,
+    MockScenarioView,
     MockSnapshotView,
     MockBlueprintView,
     create_mock_httpx_response,
@@ -960,6 +961,67 @@ def test_create_from_object_with_agent_setup(self, mock_client: Mock, agent_view
         )
 
 
+class TestScenarioOps:
+    """Tests for ScenarioOps class."""
+
+    def test_from_id(self, mock_client: Mock) -> None:
+        """Test from_id method."""
+        from runloop_api_client.sdk import Scenario
+        from runloop_api_client.sdk.sync import ScenarioOps
+
+        ops = ScenarioOps(mock_client)
+        scenario = ops.from_id("scn_123")
+
+        assert isinstance(scenario, Scenario)
+        assert scenario.id == "scn_123"
+
+    def test_list_empty(self, mock_client: Mock) -> None:
+        """Test list method with empty results."""
+        from runloop_api_client.sdk.sync import ScenarioOps
+
+        mock_client.scenarios.list.return_value = []
+
+        ops = ScenarioOps(mock_client)
+        scenarios = ops.list(limit=10)
+
+        assert len(scenarios) == 0
+        mock_client.scenarios.list.assert_called_once()
+
+    def test_list_single(self, mock_client: Mock, scenario_view: MockScenarioView) -> None:
+        """Test list method with single result."""
+        from runloop_api_client.sdk import Scenario
+        from runloop_api_client.sdk.sync import ScenarioOps
+
+        mock_client.scenarios.list.return_value = [scenario_view]
+
+        ops = ScenarioOps(mock_client)
+        scenarios = ops.list(limit=10)
+
+        assert len(scenarios) == 1
+        assert isinstance(scenarios[0], Scenario)
+        assert scenarios[0].id == "scn_123"
+        mock_client.scenarios.list.assert_called_once()
+
+    def test_list_multiple(self, mock_client: Mock) -> None:
+        """Test list method with multiple results."""
+        from runloop_api_client.sdk import Scenario
+        from runloop_api_client.sdk.sync import ScenarioOps
+
+        scenario_view1 = MockScenarioView(id="scn_001", name="scenario-1")
+        scenario_view2 = MockScenarioView(id="scn_002", name="scenario-2")
+        mock_client.scenarios.list.return_value = [scenario_view1, scenario_view2]
+
+        ops = ScenarioOps(mock_client)
+        scenarios = ops.list(limit=10)
+
+        assert len(scenarios) == 2
+        assert isinstance(scenarios[0], Scenario)
+        assert isinstance(scenarios[1], Scenario)
+        assert scenarios[0].id == "scn_001"
+        assert scenarios[1].id == "scn_002"
+        mock_client.scenarios.list.assert_called_once()
+
+
 class TestRunloopSDK:
     """Tests for RunloopSDK class."""