fix assert_called to assert_awaited

Author: sid-rl

tests/sdk/test_async_blueprint.py

@@ -38,7 +38,7 @@ async def test_get_info(self, mock_async_client: AsyncMock, blueprint_view: Mock
         )
 
         assert result == blueprint_view
-        mock_async_client.blueprints.retrieve.assert_called_once()
+        mock_async_client.blueprints.retrieve.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_logs(self, mock_async_client: AsyncMock) -> None:
@@ -55,7 +55,7 @@ async def test_logs(self, mock_async_client: AsyncMock) -> None:
         )
 
         assert result == logs_view
-        mock_async_client.blueprints.logs.assert_called_once()
+        mock_async_client.blueprints.logs.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_delete(self, mock_async_client: AsyncMock) -> None:
@@ -71,7 +71,7 @@ async def test_delete(self, mock_async_client: AsyncMock) -> None:
         )
 
         assert result is not None  # Verify return value is propagated
-        mock_async_client.blueprints.delete.assert_called_once()
+        mock_async_client.blueprints.delete.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_create_devbox(self, mock_async_client: AsyncMock, devbox_view: MockDevboxView) -> None:
@@ -87,4 +87,4 @@ async def test_create_devbox(self, mock_async_client: AsyncMock, devbox_view: Mo
         )
 
         assert devbox.id == "dev_123"
-        mock_async_client.devboxes.create_and_await_running.assert_called_once()
+        mock_async_client.devboxes.create_and_await_running.assert_awaited_once()

tests/sdk/test_async_execution_result.py

@@ -190,7 +190,7 @@ async def mock_iter():
         # Should stream full output
         output = await result.stdout()
         assert output == "line1\nline2\nline3\n"
-        mock_async_client.devboxes.executions.stream_stdout_updates.assert_called_once_with(
+        mock_async_client.devboxes.executions.stream_stdout_updates.assert_awaited_once_with(
             "exec_123", devbox_id="dev_123"
         )
 
@@ -226,7 +226,7 @@ async def mock_iter():
         # Should stream full output
         output = await result.stderr()
         assert output == "error1\nerror2\n"
-        mock_async_client.devboxes.executions.stream_stderr_updates.assert_called_once_with(
+        mock_async_client.devboxes.executions.stream_stderr_updates.assert_awaited_once_with(
             "exec_123", devbox_id="dev_123"
         )
 

tests/sdk/test_async_ops.py

@@ -47,7 +47,7 @@ async def test_create(self, mock_async_client: AsyncMock, devbox_view: MockDevbo
 
         assert isinstance(devbox, AsyncDevbox)
         assert devbox.id == "dev_123"
-        mock_async_client.devboxes.create_and_await_running.assert_called_once()
+        mock_async_client.devboxes.create_and_await_running.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_create_from_blueprint_id(self, mock_async_client: AsyncMock, devbox_view: MockDevboxView) -> None:
@@ -115,7 +115,7 @@ async def test_list_empty(self, mock_async_client: AsyncMock) -> None:
         devboxes = await ops.list(limit=10, status="running")
 
         assert len(devboxes) == 0
-        mock_async_client.devboxes.list.assert_called_once()
+        mock_async_client.devboxes.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_single(self, mock_async_client: AsyncMock, devbox_view: MockDevboxView) -> None:
@@ -133,7 +133,7 @@ async def test_list_single(self, mock_async_client: AsyncMock, devbox_view: Mock
         assert len(devboxes) == 1
         assert isinstance(devboxes[0], AsyncDevbox)
         assert devboxes[0].id == "dev_123"
-        mock_async_client.devboxes.list.assert_called_once()
+        mock_async_client.devboxes.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
@@ -151,7 +151,7 @@ async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
         assert isinstance(devboxes[1], AsyncDevbox)
         assert devboxes[0].id == "dev_001"
         assert devboxes[1].id == "dev_002"
-        mock_async_client.devboxes.list.assert_called_once()
+        mock_async_client.devboxes.list.assert_awaited_once()
 
 
 class TestAsyncSnapshotOps:
@@ -167,7 +167,7 @@ async def test_list_empty(self, mock_async_client: AsyncMock) -> None:
         snapshots = await ops.list(devbox_id="dev_123", limit=10)
 
         assert len(snapshots) == 0
-        mock_async_client.devboxes.disk_snapshots.list.assert_called_once()
+        mock_async_client.devboxes.disk_snapshots.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_single(self, mock_async_client: AsyncMock, snapshot_view: MockSnapshotView) -> None:
@@ -185,7 +185,7 @@ async def test_list_single(self, mock_async_client: AsyncMock, snapshot_view: Mo
         assert len(snapshots) == 1
         assert isinstance(snapshots[0], AsyncSnapshot)
         assert snapshots[0].id == "snap_123"
-        mock_async_client.devboxes.disk_snapshots.list.assert_called_once()
+        mock_async_client.devboxes.disk_snapshots.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
@@ -203,7 +203,7 @@ async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
         assert isinstance(snapshots[1], AsyncSnapshot)
         assert snapshots[0].id == "snap_001"
         assert snapshots[1].id == "snap_002"
-        mock_async_client.devboxes.disk_snapshots.list.assert_called_once()
+        mock_async_client.devboxes.disk_snapshots.list.assert_awaited_once()
 
     def test_from_id(self, mock_async_client: AsyncMock) -> None:
         """Test from_id method."""
@@ -230,7 +230,7 @@ async def test_create(self, mock_async_client: AsyncMock, blueprint_view: MockBl
 
         assert isinstance(blueprint, AsyncBlueprint)
         assert blueprint.id == "bp_123"
-        mock_async_client.blueprints.create_and_await_build_complete.assert_called_once()
+        mock_async_client.blueprints.create_and_await_build_complete.assert_awaited_once()
 
     def test_from_id(self, mock_async_client: AsyncMock) -> None:
         """Test from_id method."""
@@ -250,7 +250,7 @@ async def test_list_empty(self, mock_async_client: AsyncMock) -> None:
         blueprints = await ops.list(limit=10)
 
         assert len(blueprints) == 0
-        mock_async_client.blueprints.list.assert_called_once()
+        mock_async_client.blueprints.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_single(self, mock_async_client: AsyncMock, blueprint_view: MockBlueprintView) -> None:
@@ -268,7 +268,7 @@ async def test_list_single(self, mock_async_client: AsyncMock, blueprint_view: M
         assert len(blueprints) == 1
         assert isinstance(blueprints[0], AsyncBlueprint)
         assert blueprints[0].id == "bp_123"
-        mock_async_client.blueprints.list.assert_called_once()
+        mock_async_client.blueprints.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
@@ -286,7 +286,7 @@ async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
         assert isinstance(blueprints[1], AsyncBlueprint)
         assert blueprints[0].id == "bp_001"
         assert blueprints[1].id == "bp_002"
-        mock_async_client.blueprints.list.assert_called_once()
+        mock_async_client.blueprints.list.assert_awaited_once()
 
 
 class TestAsyncStorageObjectOps:
@@ -328,7 +328,7 @@ async def test_list_empty(self, mock_async_client: AsyncMock) -> None:
         objects = await ops.list(limit=10)
 
         assert len(objects) == 0
-        mock_async_client.objects.list.assert_called_once()
+        mock_async_client.objects.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_single(self, mock_async_client: AsyncMock, object_view: MockObjectView) -> None:
@@ -374,7 +374,7 @@ async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
         assert isinstance(objects[1], AsyncStorageObject)
         assert objects[0].id == "obj_001"
         assert objects[1].id == "obj_002"
-        mock_async_client.objects.list.assert_called_once()
+        mock_async_client.objects.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_upload_from_file(
@@ -653,7 +653,7 @@ async def test_create(self, mock_async_client: AsyncMock, scorer_view: MockScore
 
         assert isinstance(scorer, AsyncScorer)
         assert scorer.id == "scorer_123"
-        mock_async_client.scenarios.scorers.create.assert_called_once()
+        mock_async_client.scenarios.scorers.create.assert_awaited_once()
 
     def test_from_id(self, mock_async_client: AsyncMock) -> None:
         """Test from_id method."""
@@ -677,7 +677,7 @@ async def async_iter():
         scorers = await ops.list(limit=10)
 
         assert len(scorers) == 0
-        mock_async_client.scenarios.scorers.list.assert_called_once()
+        mock_async_client.scenarios.scorers.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_single(self, mock_async_client: AsyncMock, scorer_view: MockScorerView) -> None:
@@ -697,7 +697,7 @@ async def async_iter():
         assert len(scorers) == 1
         assert isinstance(scorers[0], AsyncScorer)
         assert scorers[0].id == "scorer_123"
-        mock_async_client.scenarios.scorers.list.assert_called_once()
+        mock_async_client.scenarios.scorers.list.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
@@ -719,7 +719,7 @@ async def async_iter():
         assert isinstance(scorers[1], AsyncScorer)
         assert scorers[0].id == "scorer_001"
         assert scorers[1].id == "scorer_002"
-        mock_async_client.scenarios.scorers.list.assert_called_once()
+        mock_async_client.scenarios.scorers.list.assert_awaited_once()
 
 
 class TestAsyncRunloopSDK:

tests/sdk/test_async_scorer.py

@@ -38,7 +38,7 @@ async def test_get_info(self, mock_async_client: AsyncMock, scorer_view: MockSco
         result = await scorer.get_info()
 
         assert result == scorer_view
-        mock_async_client.scenarios.scorers.retrieve.assert_called_once()
+        mock_async_client.scenarios.scorers.retrieve.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_update(self, mock_async_client: AsyncMock) -> None:
@@ -53,7 +53,7 @@ async def test_update(self, mock_async_client: AsyncMock) -> None:
         )
 
         assert result == update_response
-        mock_async_client.scenarios.scorers.update.assert_called_once()
+        mock_async_client.scenarios.scorers.update.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_validate(self, mock_async_client: AsyncMock) -> None:
@@ -71,4 +71,4 @@ async def test_validate(self, mock_async_client: AsyncMock) -> None:
         )
 
         assert result == validate_response
-        mock_async_client.scenarios.scorers.validate.assert_called_once()
+        mock_async_client.scenarios.scorers.validate.assert_awaited_once()

tests/sdk/test_async_snapshot.py

@@ -39,7 +39,7 @@ async def test_get_info(self, mock_async_client: AsyncMock, snapshot_view: MockS
         )
 
         assert result == snapshot_view
-        mock_async_client.devboxes.disk_snapshots.query_status.assert_called_once()
+        mock_async_client.devboxes.disk_snapshots.query_status.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_update(self, mock_async_client: AsyncMock) -> None:
@@ -60,7 +60,7 @@ async def test_update(self, mock_async_client: AsyncMock) -> None:
         )
 
         assert result == updated_snapshot
-        mock_async_client.devboxes.disk_snapshots.update.assert_called_once()
+        mock_async_client.devboxes.disk_snapshots.update.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_delete(self, mock_async_client: AsyncMock) -> None:
@@ -77,7 +77,7 @@ async def test_delete(self, mock_async_client: AsyncMock) -> None:
         )
 
         assert result is not None  # Verify return value is propagated
-        mock_async_client.devboxes.disk_snapshots.delete.assert_called_once()
+        mock_async_client.devboxes.disk_snapshots.delete.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_await_completed(self, mock_async_client: AsyncMock, snapshot_view: MockSnapshotView) -> None:
@@ -95,7 +95,7 @@ async def test_await_completed(self, mock_async_client: AsyncMock, snapshot_view
         )
 
         assert result == snapshot_view
-        mock_async_client.devboxes.disk_snapshots.await_completed.assert_called_once()
+        mock_async_client.devboxes.disk_snapshots.await_completed.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_create_devbox(self, mock_async_client: AsyncMock, devbox_view: MockDevboxView) -> None:
@@ -111,4 +111,4 @@ async def test_create_devbox(self, mock_async_client: AsyncMock, devbox_view: Mo
         )
 
         assert devbox.id == "dev_123"
-        mock_async_client.devboxes.create_and_await_running.assert_called_once()
+        mock_async_client.devboxes.create_and_await_running.assert_awaited_once()

tests/sdk/test_async_storage_object.py

@@ -45,7 +45,7 @@ async def test_refresh(self, mock_async_client: AsyncMock, object_view: MockObje
         )
 
         assert result == object_view
-        mock_async_client.objects.retrieve.assert_called_once()
+        mock_async_client.objects.retrieve.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_complete(self, mock_async_client: AsyncMock) -> None:
@@ -66,7 +66,7 @@ async def test_complete(self, mock_async_client: AsyncMock) -> None:
 
         assert result == completed_view
         assert obj.upload_url is None
-        mock_async_client.objects.complete.assert_called_once()
+        mock_async_client.objects.complete.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_get_download_url_without_duration(self, mock_async_client: AsyncMock) -> None:
@@ -83,7 +83,7 @@ async def test_get_download_url_without_duration(self, mock_async_client: AsyncM
         )
 
         assert result == download_url_view
-        mock_async_client.objects.download.assert_called_once()
+        mock_async_client.objects.download.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_get_download_url_with_duration(self, mock_async_client: AsyncMock) -> None:
@@ -101,7 +101,7 @@ async def test_get_download_url_with_duration(self, mock_async_client: AsyncMock
         )
 
         assert result == download_url_view
-        mock_async_client.objects.download.assert_called_once()
+        mock_async_client.objects.download.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_download_as_bytes(self, mock_async_client: AsyncMock) -> None:
@@ -160,7 +160,7 @@ async def test_delete(self, mock_async_client: AsyncMock, object_view: MockObjec
         )
 
         assert result == object_view
-        mock_async_client.objects.delete.assert_called_once()
+        mock_async_client.objects.delete.assert_awaited_once()
 
     @pytest.mark.asyncio
     async def test_upload_content_string(self, mock_async_client: AsyncMock) -> None: