1818from runloop_api_client .sdk import AsyncDevbox , AsyncSnapshot , AsyncBlueprint , AsyncStorageObject
1919from runloop_api_client .sdk .async_ import (
2020 AsyncRunloopSDK ,
21- AsyncDevboxClient ,
22- AsyncSnapshotClient ,
23- AsyncBlueprintClient ,
24- AsyncStorageObjectClient ,
21+ AsyncDevboxOps ,
22+ AsyncSnapshotOps ,
23+ AsyncBlueprintOps ,
24+ AsyncStorageObjectOps ,
2525)
2626from runloop_api_client .lib .polling import PollingConfig
2727
@@ -34,7 +34,7 @@ async def test_create(self, mock_async_client: AsyncMock, devbox_view: MockDevbo
3434 """Test create method."""
3535 mock_async_client .devboxes .create_and_await_running = AsyncMock (return_value = devbox_view )
3636
37- client = AsyncDevboxClient (mock_async_client )
37+ client = AsyncDevboxOps (mock_async_client )
3838 devbox = await client .create (
3939 name = "test-devbox" ,
4040 metadata = {"key" : "value" },
@@ -50,7 +50,7 @@ async def test_create_from_blueprint_id(self, mock_async_client: AsyncMock, devb
5050 """Test create_from_blueprint_id method."""
5151 mock_async_client .devboxes .create_and_await_running = AsyncMock (return_value = devbox_view )
5252
53- client = AsyncDevboxClient (mock_async_client )
53+ client = AsyncDevboxOps (mock_async_client )
5454 devbox = await client .create_from_blueprint_id (
5555 "bp_123" ,
5656 name = "test-devbox" ,
@@ -65,7 +65,7 @@ async def test_create_from_blueprint_name(self, mock_async_client: AsyncMock, de
6565 """Test create_from_blueprint_name method."""
6666 mock_async_client .devboxes .create_and_await_running = AsyncMock (return_value = devbox_view )
6767
68- client = AsyncDevboxClient (mock_async_client )
68+ client = AsyncDevboxOps (mock_async_client )
6969 devbox = await client .create_from_blueprint_name (
7070 "my-blueprint" ,
7171 name = "test-devbox" ,
@@ -80,7 +80,7 @@ async def test_create_from_snapshot(self, mock_async_client: AsyncMock, devbox_v
8080 """Test create_from_snapshot method."""
8181 mock_async_client .devboxes .create_and_await_running = AsyncMock (return_value = devbox_view )
8282
83- client = AsyncDevboxClient (mock_async_client )
83+ client = AsyncDevboxOps (mock_async_client )
8484 devbox = await client .create_from_snapshot (
8585 "snap_123" ,
8686 name = "test-devbox" ,
@@ -92,7 +92,7 @@ async def test_create_from_snapshot(self, mock_async_client: AsyncMock, devbox_v
9292
9393 def test_from_id (self , mock_async_client : AsyncMock ) -> None :
9494 """Test from_id method."""
95- client = AsyncDevboxClient (mock_async_client )
95+ client = AsyncDevboxOps (mock_async_client )
9696 devbox = client .from_id ("dev_123" )
9797
9898 assert isinstance (devbox , AsyncDevbox )
@@ -107,7 +107,7 @@ async def test_list(self, mock_async_client: AsyncMock, devbox_view: MockDevboxV
107107 page = SimpleNamespace (devboxes = [devbox_view ])
108108 mock_async_client .devboxes .list = AsyncMock (return_value = page )
109109
110- client = AsyncDevboxClient (mock_async_client )
110+ client = AsyncDevboxOps (mock_async_client )
111111 devboxes = await client .list (
112112 limit = 10 ,
113113 status = "running" ,
@@ -129,7 +129,7 @@ async def test_list(self, mock_async_client: AsyncMock, snapshot_view: MockSnaps
129129 page = SimpleNamespace (snapshots = [snapshot_view ])
130130 mock_async_client .devboxes .disk_snapshots .list = AsyncMock (return_value = page )
131131
132- client = AsyncSnapshotClient (mock_async_client )
132+ client = AsyncSnapshotOps (mock_async_client )
133133 snapshots = await client .list (
134134 devbox_id = "dev_123" ,
135135 limit = 10 ,
@@ -143,7 +143,7 @@ async def test_list(self, mock_async_client: AsyncMock, snapshot_view: MockSnaps
143143
144144 def test_from_id (self , mock_async_client : AsyncMock ) -> None :
145145 """Test from_id method."""
146- client = AsyncSnapshotClient (mock_async_client )
146+ client = AsyncSnapshotOps (mock_async_client )
147147 snapshot = client .from_id ("snap_123" )
148148
149149 assert isinstance (snapshot , AsyncSnapshot )
@@ -158,7 +158,7 @@ async def test_create(self, mock_async_client: AsyncMock, blueprint_view: MockBl
158158 """Test create method."""
159159 mock_async_client .blueprints .create_and_await_build_complete = AsyncMock (return_value = blueprint_view )
160160
161- client = AsyncBlueprintClient (mock_async_client )
161+ client = AsyncBlueprintOps (mock_async_client )
162162 blueprint = await client .create (
163163 name = "test-blueprint" ,
164164 polling_config = PollingConfig (timeout_seconds = 60.0 ),
@@ -170,7 +170,7 @@ async def test_create(self, mock_async_client: AsyncMock, blueprint_view: MockBl
170170
171171 def test_from_id (self , mock_async_client : AsyncMock ) -> None :
172172 """Test from_id method."""
173- client = AsyncBlueprintClient (mock_async_client )
173+ client = AsyncBlueprintOps (mock_async_client )
174174 blueprint = client .from_id ("bp_123" )
175175
176176 assert isinstance (blueprint , AsyncBlueprint )
@@ -182,7 +182,7 @@ async def test_list(self, mock_async_client: AsyncMock, blueprint_view: MockBlue
182182 page = SimpleNamespace (blueprints = [blueprint_view ])
183183 mock_async_client .blueprints .list = AsyncMock (return_value = page )
184184
185- client = AsyncBlueprintClient (mock_async_client )
185+ client = AsyncBlueprintOps (mock_async_client )
186186 blueprints = await client .list (
187187 limit = 10 ,
188188 name = "test" ,
@@ -203,7 +203,7 @@ async def test_create(self, mock_async_client: AsyncMock, object_view: MockObjec
203203 """Test create method."""
204204 mock_async_client .objects .create = AsyncMock (return_value = object_view )
205205
206- client = AsyncStorageObjectClient (mock_async_client )
206+ client = AsyncStorageObjectOps (mock_async_client )
207207 obj = await client .create (name = "test.txt" , content_type = "text" , metadata = {"key" : "value" })
208208
209209 assert isinstance (obj , AsyncStorageObject )
@@ -217,7 +217,7 @@ async def test_create(self, mock_async_client: AsyncMock, object_view: MockObjec
217217
218218 def test_from_id (self , mock_async_client : AsyncMock ) -> None :
219219 """Test from_id method."""
220- client = AsyncStorageObjectClient (mock_async_client )
220+ client = AsyncStorageObjectOps (mock_async_client )
221221 obj = client .from_id ("obj_123" )
222222
223223 assert isinstance (obj , AsyncStorageObject )
@@ -230,7 +230,7 @@ async def test_list(self, mock_async_client: AsyncMock, object_view: MockObjectV
230230 page = SimpleNamespace (objects = [object_view ])
231231 mock_async_client .objects .list = AsyncMock (return_value = page )
232232
233- client = AsyncStorageObjectClient (mock_async_client )
233+ client = AsyncStorageObjectOps (mock_async_client )
234234 objects = await client .list (
235235 content_type = "text" ,
236236 limit = 10 ,
@@ -261,7 +261,7 @@ async def test_upload_from_file(
261261 http_client .put = AsyncMock (return_value = mock_response )
262262 mock_async_client ._client = http_client
263263
264- client = AsyncStorageObjectClient (mock_async_client )
264+ client = AsyncStorageObjectOps (mock_async_client )
265265 obj = await client .upload_from_file (temp_file , name = "test.txt" )
266266
267267 assert isinstance (obj , AsyncStorageObject )
@@ -281,7 +281,7 @@ async def test_upload_from_text(self, mock_async_client: AsyncMock, object_view:
281281 http_client .put = AsyncMock (return_value = mock_response )
282282 mock_async_client ._client = http_client
283283
284- client = AsyncStorageObjectClient (mock_async_client )
284+ client = AsyncStorageObjectOps (mock_async_client )
285285 obj = await client .upload_from_text ("test content" , "test.txt" , metadata = {"key" : "value" })
286286
287287 assert isinstance (obj , AsyncStorageObject )
@@ -305,7 +305,7 @@ async def test_upload_from_bytes(self, mock_async_client: AsyncMock, object_view
305305 http_client .put = AsyncMock (return_value = mock_response )
306306 mock_async_client ._client = http_client
307307
308- client = AsyncStorageObjectClient (mock_async_client )
308+ client = AsyncStorageObjectOps (mock_async_client )
309309 obj = await client .upload_from_bytes (b"test content" , "test.bin" , content_type = "binary" )
310310
311311 assert isinstance (obj , AsyncStorageObject )
@@ -321,7 +321,7 @@ async def test_upload_from_bytes(self, mock_async_client: AsyncMock, object_view
321321 @pytest .mark .asyncio
322322 async def test_upload_from_file_missing_path (self , mock_async_client : AsyncMock , tmp_path : Path ) -> None :
323323 """upload_from_file should raise when file cannot be read."""
324- client = AsyncStorageObjectClient (mock_async_client )
324+ client = AsyncStorageObjectOps (mock_async_client )
325325 missing_file = tmp_path / "missing.txt"
326326
327327 with pytest .raises (OSError , match = "Failed to read file" ):
@@ -335,10 +335,10 @@ def test_init(self) -> None:
335335 """Test AsyncRunloopSDK initialization."""
336336 sdk = AsyncRunloopSDK (bearer_token = "test-token" )
337337 assert sdk .api is not None
338- assert isinstance (sdk .devbox , AsyncDevboxClient )
339- assert isinstance (sdk .snapshot , AsyncSnapshotClient )
340- assert isinstance (sdk .blueprint , AsyncBlueprintClient )
341- assert isinstance (sdk .storage_object , AsyncStorageObjectClient )
338+ assert isinstance (sdk .devbox , AsyncDevboxOps )
339+ assert isinstance (sdk .snapshot , AsyncSnapshotOps )
340+ assert isinstance (sdk .blueprint , AsyncBlueprintOps )
341+ assert isinstance (sdk .storage_object , AsyncStorageObjectOps )
342342
343343 @pytest .mark .asyncio
344344 async def test_aclose (self ) -> None :
0 commit comments