Begin fixing attach metadata tests

src/blueapi/preprocessors/attach_metadata.py

@@ -11,9 +11,8 @@
 
 
 def attach_metadata(
-    data_groups: List[str],
-    provider: VisitDirectoryProvider,
     plan: MsgGenerator,
+    provider: VisitDirectoryProvider,
 ) -> MsgGenerator:
     """Updates a directory provider default location for file storage."""
     staging = False
@@ -26,10 +25,12 @@ def attach_metadata(
         if (message.command == "stage") and (not staging and will_write_data):
             yield from bps.wait_for([provider.update])
             staging = True
+        elif message.command == "unstage":
+            staging = False
 
         if message.command == "open_run":
-            message.kwargs[DATA_SESSION] = provider().filename_prefix
-            message.kwargs[DATA_GROUPS] = data_groups
+            directory_info = provider()
+            message.kwargs[DATA_SESSION] = directory_info.filename_prefix
 
         yield message
 

src/blueapi/service/handler.py

@@ -2,7 +2,10 @@
 from functools import partial
 from typing import Mapping, Optional
 
-from dodal.parameters.gda_directory_provider import VisitDirectoryProvider
+from dodal.parameters.gda_directory_provider import (
+    VisitDirectoryProvider,
+    VisitServiceClient,
+)
 
 from blueapi.config import ApplicationConfig
 from blueapi.core import BlueskyContext
@@ -90,9 +93,9 @@ def setup_handler(
 
     if config:
         provider = VisitDirectoryProvider(
-            url=config.env.data_writing.visit_service_url,
             data_group_name=config.env.data_writing.group_name,
             data_directory=config.env.data_writing.visit_directory,
+            client=VisitServiceClient(config.env.data_writing.visit_service_url),
         )
 
         # Make all dodal devices created by the context use provider if they can