MultipleInitiatorTest

MultipleInitiatorTest.java

package org.hibernate.service;

import java.util.Map;

import org.hibernate.SessionFactory;
import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
import org.hibernate.cache.internal.RegionFactoryInitiator;
import org.hibernate.cache.spi.RegionFactory;
import org.hibernate.cfg.Configuration;
import org.hibernate.service.spi.ServiceContributor;
import org.hibernate.service.spi.ServiceRegistryImplementor;

import org.hibernate.testing.cache.CachingRegionFactory;
import org.hibernate.testing.junit4.BaseUnitTestCase;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

public class MultipleInitiatorTest extends BaseUnitTestCase {

	@Test
	public void testTwoSessionFactories() {
		SessionFactoryImplementor sf1 = null, sf2 = null;
		try {
			// NOTE : Assumes that the initiator is discovered through a ServiceContributor...
			// NOTE : Also assumes that the fallback RegionFactory is triggered
			
			assertEquals( 0, initiator.contributeCalledCount );
			assertEquals( 0, initiator.getFallbackCalledCount );

			sf1 = new Configuration().buildSessionFactory();
			assertEquals( 1, initiator.contributeCalledCount );
			assertEquals( 1, initiator.getFallbackCalledCount );

			sf2 = new Configuration().buildSessionFactory();
			assertEquals( 2, initiator.contributeCalledCount );
			assertEquals( 2, initiator.getFallbackCalledCount );
			
			// addionally:
			final RegionFactory rf1 = ( (SessionFactoryImplementor) sf1 ).getServiceRegistry();
			final RegionFactory rf2 = ( (SessionFactoryImplementor) sf2 ).getServiceRegistry();
			
			// we should have hit the fallback (apparently)
			assert rf1 instanceof MyRegionFactory;
			assert rf2 instanceof MyRegionFactory;
			
			// we should have 2 distinct RegionFactory instances
			assert rf1 != rf2;
		}
		finally {
			if ( sf1 != null ) {
				sf1.close();
			}
			if ( sf2 != null ) {
				sf2.close();
			}
		}
	}

	class MyRegionFactoryInitiator extends RegionFactoryInitiator implements ServiceContributor, Service {
		private static int getFallbackCalledCount = 0;
		private static int contributeCalledCount = 0;

		@Override
		protected RegionFactory getFallback(
				Map configurationValues,
				ServiceRegistryImplementor registry) {
			getFallbackCalledCount++;
			return new MyRegionFactory();
		}

		@Override
		public void contribute(StandardServiceRegistryBuilder serviceRegistryBuilder) {
			contributeCalledCount++;
		}
	}

	class MyRegionFactory extends CachingRegionFactory {
	}

}