Merge pull request #709 from atlanhq/APP-8346

Aryamanz29 · web-flow · commit 7fb7919566ee · 2025-09-01T11:08:04.000+05:30
APP-8346 Fix problem in async iterator in AsyncWorkflowSearchResponse
diff --git a/pyatlan/client/aio/workflow.py b/pyatlan/client/aio/workflow.py
@@ -243,7 +243,7 @@ async def _find_runs(
         return AsyncWorkflowSearchResponse(
             client=self._client,
             endpoint=endpoint,
-            criteria=request_obj,
+            criteria=query,
             start=from_,
             size=size,
             **sync_response,
diff --git a/tests/integration/aio/test_workflow_client.py b/tests/integration/aio/test_workflow_client.py
@@ -191,6 +191,8 @@ async def test_workflow_get_runs_and_stop(
         [AtlanWorkflowPhase.FAILED], started_at="now-1h"
     )
     assert runs_status
+    async for _ in runs_status:
+        pass
 
 
 async def test_workflow_get_all_scheduled_runs(

Original file line number	Diff line number	Diff line change
`@@ -191,6 +191,8 @@ async def test_workflow_get_runs_and_stop(`
`191`	`191`	`[AtlanWorkflowPhase.FAILED], started_at="now-1h"`
`192`	`192`	`)`
`193`	`193`	`assert runs_status`
	`194`	`+ async for _ in runs_status:`
	`195`	`+ pass`
`194`	`196`
`195`	`197`
`196`	`198`	`async def test_workflow_get_all_scheduled_runs(`