ModuleMemberRegistrationDeployment.java example

Explorer
hibernate-ogm-master
/*
 * Hibernate OGM, Domain model persistence for NoSQL datastores
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
 */
package org.hibernate.ogm.test.integration.testcase.util;

import org.hibernate.ogm.test.integration.testcase.ModuleMemberRegistrationScenario;
import org.hibernate.ogm.test.integration.testcase.controller.MemberRegistration;
import org.hibernate.ogm.test.integration.testcase.controller.MemberRegistrationWithJta;
import org.hibernate.ogm.test.integration.testcase.controller.MemberRegistrationWithResourceLocal;
import org.hibernate.ogm.test.integration.testcase.controller.RegistrationExecutor;
import org.hibernate.ogm.test.integration.testcase.model.Address;
import org.hibernate.ogm.test.integration.testcase.model.Member;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.asset.EmptyAsset;
import org.jboss.shrinkwrap.api.asset.StringAsset;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.jboss.shrinkwrap.descriptor.api.persistence20.PersistenceDescriptor;
import org.jboss.shrinkwrap.descriptor.api.persistence20.PersistenceUnitTransactionType;

/**
 * Creates a commeon archive that can be customized and deployed when running integration tests.
 * <p>
 * Different classes are going to be used if the persistence unit transaction type is set to RESOURCE_LOCAL instead of
 * JTA.
 *
 * @author Davide D'Alto
 */
public class ModuleMemberRegistrationDeployment {

	private ModuleMemberRegistrationDeployment() {
		super();
	}

	public static class Builder {
		private final WebArchive archive;
		private boolean resourceLocal = true;

		public Builder(Class<?> clazz) {
			archive = ShrinkWrap
					.create( WebArchive.class, clazz.getSimpleName() + ".war" )
					.addClasses( clazz, MemberRegistration.class, RegistrationExecutor.class, Member.class, Address.class, ModuleMemberRegistrationScenario.class )
					.addAsWebInfResource( EmptyAsset.INSTANCE, "beans.xml" );
		}

		public Builder persistenceXml(PersistenceDescriptor descriptor) {
			resourceLocal = descriptor.getOrCreatePersistenceUnit().getTransactionType() == PersistenceUnitTransactionType._RESOURCE_LOCAL;
			String persistenceXml = descriptor.exportAsString();
			persistenceXml = ModulesHelper.injectVariables( persistenceXml );
			archive.addAsResource( new StringAsset( persistenceXml ), "META-INF/persistence.xml" );
			return this;
		}

		public Builder manifestDependencies(String dependencies) {
			ModulesHelper.addModulesDependencyDeclaration( archive, dependencies );
			return this;
		}

		public Builder addClasses(Class<?> clazz, Class<?>...furtherClasses) {
			archive.addClass( clazz );
			if ( furtherClasses != null ) {
				archive.addClasses( furtherClasses );
			}
			return this;
		}

		public Builder addAsWebInfResource(String resourceName, String targetResourceName) {
			final String content = ModulesHelper.loadResourceInjectingVariables( resourceName );
			archive.addAsWebInfResource( new StringAsset( content ), targetResourceName );
			return this;
		}

		public WebArchive createDeployment() {
			if ( resourceLocal ) {
				archive.addClasses( MemberRegistrationWithResourceLocal.class );
			}
			else {
				archive.addClasses( MemberRegistrationWithJta.class, JtaResources.class );
			}
			return archive;
		}

	}

}