replace Datastore uses

MorphiaOrg · Jan 23, 2025 · 033da3c · 033da3c
1 parent bfe59ee
commit 033da3c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/core/src/test/java/dev/morphia/test/mapping/TestReferences.java b/core/src/test/java/dev/morphia/test/mapping/TestReferences.java
@@ -17,6 +17,7 @@
 import com.mongodb.client.model.ReturnDocument;
 
 import dev.morphia.Datastore;
+import dev.morphia.DatastoreImpl;
 import dev.morphia.Key;
 import dev.morphia.ModifyOptions;
 import dev.morphia.aggregation.Aggregation;
@@ -315,7 +316,7 @@ public void testInQueryAgainstReferences() {
 
     @Test
     public void testLazyWithParent() {
-        Datastore datastore = getDs();
+        DatastoreImpl datastore = getDs();
         datastore.getMapper().map(Entity1.class, Entity2.class, EntityBase.class);
 
         Entity1 entity1 = new Entity1("entity1");
@@ -403,8 +404,8 @@ public final void testMultiDimArrayPersistence() {
 
     @Test
     public void testMultipleDatabasesSingleThreaded() {
-        final Datastore ds1 = createDatastore(getMongoClient(), "db1");
-        final Datastore ds2 = createDatastore(getMongoClient(), "db2");
+        final DatastoreImpl ds1 = (DatastoreImpl) createDatastore(getMongoClient(), "db1");
+        final DatastoreImpl ds2 = (DatastoreImpl) createDatastore(getMongoClient(), "db2");
 
         final FacebookUser db1Friend = new FacebookUser(3, "DB1 FaceBook Friend");
         ds1.save(db1Friend);