cp dines

Author: dines-rl

src/runloop_api_client/sdk/__init__.py

@@ -14,8 +14,8 @@
     SnapshotOps,
     BenchmarkOps,
     BlueprintOps,
-    NetworkPolicyOps,
     GatewayConfigOps,
+    NetworkPolicyOps,
     StorageObjectOps,
 )
 from .agent import Agent
@@ -29,8 +29,8 @@
     AsyncSnapshotOps,
     AsyncBenchmarkOps,
     AsyncBlueprintOps,
-    AsyncNetworkPolicyOps,
     AsyncGatewayConfigOps,
+    AsyncNetworkPolicyOps,
     AsyncStorageObjectOps,
 )
 from .devbox import Devbox, NamedShell
@@ -47,8 +47,8 @@
 from .benchmark_run import BenchmarkRun
 from .async_scenario import AsyncScenario
 from .async_snapshot import AsyncSnapshot
-from .network_policy import NetworkPolicy
 from .gateway_config import GatewayConfig
+from .network_policy import NetworkPolicy
 from .storage_object import StorageObject
 from .async_benchmark import AsyncBenchmark
 from .async_blueprint import AsyncBlueprint
@@ -57,8 +57,8 @@
 from .scenario_builder import ScenarioBuilder
 from .async_scenario_run import AsyncScenarioRun
 from .async_benchmark_run import AsyncBenchmarkRun
-from .async_network_policy import AsyncNetworkPolicy
 from .async_gateway_config import AsyncGatewayConfig
+from .async_network_policy import AsyncNetworkPolicy
 from .async_storage_object import AsyncStorageObject
 from .async_execution_result import AsyncExecutionResult
 from .async_scenario_builder import AsyncScenarioBuilder

src/runloop_api_client/sdk/_types.py

@@ -19,17 +19,17 @@
     BenchmarkUpdateParams,
     BlueprintCreateParams,
     DevboxUploadFileParams,
-    NetworkPolicyListParams,
     GatewayConfigListParams,
+    NetworkPolicyListParams,
     DevboxCreateTunnelParams,
     DevboxDownloadFileParams,
     DevboxEnableTunnelParams,
     DevboxRemoveTunnelParams,
     DevboxSnapshotDiskParams,
-    NetworkPolicyCreateParams,
-    NetworkPolicyUpdateParams,
     GatewayConfigCreateParams,
     GatewayConfigUpdateParams,
+    NetworkPolicyCreateParams,
+    NetworkPolicyUpdateParams,
     DevboxReadFileContentsParams,
     DevboxWriteFileContentsParams,
     BenchmarkRunListScenarioRunsParams,

src/runloop_api_client/sdk/async_.py

@@ -26,10 +26,10 @@
     SDKBenchmarkCreateParams,
     SDKBlueprintCreateParams,
     SDKDiskSnapshotListParams,
-    SDKNetworkPolicyListParams,
     SDKGatewayConfigListParams,
-    SDKNetworkPolicyCreateParams,
+    SDKNetworkPolicyListParams,
     SDKGatewayConfigCreateParams,
+    SDKNetworkPolicyCreateParams,
     SDKDevboxCreateFromImageParams,
 )
 from .._types import Timeout, NotGiven, not_given
@@ -43,8 +43,8 @@
 from .async_benchmark import AsyncBenchmark
 from .async_blueprint import AsyncBlueprint
 from ..lib.context_loader import TarFilter, build_directory_tar
-from .async_network_policy import AsyncNetworkPolicy
 from .async_gateway_config import AsyncGatewayConfig
+from .async_network_policy import AsyncNetworkPolicy
 from .async_storage_object import AsyncStorageObject
 from .async_scenario_builder import AsyncScenarioBuilder
 from ..types.object_create_params import ContentType

src/runloop_api_client/sdk/sync.py

@@ -26,10 +26,10 @@
     SDKBenchmarkCreateParams,
     SDKBlueprintCreateParams,
     SDKDiskSnapshotListParams,
-    SDKNetworkPolicyListParams,
     SDKGatewayConfigListParams,
-    SDKNetworkPolicyCreateParams,
+    SDKNetworkPolicyListParams,
     SDKGatewayConfigCreateParams,
+    SDKNetworkPolicyCreateParams,
     SDKDevboxCreateFromImageParams,
 )
 from .devbox import Devbox
@@ -41,8 +41,8 @@
 from .snapshot import Snapshot
 from .benchmark import Benchmark
 from .blueprint import Blueprint
-from .network_policy import NetworkPolicy
 from .gateway_config import GatewayConfig
+from .network_policy import NetworkPolicy
 from .storage_object import StorageObject
 from .scenario_builder import ScenarioBuilder
 from ..lib.context_loader import TarFilter, build_directory_tar

tests/sdk/conftest.py

@@ -28,6 +28,7 @@
     "benchmark": "bmd_123",
     "benchmark_run": "bmr_123",
     "network_policy": "np_123",
+    "gateway_config": "gwc_123",
 }
 
 # Test URL constants
@@ -179,6 +180,26 @@ class MockNetworkPolicyView:
     egress: MockEgress = field(default_factory=MockEgress)
 
 
+@dataclass
+class MockAuthMechanism:
+    """Mock AuthMechanism for testing."""
+
+    type: str = "bearer"
+    key: str | None = None
+
+
+@dataclass
+class MockGatewayConfigView:
+    """Mock GatewayConfigView for testing."""
+
+    id: str = TEST_IDS["gateway_config"]
+    name: str = "test-gateway-config"
+    endpoint: str = "https://api.example.com"
+    description: str | None = "Test gateway config description"
+    create_time_ms: int = 1234567890000
+    auth_mechanism: MockAuthMechanism = field(default_factory=MockAuthMechanism)
+
+
 class AsyncIterableMock:
     """A simple async iterable mock for testing paginated responses."""
 
@@ -316,6 +337,12 @@ def network_policy_view() -> MockNetworkPolicyView:
     return MockNetworkPolicyView()
 
 
+@pytest.fixture
+def gateway_config_view() -> MockGatewayConfigView:
+    """Create a mock GatewayConfigView."""
+    return MockGatewayConfigView()
+
+
 @pytest.fixture
 def mock_httpx_response() -> Mock:
     """Create a mock httpx.Response."""

tests/sdk/test_async_gateway_config.py

@@ -0,0 +1,89 @@
+"""Comprehensive tests for async GatewayConfig class."""
+
+from __future__ import annotations
+
+from unittest.mock import AsyncMock
+
+import pytest
+
+from tests.sdk.conftest import MockGatewayConfigView
+from runloop_api_client.sdk import AsyncGatewayConfig
+
+
+class TestAsyncGatewayConfig:
+    """Tests for AsyncGatewayConfig class."""
+
+    def test_init(self, mock_async_client: AsyncMock) -> None:
+        """Test AsyncGatewayConfig initialization."""
+        gateway_config = AsyncGatewayConfig(mock_async_client, "gwc_123")
+        assert gateway_config.id == "gwc_123"
+
+    def test_repr(self, mock_async_client: AsyncMock) -> None:
+        """Test AsyncGatewayConfig string representation."""
+        gateway_config = AsyncGatewayConfig(mock_async_client, "gwc_123")
+        assert repr(gateway_config) == "<AsyncGatewayConfig id='gwc_123'>"
+
+    @pytest.mark.asyncio
+    async def test_get_info(self, mock_async_client: AsyncMock, gateway_config_view: MockGatewayConfigView) -> None:
+        """Test get_info method."""
+        mock_async_client.gateway_configs.retrieve = AsyncMock(return_value=gateway_config_view)
+
+        gateway_config = AsyncGatewayConfig(mock_async_client, "gwc_123")
+        result = await gateway_config.get_info(
+            extra_headers={"X-Custom": "value"},
+            extra_query={"param": "value"},
+            extra_body={"key": "value"},
+            timeout=30.0,
+        )
+
+        assert result == gateway_config_view
+        mock_async_client.gateway_configs.retrieve.assert_awaited_once()
+
+    @pytest.mark.asyncio
+    async def test_update(self, mock_async_client: AsyncMock, gateway_config_view: MockGatewayConfigView) -> None:
+        """Test update method."""
+        mock_async_client.gateway_configs.update = AsyncMock(return_value=gateway_config_view)
+
+        gateway_config = AsyncGatewayConfig(mock_async_client, "gwc_123")
+        result = await gateway_config.update(
+            name="updated-gateway",
+            description="Updated description",
+            endpoint="https://api.updated.com",
+            auth_mechanism={"type": "header", "key": "x-api-key"},
+            extra_headers={"X-Custom": "value"},
+            timeout=30.0,
+        )
+
+        assert result == gateway_config_view
+        mock_async_client.gateway_configs.update.assert_awaited_once()
+
+    @pytest.mark.asyncio
+    async def test_update_partial(
+        self, mock_async_client: AsyncMock, gateway_config_view: MockGatewayConfigView
+    ) -> None:
+        """Test update method with partial fields."""
+        mock_async_client.gateway_configs.update = AsyncMock(return_value=gateway_config_view)
+
+        gateway_config = AsyncGatewayConfig(mock_async_client, "gwc_123")
+        result = await gateway_config.update(
+            name="renamed-gateway",
+        )
+
+        assert result == gateway_config_view
+        mock_async_client.gateway_configs.update.assert_awaited_once()
+
+    @pytest.mark.asyncio
+    async def test_delete(self, mock_async_client: AsyncMock, gateway_config_view: MockGatewayConfigView) -> None:
+        """Test delete method."""
+        mock_async_client.gateway_configs.delete = AsyncMock(return_value=gateway_config_view)
+
+        gateway_config = AsyncGatewayConfig(mock_async_client, "gwc_123")
+        result = await gateway_config.delete(
+            extra_headers={"X-Custom": "value"},
+            extra_query={"param": "value"},
+            extra_body={"key": "value"},
+            timeout=30.0,
+        )
+
+        assert result == gateway_config_view
+        mock_async_client.gateway_configs.delete.assert_awaited_once()

tests/sdk/test_async_ops.py

@@ -19,6 +19,7 @@
     MockSnapshotView,
     MockBenchmarkView,
     MockBlueprintView,
+    MockGatewayConfigView,
     MockNetworkPolicyView,
     create_mock_httpx_response,
 )
@@ -38,8 +39,10 @@
     AsyncSnapshotOps,
     AsyncBenchmarkOps,
     AsyncBlueprintOps,
+    AsyncGatewayConfig,
     AsyncNetworkPolicy,
     AsyncStorageObject,
+    AsyncGatewayConfigOps,
     AsyncNetworkPolicyOps,
     AsyncStorageObjectOps,
 )
@@ -1313,6 +1316,81 @@ async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
         mock_async_client.network_policies.list.assert_awaited_once()
 
 
+class TestAsyncGatewayConfigOps:
+    """Tests for AsyncGatewayConfigOps class."""
+
+    @pytest.mark.asyncio
+    async def test_create(self, mock_async_client: AsyncMock, gateway_config_view: MockGatewayConfigView) -> None:
+        """Test create method."""
+        mock_async_client.gateway_configs.create = AsyncMock(return_value=gateway_config_view)
+
+        ops = AsyncGatewayConfigOps(mock_async_client)
+        gateway_config = await ops.create(
+            name="test-gateway-config",
+            endpoint="https://api.example.com",
+            auth_mechanism={"type": "bearer"},
+        )
+
+        assert isinstance(gateway_config, AsyncGatewayConfig)
+        assert gateway_config.id == "gwc_123"
+        mock_async_client.gateway_configs.create.assert_awaited_once()
+
+    def test_from_id(self, mock_async_client: AsyncMock) -> None:
+        """Test from_id method."""
+        ops = AsyncGatewayConfigOps(mock_async_client)
+        gateway_config = ops.from_id("gwc_123")
+
+        assert isinstance(gateway_config, AsyncGatewayConfig)
+        assert gateway_config.id == "gwc_123"
+
+    @pytest.mark.asyncio
+    async def test_list_empty(self, mock_async_client: AsyncMock) -> None:
+        """Test list method with empty results."""
+        page = SimpleNamespace(gateway_configs=[])
+        mock_async_client.gateway_configs.list = AsyncMock(return_value=page)
+
+        ops = AsyncGatewayConfigOps(mock_async_client)
+        gateway_configs = await ops.list(limit=10)
+
+        assert len(gateway_configs) == 0
+        mock_async_client.gateway_configs.list.assert_awaited_once()
+
+    @pytest.mark.asyncio
+    async def test_list_single(self, mock_async_client: AsyncMock, gateway_config_view: MockGatewayConfigView) -> None:
+        """Test list method with single result."""
+        page = SimpleNamespace(gateway_configs=[gateway_config_view])
+        mock_async_client.gateway_configs.list = AsyncMock(return_value=page)
+
+        ops = AsyncGatewayConfigOps(mock_async_client)
+        gateway_configs = await ops.list(
+            limit=10,
+            starting_after="gwc_000",
+        )
+
+        assert len(gateway_configs) == 1
+        assert isinstance(gateway_configs[0], AsyncGatewayConfig)
+        assert gateway_configs[0].id == "gwc_123"
+        mock_async_client.gateway_configs.list.assert_awaited_once()
+
+    @pytest.mark.asyncio
+    async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
+        """Test list method with multiple results."""
+        gateway_config_view1 = MockGatewayConfigView(id="gwc_001", name="gateway-1")
+        gateway_config_view2 = MockGatewayConfigView(id="gwc_002", name="gateway-2")
+        page = SimpleNamespace(gateway_configs=[gateway_config_view1, gateway_config_view2])
+        mock_async_client.gateway_configs.list = AsyncMock(return_value=page)
+
+        ops = AsyncGatewayConfigOps(mock_async_client)
+        gateway_configs = await ops.list(limit=10)
+
+        assert len(gateway_configs) == 2
+        assert isinstance(gateway_configs[0], AsyncGatewayConfig)
+        assert isinstance(gateway_configs[1], AsyncGatewayConfig)
+        assert gateway_configs[0].id == "gwc_001"
+        assert gateway_configs[1].id == "gwc_002"
+        mock_async_client.gateway_configs.list.assert_awaited_once()
+
+
 class TestAsyncRunloopSDK:
     """Tests for AsyncRunloopSDK class."""
 
@@ -1323,6 +1401,7 @@ def test_init(self) -> None:
         assert isinstance(runloop.agent, AsyncAgentOps)
         assert isinstance(runloop.benchmark, AsyncBenchmarkOps)
         assert isinstance(runloop.devbox, AsyncDevboxOps)
+        assert isinstance(runloop.gateway_config, AsyncGatewayConfigOps)
         assert isinstance(runloop.network_policy, AsyncNetworkPolicyOps)
         assert isinstance(runloop.scorer, AsyncScorerOps)
         assert isinstance(runloop.snapshot, AsyncSnapshotOps)