Java tutorial
package com.canoo.webtest.boundary; /* * Copyright 2003-2004 The Apache Software Foundation * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * * This is a MODIFIED VERSION of ScriptRunner from Ant. * Modifications Copyright (c) ASERT 2005. * Modifications released under the Canoo WebTest license. */ import org.apache.bsf.BSFException; import org.apache.bsf.BSFManager; import org.apache.commons.io.IOUtils; import org.apache.tools.ant.BuildException; import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStream; import java.util.HashMap; import java.util.Iterator; import java.util.Map; /** * This class is used to run BSF scripts */ public class ResetScriptRunner { // Register Groovy ourselves, since BSF does not // natively support it (yet). // This "hack" can be removed once BSF has been // modified to support Groovy or more dynamic // registration. static { BSFManager.registerScriptingEngine("groovy", "org.codehaus.groovy.bsf.GroovyEngine", new String[] { "groovy", "gy" }); } /** * Script language */ private String fLanguage; /** * Script content */ private String fScript; /** * Beans to be provided to the script */ private Map fBeans = new HashMap(); private BSFManager fManager; { reset(); } public void reset() { fScript = ""; } /** * Add a list of named objects to the list to be exported to the script * * @param dictionary a map of objects to be placed into the script context * indexed by String names. */ public void addBeans(final Map dictionary) { for (Iterator i = dictionary.keySet().iterator(); i.hasNext();) { final String key = (String) i.next(); try { final Object val = dictionary.get(key); addBean(key, val); } catch (BuildException ex) { // The key is in the dictionary but cannot be retrieved // This is usually due to references that refer to tasks // that have not been taskdefed in the current run. // Ignore } } } /** * Add a single object into the script context. * * @param key the name in the context this object is to stored under. * @param bean the object to be stored in the script context. */ public void addBean(final String key, final Object bean) { boolean isValid = key.length() > 0 && Character.isJavaIdentifierStart(key.charAt(0)); for (int i = 1; isValid && i < key.length(); i++) { isValid = Character.isJavaIdentifierPart(key.charAt(i)); } if (isValid) { fBeans.put(key, bean); } } /** * Do the work. * * @param execName the name that will be passed to BSF for this script * execution. * @throws BuildException if someting goes wrong exectuing the script. */ public void executeScript(final String execName) throws BuildException { if (fLanguage == null) { throw new BuildException("script language must be specified"); } try { if (fManager == null) { fManager = prepareManager(); } // execute the script fManager.exec(fLanguage, execName, 0, 0, fScript); } catch (BSFException be) { throw createBuildException(be); } } /** * Do the work. * * @param execName the name that will be passed to BSF for this script * execution. * @throws BuildException if someting goes wrong exectuing the script. */ public String evalScript(final String execName) throws BuildException { if (fLanguage == null) { throw new BuildException("script language must be specified"); } try { if (fManager == null) { fManager = prepareManager(); } // evaluate the expression return fManager.eval(fLanguage, execName, 0, 0, fScript).toString(); } catch (BSFException be) { throw createBuildException(be); } } private BSFManager prepareManager() throws BSFException { final BSFManager manager = new BSFManager(); for (Iterator i = fBeans.keySet().iterator(); i.hasNext();) { final String key = (String) i.next(); final Object value = fBeans.get(key); if (value != null) { manager.declareBean(key, value, value.getClass()); } else { // BSF uses a hashtable to store values // so cannot declareBean with a null value // So need to remove any bean of this name as // that bean should not be visible manager.undeclareBean(key); } } return manager; } private static BuildException createBuildException(final BSFException be) { Throwable t = be; final Throwable te = be.getTargetException(); if (te != null) { if (te instanceof BuildException) { return (BuildException) te; } else { t = te; } } return new BuildException(t); } /** * Defines the language (required). * * @param lang the scripting language name for the script. */ public void setLanguage(final String lang) { this.fLanguage = lang; } /** * Get the script language * * @return the script language */ public String getLanguage() { return fLanguage; } /** * Load the script from an external file ; optional. * * @param file the file containing the script source. */ public void setSrc(final File file) { if (!file.exists()) { throw new BuildException("file " + file.getPath() + " not found."); } InputStream inputStream = null; try { inputStream = new FileInputStream(file); fScript += IOUtils.toString(inputStream); } catch (IOException e) { throw new BuildException(e); } finally { IOUtils.closeQuietly(inputStream); } } /** * Set the script text. * * @param text a component of the script text to be added. */ public void addText(final String text) { fScript += text; } }