HHH-18691 Add Testcase for Get Property from Embedded Record

Author: Aftermoon-dev

hibernate-envers/src/test/java/org/hibernate/envers/test/integration/data/RecordFieldEntityTest.java

@@ -0,0 +1,146 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.envers.test.integration.data;
+
+
+import jakarta.persistence.Embedded;
+import jakarta.persistence.Entity;
+import jakarta.persistence.EntityManager;
+import jakarta.persistence.Id;
+import org.hibernate.envers.AuditReader;
+import org.hibernate.envers.Audited;
+import org.hibernate.orm.test.envers.BaseEnversJPAFunctionalTestCase;
+import org.hibernate.orm.test.envers.Priority;
+import org.hibernate.testing.orm.junit.JiraKey;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+
+/**
+ * Tests that {@link jakarta.persistence.Embedded} works correctly when combined with Java record classes
+ * in the context of Envers auditing.
+ *
+ * @author Minjae Seon
+ */
+@JiraKey(value = "HHH-18691")
+public class RecordFieldEntityTest extends BaseEnversJPAFunctionalTestCase {
+	record TestRecord(String foo, String bar) {}
+
+	static class TestEmbeddedClass {
+		public TestEmbeddedClass() {}
+		public TestEmbeddedClass(String foo, String bar) {
+			this.foo = foo;
+			this.bar = bar;
+		}
+
+		private String foo;
+		private String bar;
+
+		public String getFoo() {
+			return foo;
+		}
+
+		public String getBar() {
+			return bar;
+		}
+
+		public void setFoo(String foo) {
+			this.foo = foo;
+		}
+	}
+
+	@Entity
+	@Audited
+	static class WithRecord {
+		@Id
+		private Integer id;
+
+		@Embedded
+		private TestRecord testRecord;
+
+		public Integer getId() {
+			return id;
+		}
+
+		static WithRecord of(int id, String foo, String bar) {
+			WithRecord withRecord = new WithRecord();
+
+			withRecord.id = id;
+			withRecord.testRecord = new TestRecord(foo, bar);
+			return withRecord;
+		}
+	}
+
+	@Entity
+	@Audited
+	static class WithoutRecord {
+		@Id
+		private Integer id;
+
+		@Embedded
+		private TestEmbeddedClass testEmbeddedClass;
+
+		public Integer getId() {
+			return id;
+		}
+
+		public TestEmbeddedClass getTestEmbeddedClass() {
+			return testEmbeddedClass;
+		}
+
+		static WithoutRecord of(int id, String foo, String bar) {
+			WithoutRecord withoutRecord = new WithoutRecord();
+
+			withoutRecord.id = id;
+			withoutRecord.testEmbeddedClass = new TestEmbeddedClass(foo, bar);
+
+			return withoutRecord;
+		}
+	}
+
+	@Override
+	protected Class<?>[] getAnnotatedClasses() {
+		return new Class<?>[]{ WithRecord.class, WithoutRecord.class };
+	}
+
+	@Test
+	@Priority(10)
+	public void initData() {
+		EntityManager em = getEntityManager();
+
+		em.getTransaction().begin();
+
+		// Create WithRecord Entity
+		WithRecord withRecord = WithRecord.of( 1, "foo", "bar" );
+		em.persist( withRecord );
+
+		// Create WithoutRecord
+		WithoutRecord withoutRecord = WithoutRecord.of( 1, "foo", "bar" );
+		em.persist( withoutRecord );
+
+		em.getTransaction().commit();
+
+		em.close();
+	}
+
+	@Test
+	public void testLoadRecordData() {
+		AuditReader auditReader = getAuditReader();
+		WithRecord recordRev = auditReader.find( WithRecord.class, 1, 1 );
+
+		assertEquals("WithRecord.TestRecord.foo equals foo", "foo", recordRev.testRecord.foo());
+		assertEquals("WithRecord.TestRecord.bar equals bar", "bar", recordRev.testRecord.bar());
+	}
+
+	@Test
+	public void testLoadWithoutRecordData() {
+		AuditReader auditReader = getAuditReader();
+		WithoutRecord withoutRecordRev = auditReader.find( WithoutRecord.class, 1, 1 );
+
+		assertEquals("WithoutRecord.TestEmbeddedClass.foo equals foo", "foo", withoutRecordRev.getTestEmbeddedClass().getFoo());
+		assertEquals("WithoutRecord.TestEmbeddedClass.bar equals bar", "bar", withoutRecordRev.getTestEmbeddedClass().getBar());
+	}
+}