moved code from bluesky

src/ophyd_async/plan_stubs/_fly.py

@@ -1,6 +1,7 @@
 from typing import List, Optional
 
 import bluesky.plan_stubs as bps
+from bluesky.utils import short_uid
 
 from ophyd_async.core import (
     DetectorTrigger,
@@ -117,10 +118,19 @@ def fly_and_collect(
     yield from bps.kickoff(flyer, wait=True)
     for detector in detectors:
         yield from bps.kickoff(detector)
-
-    yield from bps.collect_while_completing(
-        flyers=[flyer], dets=detectors, flush_period=0.5, stream_name=stream_name
-    )
+    group = short_uid(label="complete")
+    for detector in detectors:
+        yield from bps.complete(detector, group=group, wait=False)
+    yield from bps.complete(flyer, group=group, wait=False)
+    done = False
+    while not done:
+        try:
+            yield from bps.wait(group=group, timeout=0.5, move_on=True)
+        except TimeoutError:
+            pass
+        else:
+            done = True
+        yield from bps.collect(*detectors, name=stream_name)
 
 
 def fly_and_collect_with_static_pcomp(

tests/plan_stubs/test_fly.py

@@ -114,8 +114,9 @@ def __init__(
         super().__init__(controller, writer, config_sigs, name)
 
     @WatchableAsyncStatus.wrap
-    async def kickoff(self):
-        super().kickoff()
+    async def complete(self):
+        assert self._arm_status, "Prepare not run"
+        assert self._trigger_info
         self.writer.increment_index()
         async for index in self.writer.observe_indices_written(
             self._trigger_info.frame_timeout