Add smoketests

Author: alb-rl

src/runloop_api_client/sdk/_types.py

@@ -22,6 +22,7 @@
     NetworkPolicyListParams,
     DevboxCreateTunnelParams,
     DevboxDownloadFileParams,
+    DevboxEnableTunnelParams,
     DevboxRemoveTunnelParams,
     DevboxSnapshotDiskParams,
     NetworkPolicyCreateParams,
@@ -121,6 +122,10 @@ class SDKDevboxCreateTunnelParams(DevboxCreateTunnelParams, LongRequestOptions):
     pass
 
 
+class SDKDevboxEnableTunnelParams(DevboxEnableTunnelParams, LongRequestOptions):
+    pass
+
+
 class SDKDevboxRemoveTunnelParams(DevboxRemoveTunnelParams, LongRequestOptions):
     pass
 

src/runloop_api_client/sdk/async_devbox.py

@@ -10,6 +10,7 @@
 
 from ..types import (
     DevboxView,
+    TunnelView,
     DevboxTunnelView,
     DevboxExecutionDetailView,
     DevboxCreateSSHKeyResponse,
@@ -24,6 +25,7 @@
     SDKDevboxUploadFileParams,
     SDKDevboxCreateTunnelParams,
     SDKDevboxDownloadFileParams,
+    SDKDevboxEnableTunnelParams,
     SDKDevboxExecuteAsyncParams,
     SDKDevboxRemoveTunnelParams,
     SDKDevboxSnapshotDiskParams,
@@ -731,7 +733,9 @@ async def create_tunnel(
         self,
         **params: Unpack[SDKDevboxCreateTunnelParams],
     ) -> DevboxTunnelView:
-        """Create a network tunnel to expose a devbox port publicly.
+        """[Deprecated] Create a legacy tunnel to expose a devbox port publicly.
+
+        Use :meth:`enable_tunnel` instead for the V2 tunnel API.
 
         :param params: See :typeddict:`~runloop_api_client.sdk._types.SDKDevboxCreateTunnelParams` for available parameters
         :return: Details about the public endpoint
@@ -748,6 +752,30 @@ async def create_tunnel(
                 **params,
             )
 
+    async def enable_tunnel(
+        self,
+        **params: Unpack[SDKDevboxEnableTunnelParams],
+    ) -> TunnelView:
+        """Enable a V2 tunnel for secure HTTP access to the devbox.
+
+        V2 tunnels provide encrypted URL-based access without exposing internal IDs.
+        Each devbox can have one tunnel. The tunnel URL format is:
+        ``https://{port}-{tunnel_key}.tunnel.runloop.ai``
+
+        :param params: See :typeddict:`~runloop_api_client.sdk._types.SDKDevboxEnableTunnelParams` for available parameters
+        :return: Tunnel details including the tunnel key for constructing URLs
+        :rtype: TunnelView
+
+        Example:
+            >>> tunnel = await devbox.net.enable_tunnel(auth_mode="open")
+            >>> print(f"Tunnel key: {tunnel.tunnel_key}")
+            >>> # Access via: https://8080-{tunnel.tunnel_key}.tunnel.runloop.ai
+        """
+        return await self._devbox._client.devboxes.enable_tunnel(
+            self._devbox.id,
+            **params,
+        )
+
     async def remove_tunnel(
         self,
         **params: Unpack[SDKDevboxRemoveTunnelParams],

src/runloop_api_client/sdk/devbox.py

@@ -10,6 +10,7 @@
 
 from ..types import (
     DevboxView,
+    TunnelView,
     DevboxTunnelView,
     DevboxExecutionDetailView,
     DevboxCreateSSHKeyResponse,
@@ -25,6 +26,7 @@
     SDKDevboxUploadFileParams,
     SDKDevboxCreateTunnelParams,
     SDKDevboxDownloadFileParams,
+    SDKDevboxEnableTunnelParams,
     SDKDevboxExecuteAsyncParams,
     SDKDevboxRemoveTunnelParams,
     SDKDevboxSnapshotDiskParams,
@@ -734,7 +736,9 @@ def create_tunnel(
         self,
         **params: Unpack[SDKDevboxCreateTunnelParams],
     ) -> DevboxTunnelView:
-        """Create a network tunnel to expose a devbox port publicly.
+        """[Deprecated] Create a legacy tunnel to expose a devbox port publicly.
+
+        Use :meth:`enable_tunnel` instead for the V2 tunnel API.
 
         :param params: See :typeddict:`~runloop_api_client.sdk._types.SDKDevboxCreateTunnelParams` for available parameters
         :return: Details about the public endpoint
@@ -751,6 +755,30 @@ def create_tunnel(
                 **params,
             )
 
+    def enable_tunnel(
+        self,
+        **params: Unpack[SDKDevboxEnableTunnelParams],
+    ) -> TunnelView:
+        """Enable a V2 tunnel for secure HTTP access to the devbox.
+
+        V2 tunnels provide encrypted URL-based access without exposing internal IDs.
+        Each devbox can have one tunnel. The tunnel URL format is:
+        ``https://{port}-{tunnel_key}.tunnel.runloop.ai``
+
+        :param params: See :typeddict:`~runloop_api_client.sdk._types.SDKDevboxEnableTunnelParams` for available parameters
+        :return: Tunnel details including the tunnel key for constructing URLs
+        :rtype: :class:`~runloop_api_client.types.tunnel_view.TunnelView`
+
+        Example:
+            >>> tunnel = devbox.net.enable_tunnel(auth_mode="open")
+            >>> print(f"Tunnel key: {tunnel.tunnel_key}")
+            >>> # Access via: https://8080-{tunnel.tunnel_key}.tunnel.runloop.ai
+        """
+        return self._devbox._client.devboxes.enable_tunnel(
+            self._devbox.id,
+            **params,
+        )
+
     def remove_tunnel(
         self,
         **params: Unpack[SDKDevboxRemoveTunnelParams],

tests/smoketests/sdk/test_async_devbox.py

@@ -458,6 +458,53 @@ async def test_create_and_remove_tunnel(self, async_sdk_client: AsyncRunloopSDK)
         finally:
             await devbox.shutdown()
 
+    @pytest.mark.timeout(TWO_MINUTE_TIMEOUT)
+    async def test_create_with_tunnel_param(self, async_sdk_client: AsyncRunloopSDK) -> None:
+        """Test creating a devbox with tunnel configuration in create params."""
+        devbox = await async_sdk_client.devbox.create(
+            name=unique_name("sdk-async-devbox-tunnel-param"),
+            launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
+            tunnel={"auth_mode": "open"},
+        )
+
+        try:
+            # Verify devbox is running
+            info = await devbox.get_info()
+            assert info.status == "running"
+
+            # Verify tunnel was created at launch
+            assert info.tunnel is not None
+            assert info.tunnel.tunnel_key is not None
+            assert info.tunnel.auth_mode is not None
+        finally:
+            await devbox.shutdown()
+
+    @pytest.mark.timeout(TWO_MINUTE_TIMEOUT)
+    async def test_enable_tunnel(self, async_sdk_client: AsyncRunloopSDK) -> None:
+        """Test enabling a V2 tunnel on an existing devbox."""
+        devbox = await async_sdk_client.devbox.create(
+            name=unique_name("sdk-async-devbox-enable-tunnel"),
+            launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
+        )
+
+        try:
+            # Verify no tunnel exists initially
+            info = await devbox.get_info()
+            assert info.tunnel is None
+
+            # Enable tunnel using the V2 API via SDK
+            tunnel = await devbox.net.enable_tunnel(auth_mode="open")
+            assert tunnel is not None
+            assert tunnel.tunnel_key is not None
+            assert tunnel.auth_mode is not None
+
+            # Verify tunnel is now present in devbox info
+            info = await devbox.get_info()
+            assert info.tunnel is not None
+            assert info.tunnel.tunnel_key == tunnel.tunnel_key
+        finally:
+            await devbox.shutdown()
+
 
 class TestAsyncDevboxCreationMethods:
     """Test various async devbox creation methods."""

tests/smoketests/sdk/test_devbox.py

@@ -455,6 +455,51 @@ def test_create_and_remove_tunnel(self, sdk_client: RunloopSDK) -> None:
         finally:
             devbox.shutdown()
 
+    @pytest.mark.timeout(TWO_MINUTE_TIMEOUT)
+    def test_create_with_tunnel_param(self, sdk_client: RunloopSDK) -> None:
+        """Test creating a devbox with tunnel configuration in create params."""
+        devbox = sdk_client.devbox.create(
+            name=unique_name("sdk-devbox-tunnel-param"),
+            launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
+            tunnel={"auth_mode": "open"},
+        )
+
+        try:
+            # Verify devbox is running
+            info = devbox.get_info()
+            assert info.status == "running"
+
+            # Verify tunnel was created at launch
+            assert info.tunnel is not None
+            assert info.tunnel.tunnel_key is not None
+            assert info.tunnel.auth_mode is not None
+        finally:
+            devbox.shutdown()
+
+    @pytest.mark.timeout(TWO_MINUTE_TIMEOUT)
+    def test_enable_tunnel(self, sdk_client: RunloopSDK) -> None:
+        """Test enabling a V2 tunnel on an existing devbox."""
+        devbox = sdk_client.devbox.create(
+            name=unique_name("sdk-devbox-enable-tunnel"),
+            launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
+        )
+
+        try:
+            info = devbox.get_info()
+            assert info.tunnel is None
+
+            # Enable tunnel using the V2 API via SDK
+            tunnel = devbox.net.enable_tunnel(auth_mode="open")
+            assert tunnel is not None
+            assert tunnel.tunnel_key is not None
+            assert tunnel.auth_mode is not None
+
+            info = devbox.get_info()
+            assert info.tunnel is not None
+            assert info.tunnel.tunnel_key == tunnel.tunnel_key
+        finally:
+            devbox.shutdown()
+
 
 class TestDevboxCreationMethods:
     """Test various devbox creation methods."""