Added lazyWithComplexId tests

Author: JohannesRabauer

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

@@ -0,0 +1,20 @@
+/*
+ * Copyright © 2024 XDEV Software (https://xdev.software)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package software.xdev.spring.data.eclipse.store.integration.isolated.tests.lazy;
+
+public record CompositeKeyAsRecord(int keyPart1, int keyPart2)
+{
+}

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

@@ -513,6 +513,23 @@ void simpleEntityWithIdLazyWrappedRepository_FindById(
 		Assertions.assertThrows(NoIdFieldFoundException.class, () -> repository.findById(all.get(0).get().getId()));
 	}
 
+	@Test
+	void simpleEntityWithIdLazyRepository_SimpleStore(@Autowired final SimpleEntityWithIdLazyRepository repository)
+	{
+		final SimpleEntityWithId objectToStore1 = new SimpleEntityWithId(TestData.DUMMY_STRING);
+		final Long object1Id = repository.save(objectToStore1).getId();
+		
+		TestUtil.doBeforeAndAfterRestartOfDatastore(
+			this.configuration,
+			() -> {
+				LazyReferenceManager.get().cleanUp();
+				final Optional<SimpleEntityWithId> reloadedObject = repository.findById(object1Id);
+				Assertions.assertTrue(reloadedObject.isPresent());
+				Assertions.assertEquals(objectToStore1, reloadedObject.get());
+			}
+		);
+	}
+	
 	@Test
 	void simpleEntityWithIdLazyRepository_FindById(@Autowired final SimpleEntityWithIdLazyRepository repository)
 	{
@@ -538,4 +555,34 @@ void simpleEntityWithIdLazyRepository_FindById(@Autowired final SimpleEntityWith
 		Assertions.assertTrue(lazyEntitiesById.get(object1Id).isLoaded());
 		Assertions.assertFalse(lazyEntitiesById.get(object2Id).isLoaded());
 	}
+	
+	@Test
+	void simpleEntityWithComplexIdLazyRepository_FindById(
+		@Autowired final SimpleEntityWithComplexIdLazyRepository repository)
+	{
+		final SimpleEntityWithComplexId objectToStore1 =
+			new SimpleEntityWithComplexId(new CompositeKeyAsRecord(1, 1), TestData.DUMMY_STRING);
+		final CompositeKeyAsRecord object1Id = repository.save(objectToStore1).getId();
+		final SimpleEntityWithComplexId objectToStore2 =
+			new SimpleEntityWithComplexId(new CompositeKeyAsRecord(1, 2), TestData.DUMMY_STRING);
+		final CompositeKeyAsRecord object2Id = repository.save(objectToStore2).getId();
+		
+		TestUtil.doBeforeAndAfterRestartOfDatastore(
+			this.configuration,
+			() -> {
+				LazyReferenceManager.get().cleanUp();
+				final Optional<SimpleEntityWithComplexId> reloadedObject = repository.findById(object1Id);
+				Assertions.assertTrue(reloadedObject.isPresent());
+			}
+		);
+		final EntityData<SimpleEntityWithComplexId, CompositeKeyAsRecord> entityData =
+			this.configuration.getStorageInstance()
+				.getRoot()
+				.getCurrentRootData()
+				.getEntityData(SimpleEntityWithComplexId.class);
+		final HashMap<CompositeKeyAsRecord, Lazy<SimpleEntityWithComplexId>> lazyEntitiesById =
+			((LazyEntityData<SimpleEntityWithComplexId, CompositeKeyAsRecord>)entityData).getNativeLazyEntitiesById();
+		Assertions.assertTrue(lazyEntitiesById.get(object1Id).isLoaded());
+		Assertions.assertFalse(lazyEntitiesById.get(object2Id).isLoaded());
+	}
 }

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

@@ -0,0 +1,55 @@
+/*
+ * Copyright © 2024 XDEV Software (https://xdev.software)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package software.xdev.spring.data.eclipse.store.integration.isolated.tests.lazy;
+
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.GenerationType;
+import jakarta.persistence.Id;
+
+
+public class SimpleEntityWithComplexId
+{
+	@Id
+	private CompositeKeyAsRecord id;
+	
+	private String name;
+	
+	public SimpleEntityWithComplexId(final CompositeKeyAsRecord id, final String name)
+	{
+		this.id = id;
+		this.name = name;
+	}
+	
+	public String getName()
+	{
+		return this.name;
+	}
+	
+	public void setName(final String name)
+	{
+		this.name = name;
+	}
+	
+	public CompositeKeyAsRecord getId()
+	{
+		return this.id;
+	}
+	
+	public void setId(final CompositeKeyAsRecord id)
+	{
+		this.id = id;
+	}
+}

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

@@ -0,0 +1,23 @@
+/*
+ * Copyright © 2024 XDEV Software (https://xdev.software)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package software.xdev.spring.data.eclipse.store.integration.isolated.tests.lazy;
+
+import software.xdev.spring.data.eclipse.store.repository.interfaces.lazy.LazyEclipseStoreRepository;
+
+
+public interface SimpleEntityWithComplexIdLazyRepository extends LazyEclipseStoreRepository<SimpleEntityWithComplexId, CompositeKeyAsRecord>
+{
+}

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

@@ -15,6 +15,8 @@
  */
 package software.xdev.spring.data.eclipse.store.integration.isolated.tests.lazy;
 
+import java.util.Objects;
+
 import jakarta.persistence.GeneratedValue;
 import jakarta.persistence.GenerationType;
 import jakarta.persistence.Id;
@@ -52,4 +54,21 @@ public void setId(final long id)
 	{
 		this.id = id;
 	}
+	
+	@Override
+	public boolean equals(final Object o)
+	{
+		if(o == null || this.getClass() != o.getClass())
+		{
+			return false;
+		}
+		final SimpleEntityWithId that = (SimpleEntityWithId)o;
+		return this.id == that.id && Objects.equals(this.name, that.name);
+	}
+	
+	@Override
+	public int hashCode()
+	{
+		return Objects.hash(this.id, this.name);
+	}
 }