HHH-19846 Drop JUnit 4 usage

Author: dreab8

hibernate-core/src/test/java/org/hibernate/orm/test/bulkid/AbstractMutationStrategyCompositeIdTest.java

@@ -4,157 +4,133 @@
  */
 package org.hibernate.orm.test.bulkid;
 
-import java.io.Serializable;
-import java.util.Objects;
 import jakarta.persistence.Entity;
 import jakarta.persistence.Id;
 import jakarta.persistence.Inheritance;
 import jakarta.persistence.InheritanceType;
+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.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
 
-import org.hibernate.cfg.AvailableSettings;
-import org.hibernate.cfg.Configuration;
-import org.hibernate.query.sqm.mutation.spi.SqmMultiTableInsertStrategy;
-import org.hibernate.query.sqm.mutation.spi.SqmMultiTableMutationStrategy;
+import java.io.Serializable;
+import java.util.Objects;
 
-import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
-import org.junit.Before;
-import org.junit.Test;
+import static org.assertj.core.api.Assertions.assertThat;
 
-import static org.hibernate.testing.transaction.TransactionUtil.doInHibernate;
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertTrue;
 
 /**
  * @author Vlad Mihalcea
  */
-public abstract class AbstractMutationStrategyCompositeIdTest extends BaseCoreFunctionalTestCase {
-
-	@Override
-	protected Class<?>[] getAnnotatedClasses() {
-		return new Class<?>[] {
-				Person.class,
-				Doctor.class,
-				Engineer.class
-		};
-	}
-
-	@Override
-	protected void configure(Configuration configuration) {
-		super.configure( configuration );
-		Class<? extends SqmMultiTableMutationStrategy> mutationStrategyClass = getMultiTableMutationStrategyClass();
-		if ( mutationStrategyClass != null ) {
-			configuration.setProperty( AvailableSettings.QUERY_MULTI_TABLE_MUTATION_STRATEGY, mutationStrategyClass );
+@DomainModel(
+		annotatedClasses = {
+				AbstractMutationStrategyCompositeIdTest.Person.class,
+				AbstractMutationStrategyCompositeIdTest.Doctor.class,
+				AbstractMutationStrategyCompositeIdTest.Engineer.class
 		}
-		Class<? extends SqmMultiTableInsertStrategy> insertStrategyClass = getMultiTableInsertStrategyClass();
-		if ( insertStrategyClass != null ) {
-			configuration.setProperty( AvailableSettings.QUERY_MULTI_TABLE_INSERT_STRATEGY, insertStrategyClass );
-		}
-	}
-
-	protected abstract Class<? extends SqmMultiTableMutationStrategy> getMultiTableMutationStrategyClass();
-
-	protected abstract Class<? extends SqmMultiTableInsertStrategy> getMultiTableInsertStrategyClass();
-
-	@Override
-	protected boolean isCleanupTestDataRequired() {
-		return true;
-	}
-
-	@Override
-	protected boolean isCleanupTestDataUsingBulkDelete() {
-		return true;
-	}
+)
+@SessionFactory
+public abstract class AbstractMutationStrategyCompositeIdTest {
 
-	@Before
-	public void setUp() {
-		doInHibernate( this::sessionFactory, session -> {
+	@BeforeEach
+	public void setUp(SessionFactoryScope scope) {
+		scope.inTransaction( session -> {
 			for ( int i = 0; i < entityCount(); i++ ) {
 				Doctor doctor = new Doctor();
 				doctor.setId( i + 1 );
 				doctor.setCompanyName( "Red Hat USA" );
-				doctor.setEmployed( ( i % 2 ) == 0 );
+				doctor.setEmployed( (i % 2) == 0 );
 				session.persist( doctor );
 			}
 
 			for ( int i = 0; i < entityCount(); i++ ) {
 				Engineer engineer = new Engineer();
 				engineer.setId( i + 1 + entityCount() );
 				engineer.setCompanyName( "Red Hat Europe" );
-				engineer.setEmployed( ( i % 2 ) == 0 );
-				engineer.setFellow( ( i % 2 ) == 1 );
+				engineer.setEmployed( (i % 2) == 0 );
+				engineer.setFellow( (i % 2) == 1 );
 				session.persist( engineer );
 			}
-		});
+		} );
+	}
+
+	@AfterEach
+	public void tearDown(SessionFactoryScope scope) {
+		scope.getSessionFactory().getSchemaManager().truncateMappedObjects();
 	}
 
 	protected int entityCount() {
 		return 4;
 	}
 
 	@Test
-	public void testUpdate() {
-		doInHibernate( this::sessionFactory, session -> {
-			int updateCount = session.createQuery( "update Person set name = :name where employed = :employed" )
+	public void testUpdate(SessionFactoryScope scope) {
+		scope.inTransaction( session -> {
+			int updateCount = session.createMutationQuery( "update Person set name = :name where employed = :employed" )
 					.setParameter( "name", "John Doe" )
 					.setParameter( "employed", true )
 					.executeUpdate();
 
-			assertEquals(entityCount(), updateCount);
-		});
+			assertThat( updateCount ).isEqualTo( entityCount() );
+		} );
 	}
 
 	@Test
-	public void testDeleteFromPerson() {
-		doInHibernate( this::sessionFactory, session -> {
+	public void testDeleteFromPerson(SessionFactoryScope scope) {
+		scope.inTransaction( session -> {
 			//tag::batch-bulk-hql-temp-table-delete-query-example[]
-			int updateCount = session.createQuery(
-				"delete from Person where employed = :employed" )
-			.setParameter( "employed", false )
-			.executeUpdate();
+			int updateCount = session.createMutationQuery(
+							"delete from Person where employed = :employed" )
+					.setParameter( "employed", false )
+					.executeUpdate();
 			//end::batch-bulk-hql-temp-table-delete-query-example[]
-			assertEquals( entityCount(), updateCount );
-		});
+			assertThat( updateCount ).isEqualTo( entityCount() );
+		} );
 	}
 
 	@Test
-	public void testDeleteFromEngineer() {
-		doInHibernate( this::sessionFactory, session -> {
-			int updateCount = session.createQuery( "delete from Engineer where fellow = :fellow" )
+	public void testDeleteFromEngineer(SessionFactoryScope scope) {
+		scope.inTransaction( session -> {
+			int updateCount = session.createMutationQuery( "delete from Engineer where fellow = :fellow" )
 					.setParameter( "fellow", true )
 					.executeUpdate();
-			assertEquals( entityCount() / 2, updateCount );
-		});
+			assertThat( updateCount ).isEqualTo( entityCount() / 2 );
+
+		} );
 	}
 
 	@Test
-	public void testInsert() {
-		doInHibernate( this::sessionFactory, session -> {
-			session.createQuery( "insert into Engineer(id, companyName, name, employed, fellow) values (0, 'Red Hat', :name, :employed, false)" )
+	public void testInsert(SessionFactoryScope scope) {
+		scope.inTransaction( session -> {
+			session.createMutationQuery(
+							"insert into Engineer(id, companyName, name, employed, fellow) values (0, 'Red Hat', :name, :employed, false)" )
 					.setParameter( "name", "John Doe" )
 					.setParameter( "employed", true )
 					.executeUpdate();
 			final Engineer engineer = session.find( Engineer.class,
 					new AbstractMutationStrategyCompositeIdTest_.Person_.Id( 0, "Red Hat" ) );
-			assertEquals( "John Doe", engineer.getName() );
-			assertTrue( engineer.isEmployed() );
-			assertFalse( engineer.isFellow() );
-		});
+			assertThat( engineer.getName() ).isEqualTo( "John Doe" );
+			assertThat( engineer.isEmployed() ).isTrue();
+			assertThat( engineer.isFellow() ).isFalse();
+		} );
 	}
 
 	@Test
-	public void testInsertSelect() {
-		doInHibernate( this::sessionFactory, session -> {
-			final int insertCount = session.createQuery( "insert into Engineer(id, companyName, name, employed, fellow) "
-										+ "select d.id + " + (entityCount() * 2) + ", 'Red Hat', 'John Doe', true, false from Doctor d" )
+	public void testInsertSelect(SessionFactoryScope scope) {
+		scope.inTransaction( session -> {
+			final int insertCount = session.createMutationQuery(
+							"insert into Engineer(id, companyName, name, employed, fellow) "
+							+ "select d.id + " + (entityCount() * 2) + ", 'Red Hat', 'John Doe', true, false from Doctor d" )
 					.executeUpdate();
 			final Engineer engineer = session.find( Engineer.class,
 					new AbstractMutationStrategyCompositeIdTest_.Person_.Id( entityCount() * 2 + 1, "Red Hat" ) );
-			assertEquals( entityCount(), insertCount );
-			assertEquals( "John Doe", engineer.getName() );
-			assertTrue( engineer.isEmployed() );
-			assertFalse( engineer.isFellow() );
-		});
+			assertThat( insertCount ).isEqualTo( entityCount() );
+			assertThat( engineer.getName() ).isEqualTo( "John Doe" );
+			assertThat( engineer.isEmployed() ).isTrue();
+			assertThat( engineer.isFellow() ).isFalse();
+		} );
 	}
 
 	//tag::batch-bulk-hql-temp-table-base-class-example[]
@@ -174,7 +150,7 @@ public static class Person implements Serializable {
 
 		//Getters and setters are omitted for brevity
 
-	//end::batch-bulk-hql-temp-table-base-class-example[]
+		//end::batch-bulk-hql-temp-table-base-class-example[]
 
 		public Integer getId() {
 			return id;
@@ -213,19 +189,19 @@ public boolean equals(Object o) {
 			if ( this == o ) {
 				return true;
 			}
-			if ( !( o instanceof Person ) ) {
+			if ( !(o instanceof Person) ) {
 				return false;
 			}
 			Person person = (Person) o;
 			return Objects.equals( getId(), person.getId() ) &&
-					Objects.equals( getCompanyName(), person.getCompanyName() );
+				Objects.equals( getCompanyName(), person.getCompanyName() );
 		}
 
 		@Override
 		public int hashCode() {
 			return Objects.hash( getId(), getCompanyName() );
 		}
-	//tag::batch-bulk-hql-temp-table-base-class-example[]
+		//tag::batch-bulk-hql-temp-table-base-class-example[]
 	}
 	//end::batch-bulk-hql-temp-table-base-class-example[]
 
@@ -241,7 +217,7 @@ public static class Engineer extends Person {
 
 		//Getters and setters are omitted for brevity
 
-	//end::batch-bulk-hql-temp-table-sub-classes-example[]
+		//end::batch-bulk-hql-temp-table-sub-classes-example[]
 
 		public boolean isFellow() {
 			return fellow;
@@ -250,7 +226,7 @@ public boolean isFellow() {
 		public void setFellow(boolean fellow) {
 			this.fellow = fellow;
 		}
-	//tag::batch-bulk-hql-temp-table-sub-classes-example[]
+		//tag::batch-bulk-hql-temp-table-sub-classes-example[]
 	}
 	//end::batch-bulk-hql-temp-table-sub-classes-example[]
 }