Expanded Test to look for lazy implementation

Author: JohannesRabauer

spring-data-eclipse-store/src/main/java/software/xdev/spring/data/eclipse/store/repository/EclipseStoreStorage.java

@@ -498,4 +498,13 @@ public ReadWriteLock getReadWriteLock()
 	{
 		return this.readWriteLock;
 	}
+	
+	/**
+	 * <b>Warning!</b> Please be very cautious if you access the root object.
+	 * This should only be done if absolutely necessary!
+	 */
+	public VersionedRoot getRoot()
+	{
+		return root;
+	}
 }

spring-data-eclipse-store/src/main/java/software/xdev/spring/data/eclipse/store/repository/root/v2_4/LazyEntityData.java

@@ -186,4 +186,9 @@ public Collection<Object> removeAllEntitiesAndReturnObjectsToStore()
 		this.entitiesById.clear();
 		return this.getObjectsToStore();
 	}
+	
+	public HashMap<ID, Lazy<T>> getNativeLazyEntitiesById()
+	{
+		return entitiesById;
+	}
 }

spring-data-eclipse-store/src/test/java/software/xdev/spring/data/eclipse/store/integration/isolated/tests/lazy/LazyTest.java

@@ -19,6 +19,7 @@
 
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Optional;
 
@@ -37,6 +38,8 @@
 import software.xdev.spring.data.eclipse.store.helper.TestUtil;
 import software.xdev.spring.data.eclipse.store.integration.isolated.IsolatedTestAnnotations;
 import software.xdev.spring.data.eclipse.store.repository.lazy.SpringDataEclipseStoreLazy;
+import software.xdev.spring.data.eclipse.store.repository.root.v2_4.EntityData;
+import software.xdev.spring.data.eclipse.store.repository.root.v2_4.LazyEntityData;
 
 
 @SuppressWarnings("checkstyle:MethodName")
@@ -302,7 +305,6 @@ void lazyClearBeforeSave()
 	}
 
 	@Test
-	@Disabled("This should work at some point. At least a warning should be displayed.")
 	void lazyUseEclipseStoreLazy(@Autowired final ObjectWithLazyRepository<SimpleObject> repository)
 	{
 		final ObjectWithLazy<SimpleObject> newLazy = new ObjectWithLazy<>();
@@ -515,19 +517,25 @@ void simpleEntityWithIdLazyWrappedRepository_FindById(
 	void simpleEntityWithIdLazyRepository_FindById(@Autowired final SimpleEntityWithIdLazyRepository repository)
 	{
 		final SimpleEntityWithId objectToStore1 = new SimpleEntityWithId(TestData.DUMMY_STRING);
-		repository.save(objectToStore1);
+		final Long object1Id = repository.save(objectToStore1).getId();
 		final SimpleEntityWithId objectToStore2 = new SimpleEntityWithId(TestData.DUMMY_STRING);
-		repository.save(objectToStore2);
-		
-		final List<SimpleEntityWithId> all = repository.findAll();
+		final Long object2Id = repository.save(objectToStore2).getId();
 
 		TestUtil.doBeforeAndAfterRestartOfDatastore(
 			this.configuration,
 			() -> {
-				final Optional<SimpleEntityWithId> reloadedObject =
-					repository.findById(all.get(0).getId());
+				LazyReferenceManager.get().cleanUp();
+				final Optional<SimpleEntityWithId> reloadedObject = repository.findById(object1Id);
 				Assertions.assertTrue(reloadedObject.isPresent());
 			}
 		);
+		final EntityData<SimpleEntityWithId, Long> entityData = this.configuration.getStorageInstance()
+			.getRoot()
+			.getCurrentRootData()
+			.getEntityData(SimpleEntityWithId.class);
+		final HashMap<Long, Lazy<SimpleEntityWithId>> lazyEntitiesById =
+			((LazyEntityData<SimpleEntityWithId, Long>)entityData).getNativeLazyEntitiesById();
+		Assertions.assertTrue(lazyEntitiesById.get(object1Id).isLoaded());
+		Assertions.assertFalse(lazyEntitiesById.get(object2Id).isLoaded());
 	}
 }