From 8393b6109ae4e6fd56eb3650babd7b7ae8c35ab6 Mon Sep 17 00:00:00 2001
From: David Erb <david.erb@diamond.ac.uk>
Date: Tue, 25 Apr 2023 04:56:53 +0000
Subject: [PATCH] fixes test by inserting soakdb3 head record

---
 tests/test_export.py | 63 +++++++++++++++++++++++++++++++++++++-------
 1 file changed, 53 insertions(+), 10 deletions(-)

diff --git a/tests/test_export.py b/tests/test_export.py
index 1e70bf5..359f788 100644
--- a/tests/test_export.py
+++ b/tests/test_export.py
@@ -2,11 +2,19 @@
 import logging
 from pathlib import Path
 
-# Base class for the tester.
 # API constants.
 from dls_servbase_lib.datafaces.context import Context as DlsServbaseDatafaceContext
 from dls_utilpack.visit import get_xchem_subdirectory
 
+# Soakdb3 database.
+from soakdb3_api.databases.constants import Tablenames
+
+# Client for direct access to the soakdb3 database for seeding it.
+from soakdb3_api.datafaces.context import Context as Soakdb3DatafaceClientContext
+from soakdb3_api.datafaces.datafaces import (
+    datafaces_get_default as soakdb3_datafaces_get_default,
+)
+
 # The service process startup/teardown context.
 from soakdb3_lib.datafaces.context import Context as Soakdb3DatafaceServerContext
 
@@ -67,6 +75,11 @@ async def _main_coroutine(self, constants, output_directory):
             soakdb3_dataface_specification
         )
 
+        # Make the soakdb3 CLIENT context.
+        soakdb3_client_context = Soakdb3DatafaceClientContext(
+            soakdb3_dataface_specification
+        )
+
         # Reference the dict entry for the xchembku dataface.
         xchembku_dataface_specification = multiconf_dict[
             "xchembku_dataface_specification"
@@ -93,20 +106,26 @@ async def _main_coroutine(self, constants, output_directory):
 
         # Start the soakdb3 server context which includes the direct or network-addressable service.
         async with soakdb3_server_context:
-            # Start the client context for the direct access to the xchembku.
-            async with xchembku_client_context:
-                # Start the dataface the gui uses for cookies.
-                async with servbase_dataface_context:
-                    # Start the gui client context.
-                    async with gui_client_context:
-                        # And the gui server context which starts the coro.
-                        async with gui_server_context:
-                            await self.__run_part1(constants, output_directory)
+            # Client for direct access to the soakdb3 database for seeding it.
+            async with soakdb3_client_context:
+                # Start the client context for the direct access to the xchembku.
+                async with xchembku_client_context:
+                    # Start the dataface the gui uses for cookies.
+                    async with servbase_dataface_context:
+                        # Start the gui client context.
+                        async with gui_client_context:
+                            # And the gui server context which starts the coro.
+                            async with gui_server_context:
+                                await self.__run_part1(constants, output_directory)
 
     # ----------------------------------------------------------------------------------------
 
     async def __run_part1(self, constants, output_directory):
         """ """
+
+        # Reference the soakdb3 dataface object which the context has set up as the default.
+        soakdb3_dataface = soakdb3_datafaces_get_default()
+
         # Reference the xchembku object which the context has set up as the default.
         self.__xchembku = xchembku_datafaces_get_default()
 
@@ -120,12 +139,36 @@ async def __run_part1(self, constants, output_directory):
             / get_xchem_subdirectory(self.visit)
         )
 
+        # Soakdb3 expects visitid to be a visit directory.
+        # This is because of how the soadkb3 VBA in the Excel works.
+        visitid = str(self.__visit_directory / "processing")
+
         self.__crystal_targets_directory = (
             self.__visit_directory / "processing/lab36/crystal-targets"
         )
 
         self.__crystal_targets_directory.mkdir(parents=True)
 
+        # ----------------------------------------------------------------
+        # Seed the necessary fields in the head table.
+
+        protein = "P1"
+        drop_volume = 3.1
+
+        head_record = {
+            "Protein": protein,
+            "DropVolume": drop_volume,
+        }
+
+        # Insert these fields as the (single) row in the soakdb3 database's head table.
+        await soakdb3_dataface.insert(  # type: ignore
+            visitid,
+            Tablenames.HEAD,
+            [head_record],
+        )
+
+        # ----------------------------------------------------------------
+
         await self.__export_initial()
 
         crystal_wells = []