From 5632b5484612706d6bc9d2a776a3c9a1a58a4648 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Yoann=20Rodi=C3=A8re?= <yoann@hibernate.org>
Date: Tue, 1 Mar 2022 14:21:11 +0100
Subject: [PATCH] Test access to public field for Hibernate entity lazy-loaded
 from polymorphic toOne association

---
 ...ProxyAndLazyLoadingAndInheritanceTest.java | 122 ++++++++++++++++++
 1 file changed, 122 insertions(+)
 create mode 100644 extensions/hibernate-orm/deployment/src/test/java/io/quarkus/hibernate/orm/publicfields/PublicFieldWithProxyAndLazyLoadingAndInheritanceTest.java

diff --git a/extensions/hibernate-orm/deployment/src/test/java/io/quarkus/hibernate/orm/publicfields/PublicFieldWithProxyAndLazyLoadingAndInheritanceTest.java b/extensions/hibernate-orm/deployment/src/test/java/io/quarkus/hibernate/orm/publicfields/PublicFieldWithProxyAndLazyLoadingAndInheritanceTest.java
new file mode 100644
index 0000000000000..b0a8ae6d00c0b
--- /dev/null
+++ b/extensions/hibernate-orm/deployment/src/test/java/io/quarkus/hibernate/orm/publicfields/PublicFieldWithProxyAndLazyLoadingAndInheritanceTest.java
@@ -0,0 +1,122 @@
+/*
+ * Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * License: GNU Lesser General Public License (LGPL), version 2.1 or later.
+ * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
+ */
+package io.quarkus.hibernate.orm.publicfields;
+
+import static io.quarkus.hibernate.orm.TransactionTestUtils.inTransaction;
+import static org.assertj.core.api.Assertions.assertThat;
+
+import javax.inject.Inject;
+import javax.persistence.Entity;
+import javax.persistence.EntityManager;
+import javax.persistence.FetchType;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+import javax.persistence.ManyToOne;
+import javax.transaction.UserTransaction;
+
+import org.hibernate.Hibernate;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.RegisterExtension;
+
+import io.quarkus.hibernate.orm.TransactionTestUtils;
+import io.quarkus.test.QuarkusUnitTest;
+
+public class PublicFieldWithProxyAndLazyLoadingAndInheritanceTest {
+
+    @RegisterExtension
+    static QuarkusUnitTest runner = new QuarkusUnitTest()
+            .withApplicationRoot((jar) -> jar
+                    .addClass(TransactionTestUtils.class)
+                    .addClasses(Containing.class, Contained.class, ContainedExtended.class)
+                    .addClass(FieldAccessEnhancedDelegate.class))
+            .withConfigurationResource("application.properties");
+
+    @Inject
+    EntityManager em;
+
+    @Inject
+    UserTransaction transaction;
+
+    private Long containingID;
+
+    @Test
+    public void test() {
+        FieldAccessEnhancedDelegate delegate = new FieldAccessEnhancedDelegate();
+        inTransaction(transaction, () -> {
+            containingID = delegate.createEntities(em, "George");
+        });
+        inTransaction(transaction, () -> {
+            delegate.testLazyLoading(em, containingID);
+        });
+    }
+
+    @Entity(name = "Containing")
+    public static class Containing {
+
+        @Id
+        @GeneratedValue(strategy = GenerationType.AUTO)
+        public Long id;
+
+        @ManyToOne(fetch = FetchType.LAZY, optional = false)
+        public Contained contained;
+    }
+
+    @Entity(name = "Contained")
+    public static class Contained {
+
+        @Id
+        @GeneratedValue(strategy = GenerationType.AUTO)
+        public Long id;
+
+        public String name;
+
+        Contained() {
+        }
+
+        Contained(String name) {
+            this.name = name;
+        }
+    }
+
+    @Entity(name = "ContainedExtended")
+    public static class ContainedExtended extends Contained {
+
+        ContainedExtended() {
+        }
+
+        ContainedExtended(String name) {
+            this.name = name;
+        }
+
+    }
+
+    /**
+     * A class whose bytecode is transformed by Quarkus to replace public field access with getter/setter access.
+     * <p>
+     * (Test bytecode was not transformed by Quarkus when using QuarkusUnitTest last time I checked).
+     */
+    private static class FieldAccessEnhancedDelegate {
+
+        public long createEntities(EntityManager entityManager, String name) {
+            Containing containing = new Containing();
+            ContainedExtended contained = new ContainedExtended(name);
+            containing.contained = contained;
+            entityManager.persist(contained);
+            entityManager.persist(containing);
+            return containing.id;
+        }
+
+        public void testLazyLoading(EntityManager entityManager, Long containingID) {
+            Containing containing = entityManager.find(Containing.class, containingID);
+            Contained contained = containing.contained;
+            assertThat(contained).isNotNull();
+            assertThat(Hibernate.isPropertyInitialized(contained, "name")).isFalse();
+            assertThat(contained.name).isNotNull();
+        }
+    }
+}