fixed test after the sub branch merged

cellpack/autopack/DBRecipeHandler.py

@@ -411,7 +411,7 @@ def upload_gradients(self, gradients):
             gradient_doc = GradientDoc(settings=gradient)
             _, doc_id = gradient_doc.should_write(self.db, gradient_name)
             if doc_id:
-                print(f"gradients/{gradient_name} already exists in firestore")
+                print(f"gradients/{gradient_name} is already in firestore")
                 self.grad_to_path_map[gradient_name] = self.db.create_path(
                     "gradients", doc_id
                 )
@@ -422,8 +422,9 @@ def upload_gradients(self, gradients):
     def upload_objects(self, objects):
         for obj_name in objects:
             objects[obj_name]["name"] = obj_name
+            # modify a copy of objects to avoid key error when resolving local regions
             modify_objects = copy.deepcopy(objects)
-            # replace gradient name with path before uploading
+            # replace gradient name with path to check if gradient exists in db
             if "gradient" in modify_objects[obj_name]:
                 grad_name = modify_objects[obj_name]["gradient"]
                 modify_objects[obj_name]["gradient"] = self.grad_to_path_map[grad_name]

cellpack/tests/test_db_recipe_handler.py

@@ -1,6 +1,6 @@
 from cellpack.autopack.DBRecipeHandler import DBRecipeHandler
 from cellpack.tests.mocks.mock_db import MockDB
-
+from unittest.mock import MagicMock, patch
 mock_db = MockDB({})
 
 
@@ -65,12 +65,18 @@ def test_upload_objects():
 
 
 def test_upload_objects_with_gradient():
-    # the value of gradient in obj should be changed to the path instead of the name before uploading
     data = {"test": {"test_key": "test_value", "gradient": "test_grad_name"}}
-    object_doc = DBRecipeHandler(mock_db)
-    object_doc.grad_to_path_map = {"test_grad_name": "firebase:gradients/test_id"}
-    object_doc.upload_objects(data)
-    assert data["test"]["gradient"] == "firebase:gradients/test_id"
+    object_handler = DBRecipeHandler(mock_db)
+    object_handler.grad_to_path_map = {"test_grad_name": "firebase:gradients/test_id"}
+
+    with patch('cellpack.autopack.DBRecipeHandler.ObjectDoc', return_value=MagicMock()) as mock_object_doc:
+        mock_object_doc.return_value.should_write.return_value = (None, "firebase:gradients/test_id")
+        object_handler.upload_objects(data)
+        mock_object_doc.assert_called()
+        called_with_settings = mock_object_doc.call_args.kwargs["settings"] 
+    assert data["test"]["gradient"] == "test_grad_name"
+    assert called_with_settings["gradient"] == "firebase:gradients/test_id"
+
 
 
 def test_upload_compositions():