HHH-11570 - Add test for issue

hibernate-envers/src/test/java/org/hibernate/envers/test/integration/jta/OneToManyJtaSessionClosedBeforeCommitTest.java

@@ -0,0 +1,89 @@
+/*
+ * 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 org.hibernate.envers.test.integration.jta;
+
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
+import javax.persistence.EntityManager;
+
+import org.hibernate.cfg.AvailableSettings;
+import org.hibernate.envers.test.BaseEnversJPAFunctionalTestCase;
+import org.hibernate.envers.test.Priority;
+import org.hibernate.envers.test.entities.onetomany.SetRefEdEntity;
+import org.hibernate.envers.test.entities.onetomany.SetRefIngEntity;
+
+import org.hibernate.testing.TestForIssue;
+import org.hibernate.testing.jta.TestingJtaBootstrap;
+import org.hibernate.testing.jta.TestingJtaPlatformImpl;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * @author Andrea Boriero
+ */
+@TestForIssue( jiraKey = "HHH-11570")
+public class OneToManyJtaSessionClosedBeforeCommitTest extends BaseEnversJPAFunctionalTestCase {
+	private Integer entityId;
+
+	@Override
+	protected Class<?>[] getAnnotatedClasses() {
+		return new Class<?>[] {SetRefIngEntity.class, SetRefEdEntity.class};
+	}
+
+	@Override
+	protected void addConfigOptions(Map options) {
+		TestingJtaBootstrap.prepare( options );
+		options.put( AvailableSettings.ALLOW_JTA_TRANSACTION_ACCESS, "true" );
+	}
+
+	@Test
+	@Priority(10)
+	public void initData() throws Exception {
+		TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
+		EntityManager entityManager = getEntityManager();
+		try {
+			SetRefEdEntity edEntity = new SetRefEdEntity( 2, "edEntity" );
+			entityManager.persist( edEntity );
+
+			SetRefIngEntity ingEntity = new SetRefIngEntity( 1, "ingEntity" );
+
+			Set<SetRefIngEntity> sries = new HashSet<>();
+			sries.add( ingEntity );
+			ingEntity.setReference( edEntity );
+			edEntity.setReffering( sries );
+
+			entityManager.persist( ingEntity );
+
+			entityId = ingEntity.getId();
+			// simulates spring JtaTransactionManager.triggerBeforeCompletion()
+			// this closes the entity manager prior to the JTA transaction.
+			entityManager.close();
+		}
+		finally {
+			TestingJtaPlatformImpl.tryCommit();
+		}
+	}
+
+	@Test
+	public void testRevisionCounts() {
+		assertEquals(
+				Arrays.asList( 1 ),
+				getAuditReader().getRevisions( SetRefIngEntity.class, entityId )
+		);
+	}
+
+	@Test
+	public void testRevisionHistory() {
+		assertEquals(
+				new SetRefIngEntity( 1, "ingEntity" ),
+				getAuditReader().find( SetRefIngEntity.class, entityId, 1 )
+		);
+	}
+}