fixed tests to expect updated parameter/member names

Author: sid-rl

tests/sdk/async_devbox/test_interfaces.py

@@ -14,7 +14,7 @@
 
 from tests.sdk.conftest import MockExecutionView
 from runloop_api_client.sdk import AsyncDevbox
-from runloop_api_client._types import NotGiven
+from runloop_api_client._types import NotGiven, Omit
 
 
 class TestAsyncCommandInterface:
@@ -34,7 +34,7 @@ async def test_exec_without_callbacks(
         assert await result.stdout() == "output"
         call_kwargs = mock_async_client.devboxes.execute_and_await_completion.call_args[1]
         assert call_kwargs["command"] == "echo hello"
-        assert isinstance(call_kwargs["shell_name"], NotGiven) or call_kwargs["shell_name"] is None
+        assert isinstance(call_kwargs["shell_name"], Omit)
         assert isinstance(call_kwargs["timeout"], NotGiven)
 
     @pytest.mark.asyncio

tests/sdk/devbox/test_interfaces.py

@@ -14,7 +14,7 @@
 
 from tests.sdk.conftest import MockExecutionView
 from runloop_api_client.sdk import Devbox
-from runloop_api_client._types import NotGiven
+from runloop_api_client._types import NotGiven, Omit
 
 
 class TestCommandInterface:
@@ -31,7 +31,7 @@ def test_exec_without_callbacks(self, mock_client: Mock, execution_view: MockExe
         assert result.stdout() == "output"
         call_kwargs = mock_client.devboxes.execute_and_await_completion.call_args[1]
         assert call_kwargs["command"] == "echo hello"
-        assert isinstance(call_kwargs["shell_name"], NotGiven) or call_kwargs["shell_name"] is None
+        assert isinstance(call_kwargs["shell_name"], Omit)
         assert call_kwargs["polling_config"] is None
         assert isinstance(call_kwargs["timeout"], NotGiven)
 

tests/sdk/test_async_clients.py

@@ -127,7 +127,7 @@ class TestAsyncSnapshotClient:
     @pytest.mark.asyncio
     async def test_list(self, mock_async_client: AsyncMock, snapshot_view: MockSnapshotView) -> None:
         """Test list method."""
-        page = SimpleNamespace(disk_snapshots=[snapshot_view])
+        page = SimpleNamespace(snapshots=[snapshot_view])
         mock_async_client.devboxes.disk_snapshots.list = AsyncMock(return_value=page)
 
         client = AsyncSnapshotClient(mock_async_client)

tests/sdk/test_clients.py

@@ -120,7 +120,7 @@ class TestSnapshotClient:
 
     def test_list(self, mock_client: Mock, snapshot_view: MockSnapshotView) -> None:
         """Test list method."""
-        page = SimpleNamespace(disk_snapshots=[snapshot_view])
+        page = SimpleNamespace(snapshots=[snapshot_view])
         mock_client.devboxes.disk_snapshots.list.return_value = page
 
         client = SnapshotClient(mock_client)