package org.codehaus.xfire.plexus;import java.io.ByteArrayOutputStream;import java.io.InputStream;import java.io.InputStreamReader;import java.io.Reader;import java.io.StringReader;import java.util.HashMap;import java.util.List;import java.util.Map;import javax.xml.stream.XMLInputFactory;import javax.xml.stream.XMLStreamException;import org.codehaus.plexus.PlexusTestCase;import org.codehaus.xfire.MessageContext;import org.codehaus.xfire.XFire;import org.codehaus.xfire.exchange.InMessage;import org.codehaus.xfire.service.Service;import org.codehaus.xfire.service.ServiceFactory;import org.codehaus.xfire.service.ServiceRegistry;import org.codehaus.xfire.service.binding.MessageBindingProvider;import org.codehaus.xfire.service.binding.ObjectServiceFactory;import org.codehaus.xfire.soap.Soap11;import org.codehaus.xfire.soap.Soap12;import org.codehaus.xfire.soap.SoapConstants;import org.codehaus.xfire.test.XPathAssert;import org.codehaus.xfire.transport.Channel;import org.codehaus.xfire.transport.Transport;import org.codehaus.xfire.transport.local.LocalTransport;import org.codehaus.xfire.util.STAXUtils;import org.codehaus.xfire.util.jdom.StaxBuilder;import org.codehaus.xfire.wsdl.WSDLWriter;import org.jdom.Document;import org.jdom.Element;import org.jdom.output.XMLOutputter;/** * Contains helpful methods to test SOAP services. * * @author <a href="mailto:dan@envoisolutions.com">Dan Diephouse</a> */public class PlexusXFireTest        extends PlexusTestCase{    private ServiceFactory factory;        private XMLInputFactory defaultInputFactory = XMLInputFactory.newInstance();        /**     * Namespaces for the XPath expressions.     */    private Map namespaces = new HashMap();    protected void printNode(Document node)        throws Exception    {        XMLOutputter writer = new XMLOutputter();        writer.output(node, System.out);    }        protected void printNode(Element node)            throws Exception    {        XMLOutputter writer = new XMLOutputter();        writer.output(node, System.out);    }    /**     * Invoke a service with the specified document.     *     * @param service  The name of the service.     * @param document The request as an xml document in the classpath.     */    protected Document invokeService(String service, String document)            throws Exception    {        ByteArrayOutputStream out = new ByteArrayOutputStream();        MessageContext context = new MessageContext();        context.setXFire(getXFire());        context.setProperty(Channel.BACKCHANNEL_URI, out);        if (service != null)            context.setService(getServiceRegistry().getService(service));                InputStream stream = getResourceAsStream(document);         InMessage msg = new InMessage(STAXUtils.createXMLStreamReader(stream, "UTF-8",null));        Transport t = getXFire().getTransportManager().getTransport(LocalTransport.BINDING_ID);        Channel c = t.createChannel();                c.receive(context, msg);                String response = out.toString();        if (response == null || response.length() == 0)            return null;        return readDocument(response);    }    protected Document readDocument(String text)            throws XMLStreamException    {        return readDocument(text, defaultInputFactory);    }    protected Document readDocument(String text, XMLInputFactory ifactory)            throws XMLStreamException    {        try        {            StaxBuilder builder = new StaxBuilder(ifactory);            return builder.build(new StringReader(text));        }        catch (XMLStreamException e)        {            System.err.println("Could not read the document!");            System.err.println(text);            throw e;        }    }        protected Document getWSDLDocument(String service)            throws Exception    {        ByteArrayOutputStream out = new ByteArrayOutputStream();        getXFire().generateWSDL(service, out);        return readDocument(out.toString());    }    /**     * @see junit.framework.TestCase#setUp()     */    protected void setUp()            throws Exception    {        super.setUp();                addNamespace("s", Soap11.getInstance().getNamespace());        addNamespace("soap12", Soap12.getInstance().getNamespace());    }    /**     * Assert that the following XPath query selects one or more nodes.     *     * @param xpath     */    public List assertValid(String xpath, Object node)            throws Exception    {        return XPathAssert.assertValid(xpath, node, namespaces);    }    /**     * Assert that the following XPath query selects no nodes.     *     * @param xpath     */    public List assertInvalid(String xpath, Object node)            throws Exception    {        return XPathAssert.assertInvalid(xpath, node, namespaces);    }    /**     * Asser that the text of the xpath node retrieved is equal to the value specified.     *     * @param xpath     * @param value     * @param node     */    public void assertXPathEquals(String xpath, String value, Document node)            throws Exception    {        XPathAssert.assertXPathEquals(xpath, value, node, namespaces);    }    public void assertNoFault(Document node)            throws Exception    {        XPathAssert.assertNoFault(node);    }    /**     * Add a namespace that will be used for XPath expressions.     *     * @param ns  Namespace name.     * @param uri The namespace uri.     */    public void addNamespace(String ns, String uri)    {        namespaces.put(ns, uri);    }    /**     * Get the WSDL for a service.     *     * @param service The name of the service.     */    protected WSDLWriter getWSDL(String service)            throws Exception    {        ServiceRegistry reg = getServiceRegistry();        Service hello = reg.getService(service);        return hello.getWSDLWriter();    }    public ServiceFactory getServiceFactory() throws Exception    {        if (factory == null)        {            ObjectServiceFactory ofactory =                 new ObjectServiceFactory(getXFire().getTransportManager(),                                         new MessageBindingProvider());                        ofactory.setStyle(SoapConstants.STYLE_MESSAGE);                        factory = ofactory;        }        return factory;    }    public void setServiceFactory(ServiceFactory factory)    {        this.factory = factory;    }    protected InputStream getResourceAsStream(String resource)    {        return getClass().getResourceAsStream(resource);    }    protected Reader getResourceAsReader(String resource)    {        return new InputStreamReader(getResourceAsStream(resource));    }    protected XFire getXFire()            throws Exception    {        return (XFire) lookup(XFire.ROLE);    }    protected ServiceRegistry getServiceRegistry()            throws Exception    {        return getXFire().getServiceRegistry();    }}