update tests for upgrading

MorphiaOrg · Mar 4, 2025 · 39e8e61 · 39e8e61
1 parent 7068ab1
commit 39e8e61
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/core/src/main/java/dev/morphia/query/Operations.java b/core/src/main/java/dev/morphia/query/Operations.java
@@ -43,7 +43,7 @@ public Operations(MorphiaDatastore datastore, @Nullable EntityModel model, List<
         updates.forEach(this::add);
     }
 
-    private void add(UpdateOperator update) {
+    public void add(UpdateOperator update) {
         update.validate(validate);
         update.model(model);
         update.datastore(datastore);

diff --git a/core/src/test/java/dev/morphia/test/TestUpdateOperations.java b/core/src/test/java/dev/morphia/test/TestUpdateOperations.java
@@ -24,6 +24,8 @@
 import dev.morphia.annotations.Id;
 import dev.morphia.annotations.Indexed;
 import dev.morphia.annotations.PreLoad;
+import dev.morphia.internal.PathTarget;
+import dev.morphia.mapping.Mapper;
 import dev.morphia.query.FindOptions;
 import dev.morphia.query.Operations;
 import dev.morphia.query.Query;
@@ -643,4 +645,15 @@ public void testPathTranslations(TestUpdateOperations.TranslationParams params)
         assertTrue(contains, format("failed to find '%s' in:%n%s", format, json));
     }
 
+    private Document toDocument(Class type, UpdateOperator update) {
+        var ds = getDs();
+        Mapper mapper = ds.getMapper();
+        final Operations operations = new Operations(ds, mapper.getEntityModel(type), List.of(update), false);
+
+        PathTarget pathTarget = new PathTarget(mapper, mapper.getEntityModel(type), update.field(), false);
+        (update).datastore(ds);
+        operations.add(update);
+        return operations.toDocument(ds);
+    }
+
 }
diff --git a/rewrite/pom.xml b/rewrite/pom.xml
@@ -11,7 +11,7 @@
         <relativePath>../pom.xml</relativePath>
     </parent>
 
-    <artifactId>rewrite</artifactId>
+    <artifactId>morphia-rewrite</artifactId>
 
     <properties>
         <morphia.prior.version>2.4.14</morphia.prior.version>