_br-ode-1.X.svn/axis2-war/src/test/java/org/apache/ode/axis2/Axis2TestBase.java - ode - Git at Google

 package org.apache.ode.axis2;

 import org.apache.axis2.AxisFault;
 import org.apache.axis2.context.ConfigurationContextFactory;
 import org.apache.axis2.description.AxisOperation;
 import org.apache.axis2.description.AxisService;
 import org.apache.axis2.description.Parameter;
 import org.apache.axis2.description.WSDL11ToAxisServiceBuilder;
 import org.apache.axis2.engine.AxisServer;
 import org.apache.axis2.engine.MessageReceiver;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.apache.ode.axis2.hooks.ODEAxisService;
 import org.apache.ode.axis2.util.Axis2UriResolver;
 import org.apache.ode.axis2.util.Axis2WSDLLocator;
 import org.apache.ode.tools.sendsoap.cline.HttpSoapSender;
 import org.testng.annotations.AfterMethod;
 import org.testng.annotations.BeforeMethod;
 import org.testng.annotations.DataProvider;


 import javax.servlet.ServletException;
 import javax.wsdl.WSDLException;
 import javax.xml.namespace.QName;
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileNotFoundException;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Iterator;
 import java.util.List;

 /**
  * @author Matthieu Riou <mriou@apache.org>
  */
 public abstract class Axis2TestBase {

     public static final int DEFAULT_TEST_PORT = 8888;

     private static final Log log = LogFactory.getLog(Axis2TestBase.class);

     protected ODEAxis2Server server;

 	protected String config;

 	protected static final String DO_NOT_OVERRIDE_CONFIG = "<DO_NOT_OVERRIDE_CONFIG>";

 	private static String originalOdePersistence = System.getProperty("ode.persistence");
 	private static String originalOdeConfigDir = System.getProperty("org.apache.ode.configDir");

 	@DataProvider(name = "configs")
 	protected Iterator<Object[]> createConfigData() {
 		List<String> configDirList = new ArrayList<String>();
 		if( !(this instanceof ODEConfigDirAware) || ((ODEConfigDirAware)this).getODEConfigDir().contains("hib")) {
 			addToConfigDirList(configDirList, "org.apache.ode.hibdbs");
 		}
 		if( !(this instanceof ODEConfigDirAware) || !((ODEConfigDirAware)this).getODEConfigDir().contains("hib")) {
 			addToConfigDirList(configDirList, "org.apache.ode.jpadbs");
 		}

 		if( configDirList.isEmpty() ) {
 			// if no system property is set, fall back to default
 			if( this instanceof ODEConfigDirAware ) {
 				configDirList.add(((ODEConfigDirAware)this).getODEConfigDir());
 			} else {
 				configDirList.add(DO_NOT_OVERRIDE_CONFIG);
 			}
 		} else {
 			System.out.println("Java system properties have been set to override ode configuration: " + configDirList);
 		}

 		final Iterator<String> itr = configDirList.iterator();
 		return new Iterator<Object[]>() {
 			public boolean hasNext() {
 				return itr.hasNext();
 			}

 			public Object[] next() {
 				config = itr.next();
 				return new Object[] {};
 			}

 			public void remove() {
 			}
 		};
 	}

 	private void addToConfigDirList(List<String> configDirList, String propertyKey) {
 		String dbs = System.getProperty(propertyKey);
 		if( dbs != null ) {
 			String[] configDirs = dbs.split(",");
 			for( String configDir : configDirs ) {
 				String trimmed = configDir.trim();
 				if( trimmed.length() > 0 ) {
 					configDirList.add(trimmed);
 				}
 			}
 		}
 	}

 	public void startServer() throws Exception {
         startServer("webapp/WEB-INF", "webapp/WEB-INF/conf/axis2.xml");
     }

     public void startServer(String axis2RepoDir, String axis2ConfLocation) throws Exception {
         String odeRootAbsolutePath = getClass().getClassLoader().getResource("webapp/WEB-INF").getFile();
         String axis2RepoAbsolutePath = getClass().getClassLoader().getResource(axis2RepoDir).getFile();
         String axis2ConfAbsolutePath = axis2ConfLocation == null ? null : getClass().getClassLoader().getResource(axis2ConfLocation).getFile();
         server = new ODEAxis2Server(odeRootAbsolutePath, axis2RepoAbsolutePath, axis2ConfAbsolutePath);
         server.start();
     }

     public void stopServer() throws AxisFault {
         server.stop();
     }

     @BeforeMethod
     protected void setUp() throws Exception {
     	/**
     	 * 1. If no settings are given from buildr, the test runs with the default config directory.
     	 * 2. If no settings are given from buildr and if the test implements ODEConfigDirAware, the test runs with
     	 * the config directory from the interface.
     	 * 3. If settings are given from buildr and if it's derby and openJPA, test falls back to the above 1 or 2.
     	 * 4. If settings are given from buildr and if it's derby and hibernate, test falls back to the above 2 or
     	 * uses -Dode.persistence=hibernate.
     	 */
     	if( config == null || DO_NOT_OVERRIDE_CONFIG.equals(config) ) {
         	System.out.println("Test config: default.");
     	} else if("<jpa>".equals(config)) {
 			if( this instanceof ODEConfigDirAware ) {
 				config = ((ODEConfigDirAware)this).getODEConfigDir();
 	    		System.out.println("Test config: " + config + ".");
 	    		System.setProperty("org.apache.ode.configDir", config);
 			} else {
 				System.out.println("Test config: default.");
 			}
     	} else if("<hib>".equals(config)) {
 			if( this instanceof ODEConfigDirAware ) {
 				config = ((ODEConfigDirAware)this).getODEConfigDir();
 	    		System.out.println("Test config: " + config + ".");
 	    		System.setProperty("org.apache.ode.configDir", config);
 			} else {
 				// why does this not work?
 //	    		System.out.println("Test config: -Dode.persistence=hibernate");
 //	    		System.setProperty("ode.persistence", "hibernate");
 	    		config = getClass().getClassLoader().getResource("webapp").getFile() + "/WEB-INF/conf.hib-derby";
 	        	System.out.println("Test config: " + config + ".");
 	    		System.setProperty("org.apache.ode.configDir", config);
 			}
     	} else {
         	System.out.println("Test config: " + config + ".");
     		System.setProperty("org.apache.ode.configDir", config);
     	}

     	startServer();
     }

     @AfterMethod
     protected void tearDown() throws Exception {
         stopServer();

         if( originalOdeConfigDir != null ) {
         	System.setProperty("org.apache.ode.configDir", originalOdeConfigDir);
         } else {
         	System.clearProperty("org.apache.ode.configDir");
         }
         if( originalOdeConfigDir != null ) {
     		System.setProperty("ode.persistence", originalOdePersistence);
         } else {
         	System.clearProperty("ode.persistence");
         }
     }

     protected class ODEAxis2Server extends AxisServer {

         ODEServer _ode;
         String odeRootDir;

         protected ODEAxis2Server(String odeRootDir, String axis2RepoDir, String axis2ConfLocation) throws Exception {
             super(false);
             this.odeRootDir = odeRootDir;
             if (log.isInfoEnabled()) {
                 log.info("Ode Root Dir: " + odeRootDir);
                 log.info("Axis2 Conf file: " + axis2ConfLocation);
                 log.info("Axis2 Repo dir: " + axis2RepoDir);
             }

             configContext = ConfigurationContextFactory.createConfigurationContextFromFileSystem(axis2RepoDir, axis2ConfLocation);
             // do not use 8080 for tests
             configContext.getAxisConfiguration().getTransportIn("http").addParameter(new Parameter("port", ""+DEFAULT_TEST_PORT));
         }

         protected void start() throws AxisFault {
             super.start();
             _ode = new ODEServer();
             try {
                 _ode.init(odeRootDir, configContext.getAxisConfiguration());
             } catch (ServletException e) {
                 e.printStackTrace();
             }
         }

         public void stop() throws AxisFault {
             _ode.shutDown();
             super.stop();
         }

         public Collection<QName> deployProcess(String bundleName) {
             return _ode.getProcessStore().deploy(new File(getBundleDir(bundleName)));
         }

         public void undeployProcess(String bundleName) {
             _ode.getProcessStore().undeploy(new File(getBundleDir(bundleName)));
         }

         public boolean isDeployed(String bundleName) {
             return _ode.getProcessStore().getPackages().contains(bundleName);
         }

         /**
          * Creates and deploys an Axis service based on a provided MessageReceiver. The receiver
          * will be invoked for all invocations of that service.
          */
         protected void deployService(String bundleName, String defFile, QName serviceName, String port,
                                      MessageReceiver receiver) throws WSDLException, IOException, URISyntaxException {
             URI wsdlUri = new File(getBundleDir(bundleName) + "/" + defFile).toURI();

             InputStream is = wsdlUri.toURL().openStream();
             WSDL11ToAxisServiceBuilder serviceBuilder = new ODEAxisService.WSDL11ToAxisPatchedBuilder(is, serviceName, port);
             serviceBuilder.setBaseUri(wsdlUri.toString());
             serviceBuilder.setCustomResolver(new Axis2UriResolver());
             serviceBuilder.setCustomWSLD4JResolver(new Axis2WSDLLocator(wsdlUri));
             serviceBuilder.setServerSide(true);

             AxisService axisService = serviceBuilder.populateService();
             axisService.setName(serviceName.getLocalPart());
             axisService.setWsdlFound(true);
             axisService.setCustomWsdl(true);
             axisService.setClassLoader(getConfigurationContext().getAxisConfiguration().getServiceClassLoader());

             Iterator operations = axisService.getOperations();
             while (operations.hasNext()) {
                 AxisOperation operation = (AxisOperation) operations.next();
                 if (operation.getMessageReceiver() == null) {
                     operation.setMessageReceiver(receiver);
                 }
             }
             getConfigurationContext().getAxisConfiguration().addService(axisService);
         }

         public String sendRequestFile(String endpoint, String bundleName, String filename) {
             try {
                 return HttpSoapSender.doSend(new URL(endpoint),
                         new FileInputStream(getBundleDir(bundleName) + "/" + filename), null, 0, null, null, null);
             } catch (IOException e) {
                 throw new RuntimeException(e);
             }
         }

         protected String getBundleDir(String bundleName) {
             return getClass().getClassLoader().getResource(bundleName).getFile();
         }

         /**
          * Convenient methods to generate a WSDL for an Axis2 service. Often nice, but also often
          * generates crappy WSDL that aren't even valid (especially when faults are involved) so
          * use with care.
          *
          * @param serviceName
          * @param fileName
          * @throws AxisFault
          */
         protected void generateWSDL(String serviceName, String fileName) throws AxisFault {
             FileOutputStream fos = null;
             try {
                 fos = new FileOutputStream(fileName);
             } catch (FileNotFoundException e) {
                 e.printStackTrace();
             }
             configContext.getAxisConfiguration().getService(serviceName).printWSDL(fos);
         }

         public ODEServer getODEServer() {
             return _ode;
         }
     }

 }
	package org.apache.ode.axis2;

	import org.apache.axis2.AxisFault;
	import org.apache.axis2.context.ConfigurationContextFactory;
	import org.apache.axis2.description.AxisOperation;
	import org.apache.axis2.description.AxisService;
	import org.apache.axis2.description.Parameter;
	import org.apache.axis2.description.WSDL11ToAxisServiceBuilder;
	import org.apache.axis2.engine.AxisServer;
	import org.apache.axis2.engine.MessageReceiver;
	import org.apache.commons.logging.Log;
	import org.apache.commons.logging.LogFactory;
	import org.apache.ode.axis2.hooks.ODEAxisService;
	import org.apache.ode.axis2.util.Axis2UriResolver;
	import org.apache.ode.axis2.util.Axis2WSDLLocator;
	import org.apache.ode.tools.sendsoap.cline.HttpSoapSender;
	import org.testng.annotations.AfterMethod;
	import org.testng.annotations.BeforeMethod;
	import org.testng.annotations.DataProvider;


	import javax.servlet.ServletException;
	import javax.wsdl.WSDLException;
	import javax.xml.namespace.QName;
	import java.io.File;
	import java.io.FileInputStream;
	import java.io.FileNotFoundException;
	import java.io.FileOutputStream;
	import java.io.IOException;
	import java.io.InputStream;
	import java.net.URI;
	import java.net.URISyntaxException;
	import java.net.URL;
	import java.util.ArrayList;
	import java.util.Collection;
	import java.util.Iterator;
	import java.util.List;

	/**
	* @author Matthieu Riou <mriou@apache.org>
	*/
	public abstract class Axis2TestBase {

	public static final int DEFAULT_TEST_PORT = 8888;

	private static final Log log = LogFactory.getLog(Axis2TestBase.class);

	protected ODEAxis2Server server;

	protected String config;

	protected static final String DO_NOT_OVERRIDE_CONFIG = "<DO_NOT_OVERRIDE_CONFIG>";

	private static String originalOdePersistence = System.getProperty("ode.persistence");
	private static String originalOdeConfigDir = System.getProperty("org.apache.ode.configDir");

	@DataProvider(name = "configs")
	protected Iterator<Object[]> createConfigData() {
	List<String> configDirList = new ArrayList<String>();
	if( !(this instanceof ODEConfigDirAware) \|\| ((ODEConfigDirAware)this).getODEConfigDir().contains("hib")) {
	addToConfigDirList(configDirList, "org.apache.ode.hibdbs");
	}
	if( !(this instanceof ODEConfigDirAware) \|\| !((ODEConfigDirAware)this).getODEConfigDir().contains("hib")) {
	addToConfigDirList(configDirList, "org.apache.ode.jpadbs");
	}

	if( configDirList.isEmpty() ) {
	// if no system property is set, fall back to default
	if( this instanceof ODEConfigDirAware ) {
	configDirList.add(((ODEConfigDirAware)this).getODEConfigDir());
	} else {
	configDirList.add(DO_NOT_OVERRIDE_CONFIG);
	}
	} else {
	System.out.println("Java system properties have been set to override ode configuration: " + configDirList);
	}

	final Iterator<String> itr = configDirList.iterator();
	return new Iterator<Object[]>() {
	public boolean hasNext() {
	return itr.hasNext();
	}

	public Object[] next() {
	config = itr.next();
	return new Object[] {};
	}

	public void remove() {
	}
	};
	}

	private void addToConfigDirList(List<String> configDirList, String propertyKey) {
	String dbs = System.getProperty(propertyKey);
	if( dbs != null ) {
	String[] configDirs = dbs.split(",");
	for( String configDir : configDirs ) {
	String trimmed = configDir.trim();
	if( trimmed.length() > 0 ) {
	configDirList.add(trimmed);
	}
	}
	}
	}

	public void startServer() throws Exception {
	startServer("webapp/WEB-INF", "webapp/WEB-INF/conf/axis2.xml");
	}

	public void startServer(String axis2RepoDir, String axis2ConfLocation) throws Exception {
	String odeRootAbsolutePath = getClass().getClassLoader().getResource("webapp/WEB-INF").getFile();
	String axis2RepoAbsolutePath = getClass().getClassLoader().getResource(axis2RepoDir).getFile();
	String axis2ConfAbsolutePath = axis2ConfLocation == null ? null : getClass().getClassLoader().getResource(axis2ConfLocation).getFile();
	server = new ODEAxis2Server(odeRootAbsolutePath, axis2RepoAbsolutePath, axis2ConfAbsolutePath);
	server.start();
	}

	public void stopServer() throws AxisFault {
	server.stop();
	}

	@BeforeMethod
	protected void setUp() throws Exception {
	/**
	* 1. If no settings are given from buildr, the test runs with the default config directory.
	* 2. If no settings are given from buildr and if the test implements ODEConfigDirAware, the test runs with
	* the config directory from the interface.
	* 3. If settings are given from buildr and if it's derby and openJPA, test falls back to the above 1 or 2.
	* 4. If settings are given from buildr and if it's derby and hibernate, test falls back to the above 2 or
	* uses -Dode.persistence=hibernate.
	*/
	if( config == null \|\| DO_NOT_OVERRIDE_CONFIG.equals(config) ) {
	System.out.println("Test config: default.");
	} else if("<jpa>".equals(config)) {
	if( this instanceof ODEConfigDirAware ) {
	config = ((ODEConfigDirAware)this).getODEConfigDir();
	System.out.println("Test config: " + config + ".");
	System.setProperty("org.apache.ode.configDir", config);
	} else {
	System.out.println("Test config: default.");
	}
	} else if("<hib>".equals(config)) {
	if( this instanceof ODEConfigDirAware ) {
	config = ((ODEConfigDirAware)this).getODEConfigDir();
	System.out.println("Test config: " + config + ".");
	System.setProperty("org.apache.ode.configDir", config);
	} else {
	// why does this not work?
	// System.out.println("Test config: -Dode.persistence=hibernate");
	// System.setProperty("ode.persistence", "hibernate");
	config = getClass().getClassLoader().getResource("webapp").getFile() + "/WEB-INF/conf.hib-derby";
	System.out.println("Test config: " + config + ".");
	System.setProperty("org.apache.ode.configDir", config);
	}
	} else {
	System.out.println("Test config: " + config + ".");
	System.setProperty("org.apache.ode.configDir", config);
	}

	startServer();
	}

	@AfterMethod
	protected void tearDown() throws Exception {
	stopServer();

	if( originalOdeConfigDir != null ) {
	System.setProperty("org.apache.ode.configDir", originalOdeConfigDir);
	} else {
	System.clearProperty("org.apache.ode.configDir");
	}
	if( originalOdeConfigDir != null ) {
	System.setProperty("ode.persistence", originalOdePersistence);
	} else {
	System.clearProperty("ode.persistence");
	}
	}

	protected class ODEAxis2Server extends AxisServer {

	ODEServer _ode;
	String odeRootDir;

	protected ODEAxis2Server(String odeRootDir, String axis2RepoDir, String axis2ConfLocation) throws Exception {
	super(false);
	this.odeRootDir = odeRootDir;
	if (log.isInfoEnabled()) {
	log.info("Ode Root Dir: " + odeRootDir);
	log.info("Axis2 Conf file: " + axis2ConfLocation);
	log.info("Axis2 Repo dir: " + axis2RepoDir);
	}

	configContext = ConfigurationContextFactory.createConfigurationContextFromFileSystem(axis2RepoDir, axis2ConfLocation);
	// do not use 8080 for tests
	configContext.getAxisConfiguration().getTransportIn("http").addParameter(new Parameter("port", ""+DEFAULT_TEST_PORT));
	}

	protected void start() throws AxisFault {
	super.start();
	_ode = new ODEServer();
	try {
	_ode.init(odeRootDir, configContext.getAxisConfiguration());
	} catch (ServletException e) {
	e.printStackTrace();
	}
	}

	public void stop() throws AxisFault {
	_ode.shutDown();
	super.stop();
	}

	public Collection<QName> deployProcess(String bundleName) {
	return _ode.getProcessStore().deploy(new File(getBundleDir(bundleName)));
	}

	public void undeployProcess(String bundleName) {
	_ode.getProcessStore().undeploy(new File(getBundleDir(bundleName)));
	}

	public boolean isDeployed(String bundleName) {
	return _ode.getProcessStore().getPackages().contains(bundleName);
	}

	/**
	* Creates and deploys an Axis service based on a provided MessageReceiver. The receiver
	* will be invoked for all invocations of that service.
	*/
	protected void deployService(String bundleName, String defFile, QName serviceName, String port,
	MessageReceiver receiver) throws WSDLException, IOException, URISyntaxException {
	URI wsdlUri = new File(getBundleDir(bundleName) + "/" + defFile).toURI();

	InputStream is = wsdlUri.toURL().openStream();
	WSDL11ToAxisServiceBuilder serviceBuilder = new ODEAxisService.WSDL11ToAxisPatchedBuilder(is, serviceName, port);
	serviceBuilder.setBaseUri(wsdlUri.toString());
	serviceBuilder.setCustomResolver(new Axis2UriResolver());
	serviceBuilder.setCustomWSLD4JResolver(new Axis2WSDLLocator(wsdlUri));
	serviceBuilder.setServerSide(true);

	AxisService axisService = serviceBuilder.populateService();
	axisService.setName(serviceName.getLocalPart());
	axisService.setWsdlFound(true);
	axisService.setCustomWsdl(true);
	axisService.setClassLoader(getConfigurationContext().getAxisConfiguration().getServiceClassLoader());

	Iterator operations = axisService.getOperations();
	while (operations.hasNext()) {
	AxisOperation operation = (AxisOperation) operations.next();
	if (operation.getMessageReceiver() == null) {
	operation.setMessageReceiver(receiver);
	}
	}
	getConfigurationContext().getAxisConfiguration().addService(axisService);
	}

	public String sendRequestFile(String endpoint, String bundleName, String filename) {
	try {
	return HttpSoapSender.doSend(new URL(endpoint),
	new FileInputStream(getBundleDir(bundleName) + "/" + filename), null, 0, null, null, null);
	} catch (IOException e) {
	throw new RuntimeException(e);
	}
	}

	protected String getBundleDir(String bundleName) {
	return getClass().getClassLoader().getResource(bundleName).getFile();
	}

	/**
	* Convenient methods to generate a WSDL for an Axis2 service. Often nice, but also often
	* generates crappy WSDL that aren't even valid (especially when faults are involved) so
	* use with care.
	*
	* @param serviceName
	* @param fileName
	* @throws AxisFault
	*/
	protected void generateWSDL(String serviceName, String fileName) throws AxisFault {
	FileOutputStream fos = null;
	try {
	fos = new FileOutputStream(fileName);
	} catch (FileNotFoundException e) {
	e.printStackTrace();
	}
	configContext.getAxisConfiguration().getService(serviceName).printWSDL(fos);
	}

	public ODEServer getODEServer() {
	return _ode;
	}
	}

	}