From 67459cd09757deb7caf67f20ac0883457b16d772 Mon Sep 17 00:00:00 2001
From: Chris Cranford <chris@hibernate.org>
Date: Mon, 22 Oct 2018 18:18:55 -0400
Subject: [PATCH] HHH-13044 - Added test case.

---
 .../ManyToOneMapsIdFlushModeTest.java         | 137 ++++++++++++++++++
 1 file changed, 137 insertions(+)
 create mode 100644 hibernate-core/src/test/java/org/hibernate/test/annotations/manytoone/ManyToOneMapsIdFlushModeTest.java

diff --git a/hibernate-core/src/test/java/org/hibernate/test/annotations/manytoone/ManyToOneMapsIdFlushModeTest.java b/hibernate-core/src/test/java/org/hibernate/test/annotations/manytoone/ManyToOneMapsIdFlushModeTest.java
new file mode 100644
index 0000000000..9d3d537e3b
--- /dev/null
+++ b/hibernate-core/src/test/java/org/hibernate/test/annotations/manytoone/ManyToOneMapsIdFlushModeTest.java
@@ -0,0 +1,137 @@
+/*
+ * 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.test.annotations.manytoone;
+
+import java.util.HashSet;
+import java.util.Set;
+
+import javax.persistence.Entity;
+import javax.persistence.FlushModeType;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+import javax.persistence.ManyToOne;
+import javax.persistence.MapsId;
+import javax.persistence.OneToMany;
+
+import org.hibernate.jpa.test.BaseEntityManagerFunctionalTestCase;
+import org.junit.Test;
+
+import org.hibernate.testing.DialectChecks;
+import org.hibernate.testing.RequiresDialectFeature;
+import org.hibernate.testing.TestForIssue;
+import org.hibernate.testing.transaction.TransactionUtil;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author Chris Cranford
+ */
+@TestForIssue(jiraKey = "HHH-13044")
+@RequiresDialectFeature(DialectChecks.SupportsIdentityColumns.class)
+public class ManyToOneMapsIdFlushModeTest extends BaseEntityManagerFunctionalTestCase {
+	@Override
+	protected Class<?>[] getAnnotatedClasses() {
+		return new Class<?>[] { ParentEntity.class, ChildEntity.class };
+	}
+
+	@Test
+	public void testFlushModeCommitWithMapsIdAndIdentity() {
+		final ParentEntity parent = TransactionUtil.doInJPA( this::entityManagerFactory, entityManager -> {
+			entityManager.setFlushMode( FlushModeType.COMMIT );
+
+			final ParentEntity parentEntity = new ParentEntity();
+			parentEntity.setData( "test" );
+
+			final ChildEntity childEntity = new ChildEntity();
+			parentEntity.addChild( childEntity );
+
+			entityManager.persist( parentEntity );
+			entityManager.persist( childEntity );
+
+			return parentEntity;
+		} );
+
+		TransactionUtil.doInJPA( this::entityManagerFactory, entityManager -> {
+			final ParentEntity parentEntity = entityManager.find( ParentEntity.class, parent.getId() );
+			assertNotNull( parentEntity );
+			assertNotNull( parentEntity.getChildren() );
+			assertTrue( !parentEntity.getChildren().isEmpty() );
+
+			final ChildEntity childEntity = parentEntity.getChildren().iterator().next();
+			assertNotNull( childEntity );
+			assertEquals( parentEntity.getId(), childEntity.getId() );
+		} );
+	}
+
+	@Entity(name = "ParentEntity")
+	public static class ParentEntity {
+		@Id
+		@GeneratedValue(strategy = GenerationType.IDENTITY)
+		private Long id;
+		private String data;
+		@OneToMany(mappedBy = "parent")
+		private Set<ChildEntity> children = new HashSet<>();
+
+		public Long getId() {
+			return id;
+		}
+
+		public void setId(Long id) {
+			this.id = id;
+		}
+
+		public String getData() {
+			return data;
+		}
+
+		public void setData(String data) {
+			this.data = data;
+		}
+
+		public Set<ChildEntity> getChildren() {
+			return children;
+		}
+
+		public void setChildren(Set<ChildEntity> children) {
+			this.children = children;
+		}
+
+		public void addChild(ChildEntity child) {
+			getChildren().add( child );
+			child.setParent( this );
+		}
+	}
+
+	@Entity(name = "ChildEntity")
+	public static class ChildEntity {
+		@Id
+		private Long id;
+
+		@MapsId
+		@ManyToOne(optional = false, targetEntity = ParentEntity.class)
+		private ParentEntity parent;
+
+		public Long getId() {
+			return id;
+		}
+
+		public void setId(Long id) {
+			this.id = id;
+		}
+
+		public ParentEntity getParent() {
+			return parent;
+		}
+
+		public void setParent(ParentEntity parent) {
+			this.parent = parent;
+		}
+	}
+}