[#487] Add utility methods for cleaning up entities

Author: DavideD

hibernate-reactive-core/src/test/java/org/hibernate/reactive/BaseReactiveTest.java

@@ -36,10 +36,13 @@
 import org.junit.Rule;
 import org.junit.runner.RunWith;
 
+import java.util.Arrays;
 import java.util.concurrent.CompletionStage;
 import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
 import static org.hibernate.reactive.containers.DatabaseConfiguration.dbType;
+import static org.hibernate.reactive.util.impl.CompletionStages.loop;
 
 /**
  * Base class for unit tests that need a connection to the selected db and
@@ -127,6 +130,26 @@ protected Configuration constructConfiguration() {
 		return configuration;
 	}
 
+	public CompletionStage<Void> deleteEntities(Class<?>... entities) {
+		return deleteEntities( Arrays.stream( entities )
+								  .map( BaseReactiveTest::defaultEntityName )
+								  .collect( Collectors.toList() )
+								  .toArray( new String[entities.length] ) );
+	}
+
+	private static String defaultEntityName(Class<?> aClass) {
+		int index = aClass.getName().lastIndexOf( '.' );
+		index = index > -1 ? index + 1 : 0;
+		return aClass.getName().substring( index );
+	}
+
+	public CompletionStage<Void> deleteEntities(String... entities) {
+		return getSessionFactory()
+				.withTransaction( (s, tx) -> loop( entities, name -> s
+						.createQuery( "from " + name ).getResultList()
+						.thenCompose( list -> s.remove( list.toArray( new Object[list.size()] ) ) ) ) );
+	}
+
 	@Before
 	public void before(TestContext context) {
 		Configuration configuration = constructConfiguration();