HHH-17167 Add test for issue

Author: mbellade

hibernate-core/src/test/java/org/hibernate/orm/test/rowid/RowIdUpdateTest.java renamed to hibernate-core/src/test/java/org/hibernate/orm/test/rowid/RowIdUpdateAndDeleteTest.java

@@ -30,15 +30,21 @@
 /**
  * @author Marco Belladelli
  */
-@DomainModel( annotatedClasses = { RowIdUpdateTest.SimpleEntity.class, RowIdUpdateTest.ParentEntity.class } )
+@DomainModel( annotatedClasses = {
+		RowIdUpdateAndDeleteTest.SimpleEntity.class,
+		RowIdUpdateAndDeleteTest.ParentEntity.class
+} )
 @SessionFactory( useCollectingStatementInspector = true )
 @Jira( "https://hibernate.atlassian.net/browse/HHH-17045" )
-public class RowIdUpdateTest {
+@Jira( "https://hibernate.atlassian.net/browse/HHH-17167" )
+public class RowIdUpdateAndDeleteTest {
 	@BeforeAll
 	public void setUp(SessionFactoryScope scope) {
 		scope.inTransaction( session -> {
 			session.persist( new SimpleEntity( 1L, "initial_status" ) );
 			session.persist( new ParentEntity( 2L, new SimpleEntity( 2L, "initial_status" ) ) );
+			session.persist( new SimpleEntity( 11L, "to_delete" ) );
+			session.persist( new ParentEntity( 12L, new SimpleEntity( 12L, "to_delete" ) ) );
 		} );
 	}
 
@@ -68,6 +74,22 @@ public void testSimpleUpdateSameTransaction(SessionFactoryScope scope) {
 		).isEqualTo( "new_status" ) );
 	}
 
+	@Test
+	public void testSimpleDeleteSameTransaction(SessionFactoryScope scope) {
+		final SQLStatementInspector inspector = scope.getCollectingStatementInspector();
+		inspector.clear();
+		scope.inTransaction( session -> {
+			final SimpleEntity simpleEntity = new SimpleEntity( 13L, "to_delete" );
+			session.persist( simpleEntity );
+			session.flush();
+			session.remove( simpleEntity );
+			inspector.clear();
+		} );
+		// the update should have used the primary key, as the row-id value is not available
+		checkUpdateQuery( inspector, true );
+		scope.inTransaction( session -> assertThat( session.find( SimpleEntity.class, 13L ) ).isNull() );
+	}
+
 	@Test
 	public void testRelatedUpdateSameTransaction(SessionFactoryScope scope) {
 		final SQLStatementInspector inspector = scope.getCollectingStatementInspector();
@@ -107,6 +129,19 @@ public void testSimpleUpdateDifferentTransaction(SessionFactoryScope scope) {
 		).isEqualTo( "new_status" ) );
 	}
 
+	@Test
+	public void testSimpleDeleteDifferentTransaction(SessionFactoryScope scope) {
+		final SQLStatementInspector inspector = scope.getCollectingStatementInspector();
+		scope.inTransaction( session -> {
+			final SimpleEntity simpleEntity = session.find( SimpleEntity.class, 11L );
+			session.remove( simpleEntity );
+			inspector.clear();
+		} );
+		final Dialect dialect = scope.getSessionFactory().getJdbcServices().getDialect();
+		checkUpdateQuery( inspector, dialect.rowId( "" ) == null );
+		scope.inTransaction( session -> assertThat( session.find( SimpleEntity.class, 11L ) ).isNull() );
+	}
+
 	@Test
 	public void testRelatedUpdateRelatedDifferentTransaction(SessionFactoryScope scope) {
 		final SQLStatementInspector inspector = scope.getCollectingStatementInspector();