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 index e0600ca3d4..e2b6673b15 100644 --- 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 ) {