AbstractTestCase

package net.kwfgrid.gwes;

import java.io.*;
import java.util.List;
import java.rmi.RemoteException;

import junit.framework.TestCase;
import junit.framework.Assert;
import net.kwfgrid.gwes.exception.NoSuchWorkflowException;
import net.kwfgrid.gwes.exception.DatabaseException;
import net.kwfgrid.gwes.exception.StateTransitionException;
import net.kwfgrid.gworkflowdl.protocol.*;
import net.kwfgrid.gworkflowdl.protocol.xupdate.XUModificationHandler;
import net.kwfgrid.gworkflowdl.protocol.xupdate.XUMethodCallEncoder;
import net.kwfgrid.gworkflowdl.protocol.server.DefaultServerDelegateExt;
import net.kwfgrid.gworkflowdl.protocol.server.ServerCreator;
import net.kwfgrid.gworkflowdl.structure.*;
import org.apache.log4j.Logger;
import org.jdom.JDOMException;
import org.jdom.Document;
import org.jdom.Namespace;
import org.jdom.Element;
import org.jdom.input.SAXBuilder;
import org.jaxen.JaxenException;
import org.jaxen.XPath;
import org.jaxen.SimpleNamespaceContext;
import org.jaxen.jdom.JDOMXPath;


/**
 * Abstract base class for test cases.
 */
public abstract class AbstractTestCase
    extends TestCase  {

    static Logger logger = Logger.getLogger(AbstractTestCase.class);

    public final String userID;

    public final SimpleNamespaceContext jaxenNamespaceContext;

    public final static Namespace NAMESPACE_GWDL = Namespace.getNamespace("http://www.gridworkflow.org/gworkflowdl");

    /**
     * Basedir for all file I/O. Important when running tests from
     * the reactor.
     */
    public String basedir = System.getProperty("basedir");

    /**
     * Constructor.
     */
    public AbstractTestCase(String testName)
    {
        super(testName);
        userID = System.getProperty("user.name");
        jaxenNamespaceContext = new SimpleNamespaceContext();
        jaxenNamespaceContext.addNamespace("xsd", "http://www.w3.org/2001/XMLSchema");
        jaxenNamespaceContext.addNamespace("xsi", "http://www.w3.org/2001/XMLSchema-instance");
        jaxenNamespaceContext.addNamespace("gwdl", "http://www.gridworkflow.org/gworkflowdl");
    }

    /**
     * Get test input file.
     *
     * @param path Path to test input file.
     */
    public String getTestFile(String path)
    {

        File file;
        // absolute path
        if (path.startsWith("/")) {
            file = new File(path);
        }
        // relative path
        else {
            if (basedir == null) {
                basedir = "./";
            }
            // check <basedir>/target/test-classes/...
            file = new File(basedir+"/target/test-classes",path);
            if (!file.exists()) {
                // check <basedir>/...
                file = new File(basedir,path);
            }
        }
        return file.getAbsolutePath();
    }

     public List<Element> extractTokenChildElements(String xml) throws IOException, JDOMException, JaxenException {
         final SAXBuilder builder = new SAXBuilder(false);
//         builder.setFeature("http://apache.org/xml/features/validation/schema", true);
//         String wsdldefs = "http://schemas.xmlsoap.org/wsdl/ http://schemas.xmlsoap.org/wsdl/";
//         builder.setProperty("http://apache.org/xml/properties/schema/external-schemaLocation", wsdldefs);
         Document document = builder.build(new StringReader(xml));
         XPath xpath = new JDOMXPath("/gwdl:workflow/gwdl:place/gwdl:token/*");
         xpath.setNamespaceContext(jaxenNamespaceContext);
         return xpath.selectNodes(document);
     }

    public List<Element> extractTokenElements(String xml) throws IOException, JDOMException, JaxenException {
        final SAXBuilder builder = new SAXBuilder(false);
        Document document = builder.build(new StringReader(xml));
        XPath xpath = new JDOMXPath("/gwdl:workflow/gwdl:place/gwdl:token");
        xpath.setNamespaceContext(jaxenNamespaceContext);
        return xpath.selectNodes(document);
    }

    public String stripTokenIDs(String xml) {
        if (xml==null) return null;
        String noD = xml.replaceAll("token ID=\"d.*\"","token ID=\"dX\"");
        return noD.replaceAll("token ID=\"c.*\"","token ID=\"cX\"");
    }

     public String extractOccurrenceSequence(String xml) throws IOException, JDOMException, JaxenException {
        final SAXBuilder builder = new SAXBuilder(false);
        SimpleNamespaceContext nsContext = new SimpleNamespaceContext();
        nsContext.addNamespace("xsd", "http://www.w3.org/2001/XMLSchema");
        nsContext.addNamespace("xsi", "http://www.w3.org/2001/XMLSchema-instance");
        nsContext.addNamespace("gwdl", "http://www.gridworkflow.org/gworkflowdl");
        Document document = builder.build(new StringReader(xml));
        XPath xpath = new JDOMXPath("/gwdl:workflow/gwdl:property[@name=\"occurrence.sequence\"]");
        xpath.setNamespaceContext(nsContext);
        return xpath.stringValueOf(document);
    }


     public String readfile(String fn) throws IOException {
         BufferedReader in = (new BufferedReader(new FileReader(getTestFile(fn))));
         String line;
         StringBuffer buffer = new StringBuffer();
         line = in.readLine();
         while (line != null) {
             buffer.append(line);
             buffer.append('\n');
             line = in.readLine();
         }
         return buffer.toString();
     }

    public List xpath(String xpathS, String xml) throws IOException, JDOMException, JaxenException {
        final SAXBuilder builder = new SAXBuilder(false);
        Document document = builder.build(new StringReader(xml));
        SimpleNamespaceContext nsContext = new SimpleNamespaceContext();
        nsContext.addNamespace("xsd", "http://www.w3.org/2001/XMLSchema");
        nsContext.addNamespace("xsi", "http://www.w3.org/2001/XMLSchema-instance");
        nsContext.addNamespace("gwdl", "http://www.gridworkflow.org/gworkflowdl");
        XPath xpath = new JDOMXPath(xpathS);
        xpath.setNamespaceContext(nsContext);
        return xpath.selectNodes(document);
    }

    protected void configureFactory() {
        //configure protocol
        IMethodCallStrategy defaults = new DefaultMethodCallStrategy();
        IMethodCallStrategy protocol = new DefaultServerDelegateExt();
        IModificationHandler handler = new XUModificationHandler();
        IMethodCallEncoder encoder = new XUMethodCallEncoder();
        Protocol.setDefaultMethodCallStrategy(defaults);
        Protocol.setProtocolMethodCallStrategy(protocol);
        Protocol.setModificationHandler(handler);
        Protocol.setMethodCallEncoder(encoder);
        Creator creator = new ServerCreator(new DefaultCreator());
        Factory.setCreator(creator);
    }


}