Mock the ISPyBSession instance imported in 'register_data_collection_group'

Author: tieneupin

tests/workflows/clem/test_register_preprocessing_results.py

@@ -182,16 +182,21 @@ def test_run_with_db(
         },
     )
 
-    # Mock out module-level functions
+    # Mock the ISPyB connection where the TransportManager class is located
     mock_security_config = MagicMock()
     mock_security_config.ispyb_credentials = mock_ispyb_credentials
     mock_get_security_config = mocker.patch("murfey.server.ispyb.get_security_config")
     mock_get_security_config.return_value = mock_security_config
 
-    # Mock the ISPyB connection
     mock_ispyb_sessionmaker = mocker.patch("murfey.server.ispyb.sessionmaker")
     mock_ispyb_sessionmaker.return_value.return_value = ispyb_db_session
 
+    # Mock the ISPYB connection when registering data collection group
+    mocker.patch(
+        "murfey.workflows.register_data_collection_group.ISPyBSession",
+        return_value=ispyb_db_session,
+    )
+
     # Mock out the machine config used in the helper sanitisation function
     mock_get_machine_config = mocker.patch("murfey.workflows.clem.get_machine_config")
     mock_machine_config = MagicMock()