HHH-19134 Test case (Jakarta Data & JPA) - find entity by ID when multiple properties are annotated by @id annotation

Author: cigaly

tooling/metamodel-generator/src/jakartaData/java/org/hibernate/processor/test/data/idclass/CompositeIdClassTest.java

@@ -0,0 +1,35 @@
+/*
+ * SPDX-License-Identifier: LGPL-2.1-or-later
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.processor.test.data.idclass;
+
+import org.hibernate.processor.test.data.idclass.MyEntity.MyEntityId;
+import org.hibernate.processor.test.util.CompilationTest;
+import org.hibernate.processor.test.util.WithClasses;
+import org.junit.Test;
+
+import static org.hibernate.processor.test.util.TestUtil.assertMetamodelClassGeneratedFor;
+import static org.hibernate.processor.test.util.TestUtil.assertPresenceOfMethodInMetamodelFor;
+import static org.hibernate.processor.test.util.TestUtil.getMetaModelSourceAsString;
+
+public class CompositeIdClassTest extends CompilationTest {
+	@Test
+	@WithClasses({
+			MyRepository.class,
+			MyEntity.class,
+	})
+	public void test() {
+		System.out.println( getMetaModelSourceAsString( MyEntity.class ) );
+		System.out.println( getMetaModelSourceAsString( MyEntity.class, true ) );
+		System.out.println( getMetaModelSourceAsString( MyRepository.class ) );
+		assertMetamodelClassGeneratedFor( MyEntity.class );
+		assertMetamodelClassGeneratedFor( MyEntity.class, true );
+		assertMetamodelClassGeneratedFor( MyRepository.class );
+		assertPresenceOfMethodInMetamodelFor(
+				MyRepository.class,
+				"findById",
+				MyEntityId.class
+		);
+	}
+}

tooling/metamodel-generator/src/jakartaData/java/org/hibernate/processor/test/data/idclass/MyEntity.java

@@ -0,0 +1,26 @@
+/*
+ * SPDX-License-Identifier: LGPL-2.1-or-later
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.processor.test.data.idclass;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.IdClass;
+
+@Entity
+@IdClass(MyEntity.MyEntityId.class)
+public class MyEntity {
+
+	@Id
+	Integer topicId;
+
+	@Id
+	String userId;
+
+	String status;
+
+	public record MyEntityId(Integer topicId, String userId) {
+
+	}
+}

tooling/metamodel-generator/src/jakartaData/java/org/hibernate/processor/test/data/idclass/MyRepository.java

@@ -0,0 +1,16 @@
+/*
+ * SPDX-License-Identifier: LGPL-2.1-or-later
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.processor.test.data.idclass;
+
+import jakarta.data.repository.Query;
+import jakarta.data.repository.Repository;
+import org.hibernate.processor.test.data.idclass.MyEntity.MyEntityId;
+
+@Repository
+public interface MyRepository {
+
+	@Query("from MyEntity where id=:id")
+	MyEntity findById(MyEntityId id);
+}

tooling/metamodel-generator/src/test/java/org/hibernate/processor/test/idclass/CompositeIdClassTest.java

@@ -0,0 +1,36 @@
+/*
+ * SPDX-License-Identifier: LGPL-2.1-or-later
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.processor.test.idclass;
+
+import jakarta.persistence.EntityManager;
+import org.hibernate.processor.test.idclass.MyEntity.MyEntityId;
+import org.hibernate.processor.test.util.CompilationTest;
+import org.hibernate.processor.test.util.WithClasses;
+import org.junit.Test;
+
+import static org.hibernate.processor.test.util.TestUtil.assertMetamodelClassGeneratedFor;
+import static org.hibernate.processor.test.util.TestUtil.assertPresenceOfMethodInMetamodelFor;
+import static org.hibernate.processor.test.util.TestUtil.assertPresenceOfNameFieldInMetamodelFor;
+import static org.hibernate.processor.test.util.TestUtil.getMetaModelSourceAsString;
+
+public class CompositeIdClassTest extends CompilationTest {
+	@Test
+	@WithClasses(MyEntity.class)
+	public void test() {
+		System.out.println( getMetaModelSourceAsString( MyEntity.class ) );
+		assertMetamodelClassGeneratedFor( MyEntity.class );
+		assertPresenceOfNameFieldInMetamodelFor(
+				MyEntity.class,
+				"QUERY_FIND_BY_ID",
+				"Missing named query attribute."
+		);
+		assertPresenceOfMethodInMetamodelFor(
+				MyEntity.class,
+				"findById",
+				EntityManager.class,
+				MyEntityId.class
+		);
+	}
+}

tooling/metamodel-generator/src/test/java/org/hibernate/processor/test/idclass/MyEntity.java

@@ -0,0 +1,30 @@
+/*
+ * SPDX-License-Identifier: LGPL-2.1-or-later
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.processor.test.idclass;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.IdClass;
+import jakarta.persistence.NamedQuery;
+import org.hibernate.annotations.processing.CheckHQL;
+
+@Entity
+@IdClass(MyEntity.MyEntityId.class)
+@CheckHQL
+@NamedQuery(name = "#findById", query = "from MyEntity e where e.id=:id")
+public class MyEntity {
+
+	@Id
+	Integer topicId;
+
+	@Id
+	String userId;
+
+	String status;
+
+	public record MyEntityId(Integer topicId, String userId) {
+
+	}
+}