net.bpelunit.framework.ui.ant.BPELUnit Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ant-plugin Show documentation
The newest version!
/**
 * This file belongs to the BPELUnit utility and Eclipse plugin set. See enclosed
 * license file for more information.
 * 
 */
package net.bpelunit.framework.ui.ant;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.logging.Level;

import org.apache.commons.io.FilenameUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Task;
import net.bpelunit.framework.BPELUnitRunner;
import net.bpelunit.framework.exception.ConfigurationException;

/**
 * 
 * BPELUnit Ant Task
 * 
 * 
 * 
 * Invocation:
 * 
 * 
 *     <bpelunit testsuite="testsuite.bpts" haltOnError="true" haltOnFailure="true" bpelunitdir="c:\location"> 
 *     <output type="plain" file="false" />
 *     <logging level="plain" file="false" />
 *     </bpelunit>
 * 
 * 
 * 
 * 
 * @version $Id$
 * @author Philip Mayer
 * 
 */
public class BPELUnit extends Task {

	/**
	 * The BPELUnit home directory
	 */
	private String fBPELUnitDir = System.getenv("BPELUNIT_HOME");

	/**
	 * If true, the framework halts on errors
	 */
	private boolean fHaltOnError;

	/**
	 * if true, the framework halts on failures
	 */
	private boolean fHaltOnFailure;

	/**
	 * List of "outputters"
	 */
	private List fOutputList= new ArrayList();

	/**
	 * List of "loggers"
	 */
	private List fLoggingList= new ArrayList();

	/**
	 * The test suite file
	 */
	private File fTestSuiteFile;

	@Override
	public void execute() {
		checkAttributes();
		
		HashMap options = new HashMap();
		options.put(BPELUnitRunner.HALT_ON_ERROR, Boolean.toString(fHaltOnError));
		options.put(BPELUnitRunner.HALT_ON_FAILURE, Boolean.toString(fHaltOnFailure));

		BPELUnitAntRunner runner = new BPELUnitAntRunner(fBPELUnitDir, fLoggingList, fOutputList);
		try {
			runner.initialize(options);
			runner.run(fTestSuiteFile);
		} catch (Exception e) {
			throw new BuildException(e.getMessage(), e);
		}
	}

	private void checkAttributes() {
		if (fTestSuiteFile == null) {
			throw new BuildException("The testsuite argument is required.");
		}
		
		if (!fTestSuiteFile.exists()) {
			throw new BuildException("The testsuite file does not exist.");
		}

		if (StringUtils.isEmpty(fBPELUnitDir)) {
			throw new BuildException("The BPELUnit directory is required - either specify an argument or set environment variable BPELUNIT_HOME.");
		}
		
		for (Output output : fOutputList) {
			if (!output.hasValidStyle()) {
				throw new BuildException("An output specification is missing the style attribute, or wrong style specified.");
			}
		}

		for (Logging logging : fLoggingList) {
			if (!logging.hasValidLevel()) {
				throw new BuildException("A logging specification is missing the level attribute, or wrong level specified.");
			}
		}
	}

	public void setTestSuite(String testSuite) {
		fTestSuiteFile = new File(testSuite);
	}

	public void setBPELUnitDir(String bpelUnitDir) {
		fBPELUnitDir= bpelUnitDir;
	}

	public void setHaltOnError(boolean haltOnError) {
		fHaltOnError= haltOnError;
	}

	public void setHaltOnFailure(boolean haltOnFailure) {
		fHaltOnFailure= haltOnFailure;
	}

	public void addConfiguredOutput(Output output) {
		fOutputList.add(output);
	}

	public void addConfiguredLogging(Logging logging) {
		fLoggingList.add(logging);
	}


	public static class Output extends Extension {

		public static final String STYLE_PLAIN= "PLAIN";
		public static final String STYLE_XML= "XML";

		private String fStyle;

		public String getStyle() {
			return fStyle;
		}

		public void setStyle(String style) {
			fStyle= style;
		}


		public boolean hasValidStyle() {
			return STYLE_PLAIN.equals(fStyle) || STYLE_XML.equals(fStyle);
		}
	}

	public static class Extension {

		private String fFile;
		private OutputStream fOutput;

		public OutputStream getOutput() {
			return fOutput;
		}

		public void initialize() throws ConfigurationException {
			fOutput= createOutputStream();
		}

		public void dispose() {
			IOUtils.closeQuietly(fOutput);
		}

		private OutputStream createOutputStream() throws ConfigurationException {
			if (fFile != null) {
				String dir= FilenameUtils.getPath(fFile);
				if (!"".equals(dir)) {
					new File(dir).mkdirs();
				}
				try {
					return new FileOutputStream(fFile);
				} catch (FileNotFoundException e) {
					throw new ConfigurationException("Cannot create file with name " + fFile + " for output.", e);
				}
			} else {
				return System.out;
			}
		}

		public void write(String line) throws IOException {
			fOutput.write(line.getBytes());
		}

		public void setFile(String file) {
			fFile= file;
		}

		public String getFile() {
			return fFile;
		}
	}

	public static class Logging extends Extension {

		private String fLevel;

		public boolean hasValidLevel() {
			if (fLevel == null) {
				return false;
			}
			try {
				Level.parse(fLevel);
				return true;
			} catch (IllegalArgumentException e) {
				return false;
			}
		}

		public void setLevel(String type) {
			fLevel= type;
		}

		public String getLevel() {
			return fLevel;
		}

		public OutputStream getOutputStream() {
			return getOutput();
		}
	}
}