More layers

stephen-riggs · stephen-riggs · commit dcb94e562592 · 2025-11-25T17:13:28.000Z
diff --git a/tests/server/test_ispyb.py b/tests/server/test_ispyb.py
@@ -99,7 +99,9 @@ def test_update_data_collection_group(mock_transport, ispyb_db_session: Session)
     )
 
     transport_manager = TransportManager("PikaTransport")
-    with mock.patch("murfey.server.ispyb.ISPyBSession.__call__", ispyb_db_session):
+    with mock.patch(
+        "murfey.server.ispyb.ISPyBSession.__call__.__enter__", ispyb_db_session
+    ):
         transport_manager.do_update_data_collection_group(
             record=DataCollectionGroup(dataCollectionGroupId=1, experimentTypeId=2)
         )

Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,9 @@ def test_update_data_collection_group(mock_transport, ispyb_db_session: Session)`
`99`	`99`	`)`
`100`	`100`
`101`	`101`	`transport_manager = TransportManager("PikaTransport")`
`102`		`- with mock.patch("murfey.server.ispyb.ISPyBSession.__call__", ispyb_db_session):`
	`102`	`+ with mock.patch(`
	`103`	`+ "murfey.server.ispyb.ISPyBSession.__call__.__enter__", ispyb_db_session`
	`104`	`+ ):`
`103`	`105`	`transport_manager.do_update_data_collection_group(`
`104`	`106`	`record=DataCollectionGroup(dataCollectionGroupId=1, experimentTypeId=2)`
`105`	`107`	`)`