HHH-19523 test for collection listeners in stateless session

Author: gavinking

hibernate-core/src/test/java/org/hibernate/orm/test/stateless/events/CollectionListenerInStatelessSessionTest.java

@@ -0,0 +1,107 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.stateless.events;
+
+import org.hibernate.engine.spi.SessionFactoryImplementor;
+import org.hibernate.event.service.spi.EventListenerRegistry;
+import org.hibernate.event.spi.EventType;
+import org.hibernate.event.spi.PostCollectionRecreateEvent;
+import org.hibernate.event.spi.PostCollectionRecreateEventListener;
+import org.hibernate.event.spi.PostCollectionRemoveEvent;
+import org.hibernate.event.spi.PostCollectionRemoveEventListener;
+import org.hibernate.event.spi.PreCollectionRecreateEvent;
+import org.hibernate.event.spi.PreCollectionRecreateEventListener;
+import org.hibernate.event.spi.PreCollectionRemoveEvent;
+import org.hibernate.event.spi.PreCollectionRemoveEventListener;
+import org.hibernate.testing.orm.junit.DomainModel;
+import org.hibernate.testing.orm.junit.SessionFactory;
+import org.hibernate.testing.orm.junit.SessionFactoryScope;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+@DomainModel(annotatedClasses = {EntityA.class, EntityB.class})
+@SessionFactory
+class CollectionListenerInStatelessSessionTest {
+
+	@Test
+	void statelessInsert(SessionFactoryScope scope) {
+		EventListenerRegistry registry =
+				scope.getSessionFactory().unwrap(SessionFactoryImplementor.class)
+						.getServiceRegistry().getService(EventListenerRegistry.class);
+		var preRecreate = new MyPreCollectionRecreateEventListener();
+		var preRemove = new MyPreCollectionRemoveEventListener();
+		var postRecreate = new MyPostCollectionRecreateEventListener();
+		var postRemove = new MyPostCollectionRemoveEventListener();
+		registry.getEventListenerGroup(EventType.PRE_COLLECTION_RECREATE)
+				.appendListener( preRecreate );
+		registry.getEventListenerGroup(EventType.PRE_COLLECTION_REMOVE)
+				.appendListener( preRemove );
+		registry.getEventListenerGroup(EventType.POST_COLLECTION_RECREATE)
+				.appendListener( postRecreate );
+		registry.getEventListenerGroup(EventType.POST_COLLECTION_REMOVE)
+				.appendListener( postRemove );
+
+		scope.inStatelessTransaction(statelessSession -> {
+			EntityA a = new EntityA();
+			EntityB b = new EntityB();
+			a.children.add(b);
+			statelessSession.insert( b );
+			statelessSession.insert( a );
+			statelessSession.delete( a );
+			statelessSession.delete( b );
+		});
+
+		assertEquals(1, preRecreate.called);
+		assertEquals(1, preRemove.called);
+		assertEquals(1, postRecreate.called);
+		assertEquals(1, postRemove.called);
+	}
+
+}
+
+class MyPreCollectionRecreateEventListener implements PreCollectionRecreateEventListener {
+
+	int called = 0;
+
+	@Override
+	public void onPreRecreateCollection(PreCollectionRecreateEvent event) {
+		called++;
+	}
+
+}
+
+class MyPreCollectionRemoveEventListener implements PreCollectionRemoveEventListener {
+
+	int called = 0;
+
+	@Override
+	public void onPreRemoveCollection(PreCollectionRemoveEvent event) {
+		called++;
+	}
+
+}
+
+class MyPostCollectionRecreateEventListener implements PostCollectionRecreateEventListener {
+
+	int called = 0;
+
+	@Override
+	public void onPostRecreateCollection(PostCollectionRecreateEvent event) {
+		called++;
+	}
+
+}
+
+class MyPostCollectionRemoveEventListener implements PostCollectionRemoveEventListener {
+
+	int called = 0;
+
+	@Override
+	public void onPostRemoveCollection(PostCollectionRemoveEvent event) {
+		called++;
+	}
+
+}

hibernate-core/src/test/java/org/hibernate/orm/test/stateless/events/EntityA.java

@@ -0,0 +1,32 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.stateless.events;
+
+import java.util.ArrayList;
+import java.util.Collection;
+
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.GenerationType;
+import jakarta.persistence.Id;
+import jakarta.persistence.JoinColumn;
+import jakarta.persistence.OneToMany;
+import jakarta.persistence.Table;
+
+@Entity
+@Table(name = "ENTITY_A")
+public class EntityA {
+
+	@Id
+	@GeneratedValue(strategy = GenerationType.AUTO)
+	@Column(name = "ID")
+	Integer id;
+
+	@OneToMany
+	@JoinColumn(name = "ENTITY_A")
+	Collection<EntityB> children = new ArrayList<>();
+
+}

hibernate-core/src/test/java/org/hibernate/orm/test/stateless/events/EntityB.java

@@ -0,0 +1,21 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.stateless.events;
+
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.GenerationType;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+
+@Entity
+@Table(name = "ENTITY_B")
+public class EntityB {
+	@Id
+	@GeneratedValue(strategy = GenerationType.AUTO)
+	@Column(name = "ID")
+	Integer id;
+}