use the Impl class

don't access fields directly
MorphiaOrg · Jan 23, 2025 · a5a6ef8 · a5a6ef8
1 parent 033da3c
commit a5a6ef8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/core/src/test/java/dev/morphia/test/lifecycle/TestLifecycles.java b/core/src/test/java/dev/morphia/test/lifecycle/TestLifecycles.java
@@ -14,6 +14,7 @@
 import com.mongodb.lang.Nullable;
 
 import dev.morphia.Datastore;
+import dev.morphia.DatastoreImpl;
 import dev.morphia.EntityInterceptor;
 import dev.morphia.annotations.Entity;
 import dev.morphia.annotations.Id;
@@ -273,8 +274,9 @@ public boolean hasAnnotation(Class<? extends Annotation> type) {
 
         @Override
         public void prePersist(@NotNull Object ent, @NotNull Document document, @NotNull Datastore datastore) {
-            final List<PropertyModel> fieldsToTest = datastore.getMapper().getEntityModel(ent.getClass())
-                    .getProperties(NonNull.class);
+            DatastoreImpl ds = (DatastoreImpl) datastore;
+            final List<PropertyModel> fieldsToTest = ds.getMapper().getEntityModel(ent.getClass())
+                                                               .getProperties(NonNull.class);
             for (PropertyModel mf : fieldsToTest) {
                 if (mf.getValue(ent) == null) {
                     throw new NonNullValidationException(mf);

diff --git a/core/src/test/java/dev/morphia/test/mapping/TestMapper.java b/core/src/test/java/dev/morphia/test/mapping/TestMapper.java
@@ -39,7 +39,7 @@ private static void assertModelNotSame(EntityModel originalEntity, EntityModel c
         } else {
             assertNotSame(originalEntity.superClass, clonedEntity.superClass);
         }
-        originalEntity.subtypes.forEach(subtype -> {
+        originalEntity.getSubtypes().forEach(subtype -> {
             assertNotSame(subtype, findSubtype(clonedEntity, subtype));
         });
         originalEntity.getProperties().forEach(propertyModel -> {
@@ -49,7 +49,7 @@ private static void assertModelNotSame(EntityModel originalEntity, EntityModel c
     }
 
     private static EntityModel findSubtype(EntityModel clonedEntity, EntityModel subtype) {
-        return clonedEntity.subtypes.stream()
+        return clonedEntity.getSubtypes().stream()
                 .filter(clone -> clone.getType().equals(subtype.getType()))
                 .findFirst()
                 .orElseThrow(() -> new MappingException(