OGM-966 Add method to get the service registry in JpaTestCase

hibernate · Jan 22, 2016 · a2cda1a · a2cda1a
1 parent d6416ed
commit a2cda1a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/core/src/test/java/org/hibernate/ogm/utils/jpa/JpaTestCase.java b/core/src/test/java/org/hibernate/ogm/utils/jpa/JpaTestCase.java
@@ -27,8 +27,10 @@
 import org.hibernate.engine.transaction.jta.platform.spi.JtaPlatform;
 import org.hibernate.jpa.HibernateEntityManagerFactory;
 import org.hibernate.ogm.jpa.HibernateOgmPersistence;
+import org.hibernate.ogm.jpa.impl.OgmEntityManagerFactory;
 import org.hibernate.ogm.utils.SkippableTestRunner;
 import org.hibernate.ogm.utils.TestHelper;
+import org.hibernate.service.spi.ServiceRegistryImplementor;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.runner.RunWith;
@@ -104,6 +106,13 @@ private static TransactionManager extractJBossTransactionManager(EntityManagerFa
 		return sessionFactory.getServiceRegistry().getService( JtaPlatform.class ).retrieveTransactionManager();
 	}
 
+	protected ServiceRegistryImplementor getServiceRegistry() {
+		OgmEntityManagerFactory emFactory = ( (OgmEntityManagerFactory) getFactory() );
+		SessionFactoryImplementor sessionFactory = emFactory.getSessionFactory();
+		ServiceRegistryImplementor serviceRegistry = sessionFactory.getServiceRegistry();
+		return serviceRegistry;
+	}
+
 	@After
 	public void closeFactory() throws Exception {
 		if ( transactionManager != null && transactionManager.getStatus() == Status.STATUS_ACTIVE ) {