Fixed broken test

Author: tieneupin

tests/workflows/fib/test_register_atlas.py

@@ -312,45 +312,42 @@ def test_run_with_db(
     murfey_db_session.commit()
 
     # Mock the metadata returned from the image file
-    mock_metadata = [
-        FIBAtlasMetadata(
-            visit_name=visit_name,
-            file=test_file,
-            voltage=2000,
-            shift_x=0,
-            shift_y=0,
-            len_x=0.003072,
-            len_y=0.002048,
-            pos_x=0.003,
-            pos_y=0.0003,
-            pos_z=0.01,
-            rotation=-1.309,
-            tilt_alpha=0.8,
-            tilt_beta=0,
-            pixels_x=3072,
-            pixels_y=2048,
-            pixel_size_x=1e-6,
-            pixel_size_y=1e-6,
-        )
-        for test_file in test_files
-    ]
-    mocker.patch(
+    mock_metadata = FIBAtlasMetadata(
+        visit_name=visit_name,
+        file=test_files[0],
+        voltage=2000,
+        shift_x=0,
+        shift_y=0,
+        len_x=0.003072,
+        len_y=0.002048,
+        pos_x=0.003,
+        pos_y=0.0003,
+        pos_z=0.01,
+        rotation=-1.309,
+        tilt_alpha=0.8,
+        tilt_beta=0,
+        pixels_x=3072,
+        pixels_y=2048,
+        pixel_size_x=1e-6,
+        pixel_size_y=1e-6,
+    )
+    mock_parse = mocker.patch(
         "murfey.workflows.fib.register_atlas._parse_metadata",
-        side_effect=mock_metadata,
+        return_value=mock_metadata,
     )
 
     # Run the function and check that it's run through to completion
-    for test_file in test_files:
-        run(
-            session_id=session_id,
-            file=test_file,
-            murfey_db=murfey_db_session,
-        )
+    run(
+        session_id=session_id,
+        file=test_files[0],
+        murfey_db=murfey_db_session,
+    )
     search_results = murfey_db_session.exec(
         select(MurfeyDB.ImagingSite).where(
             MurfeyDB.ImagingSite.session_id == session_id
         )
     ).all()
 
+    assert mock_parse.call_count == 1
     assert len(search_results) == 1
-    assert search_results[0].image_path == str(mock_metadata[-1].file)
+    assert search_results[0].image_path == str(mock_metadata.file)