Added Lazy Tests

Author: JohannesRabauer

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

@@ -338,15 +338,46 @@ void lazyClearAfterSave(@Autowired final ObjectWithLazyRepository<SimpleObject>
 		final ObjectWithLazy<SimpleObject> newLazy = new ObjectWithLazy<>();
 		final SimpleObject objectToStore = new SimpleObject(TestData.DUMMY_STRING);
 		newLazy.setLazy(SpringDataEclipseStoreLazy.build(objectToStore));
+		
+		Assertions.assertTrue(newLazy.getLazy().isLoaded());
+		Assertions.assertFalse(newLazy.getLazy().isStored());
+		
 		repository.save(newLazy);
+		
+		Assertions.assertTrue(newLazy.getLazy().isLoaded());
+		Assertions.assertTrue(newLazy.getLazy().isStored());
+		
 		newLazy.getLazy().clear();
 
 		TestUtil.doBeforeAndAfterRestartOfDatastore(
 			this.configuration,
 			() -> {
 				Assertions.assertEquals(1, repository.findAll().size());
 				final Lazy<SimpleObject> lazy = repository.findAll().get(0).getLazy();
+				Assertions.assertFalse(lazy.isLoaded());
+				Assertions.assertEquals(objectToStore, lazy.get());
+				Assertions.assertTrue(lazy.isLoaded());
+			}
+		);
+	}
+	
+	@Test
+	void lazyClearThroughLazyManagerAfterSave(@Autowired final ObjectWithLazyRepository<SimpleObject> repository)
+	{
+		final ObjectWithLazy<SimpleObject> newLazy = new ObjectWithLazy<>();
+		final SimpleObject objectToStore = new SimpleObject(TestData.DUMMY_STRING);
+		newLazy.setLazy(SpringDataEclipseStoreLazy.build(objectToStore));
+		repository.save(newLazy);
+		LazyReferenceManager.get().cleanUp();
+		
+		TestUtil.doBeforeAndAfterRestartOfDatastore(
+			this.configuration,
+			() -> {
+				Assertions.assertEquals(1, repository.findAll().size());
+				final Lazy<SimpleObject> lazy = repository.findAll().get(0).getLazy();
+				Assertions.assertFalse(lazy.isLoaded());
 				Assertions.assertEquals(objectToStore, lazy.get());
+				Assertions.assertTrue(lazy.isLoaded());
 			}
 		);
 	}