fix: Fix Dataset.iter_items for apify_storage (#321)

Fix iter_items to correctly work in async iteration. Add test.

- closes #320

Author: Pijukatel

src/apify/apify_storage_client/_dataset_client.py

@@ -91,7 +91,7 @@ async def iterate_items(
         skip_empty: bool = False,
         skip_hidden: bool = False,
     ) -> AsyncIterator[dict]:
-        return self._client.iterate_items(
+        async for item in self._client.iterate_items(
             offset=offset,
             limit=limit,
             clean=clean,
@@ -101,7 +101,8 @@ async def iterate_items(
             unwind=unwind,
             skip_empty=skip_empty,
             skip_hidden=skip_hidden,
-        )
+        ):
+            yield item
 
     @override
     async def get_items_as_bytes(

tests/integration/test_actor_dataset.py

@@ -54,6 +54,26 @@ async def main() -> None:
         assert item['str'] == 'x' * 10000
 
 
+async def test_dataset_iter_items(
+    make_actor: MakeActorFunction,
+    run_actor: RunActorFunction,
+) -> None:
+    async def main() -> None:
+        inserted_data = {'Something': 'something else'}
+
+        async with Actor:
+            dataset = await Actor.open_dataset()
+            await dataset.push_data(inserted_data)
+            requested_data = [item async for item in dataset.iterate_items()]
+
+            assert len(requested_data) == 1
+            assert requested_data[0] == inserted_data
+
+    actor = await make_actor(label='test_dataset_iter_items', main_func=main)
+    run_result = await run_actor(actor)
+    assert run_result.status == 'SUCCEEDED'
+
+
 async def test_same_references_in_default_dataset(
     make_actor: MakeActorFunction,
     run_actor: RunActorFunction,