Mock the ISPyBSession object directly

Author: tieneupin

tests/workflows/clem/test_register_preprocessing_results.py

@@ -185,20 +185,22 @@ def test_run_with_db(
     # 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_ispyb_session_factory = MagicMock(return_value=ispyb_db_session)
-    mock_ispyb_session_factory.return_value = ispyb_db_session
-    # mock_ispyb_session_factory.return_value.__enter__.return_value = ispyb_db_session
-    # mock_ispyb_sessionmaker = mocker.patch(
     mocker.patch(
-        "murfey.server.ispyb.sessionmaker", return_value=mock_ispyb_session_factory
+        "murfey.server.ispyb.get_security_config",
+        return_value=mock_security_config,
     )
+
+    # mock_ispyb_session_factory = MagicMock(return_value=ispyb_db_session)
+    # mock_ispyb_session_factory.return_value = ispyb_db_session
+    # # mock_ispyb_session_factory.return_value.__enter__.return_value = ispyb_db_session
+    # # mock_ispyb_sessionmaker = mocker.patch(
     # mocker.patch(
-    #     "murfey.server.ispyb.ISPyBSession",
-    #     return_value=ispyb_db_session,
+    #     "murfey.server.ispyb.sessionmaker", return_value=mock_ispyb_session_factory
     # )
+    mocker.patch(
+        "murfey.server.ispyb.ISPyBSession",
+        return_value=ispyb_db_session,
+    )
 
     # Mock the ISPYB connection when registering data collection group
     mocker.patch(