/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2004-2013 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* http://glassfish.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package client.dispatch.header.doclit.client;
import junit.framework.TestCase;
import javax.xml.ws.Service;
import javax.xml.ws.WebServiceException;
import javax.xml.ws.Dispatch;
import javax.xml.ws.Binding;
import javax.xml.ws.soap.SOAPBinding;
import javax.xml.bind.JAXBContext;
import javax.xml.transform.*;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.sax.SAXSource;
import javax.xml.soap.SOAPMessage;
import javax.xml.soap.MessageFactory;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.namespace.QName;
import java.net.URL;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.io.*;
import client.dispatch.header.doclit.client.DispatchHeaderTest;
import testutil.ClientServerTestUtil;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.w3c.dom.Node;
public class TestCaseBase extends TestCase {
private Service service;
private Service serviceWithPorts;
static final String BINDING_ID_STRING = SOAPBinding.SOAP11HTTP_BINDING;
protected String endpointAddress;
protected QName portQName;
protected QName serviceQName;
protected JAXBContext context;
protected OutputStream log;
public void init(String endpointAddress,
QName portQName, QName serviceQName,
JAXBContext context) {
this.endpointAddress = endpointAddress;
this.portQName = portQName;
this.serviceQName = serviceQName;
this.context = context;
}
protected void setLog(OutputStream log){
this.log = log;
}
public TestCaseBase(String string) {
super(string);
}
private void createService() {
try {
service = Service.create(serviceQName);
//does service.addPort(portQName, bindingIdString, endpointAddress
addPort(service, portQName, BINDING_ID_STRING, endpointAddress);
} catch (Exception e) {
e.printStackTrace();
fail("Error creating service");
}
}
protected void addPort( Service service, QName portQName, String bindingIdString, String endpointAddress) throws URISyntaxException {
service.addPort(portQName, bindingIdString, endpointAddress);
}
private void createServiceWithWSDL(String endpointAddress, QName serviceQName) {
//still need local transport for this
URL serviceURL = null;
try {
serviceURL = new URL(endpointAddress + "?wsdl");
} catch (MalformedURLException e) {
fail("Error creating service with wsdl");
}
try {
serviceWithPorts = Service.create(serviceURL,serviceQName);
} catch (WebServiceException e) {
e.printStackTrace();
}
}
public Dispatch createDispatchJAXB() {
return service.createDispatch(portQName, context, Service.Mode.PAYLOAD);
}
public Dispatch createDispatchJAXBPortsAvailable() {
return serviceWithPorts.createDispatch(portQName, context, Service.Mode.PAYLOAD);
}
public Dispatch createDispatchSource() {
return service.createDispatch(portQName, Source.class, Service.Mode.PAYLOAD);
}
public Dispatch<SOAPMessage> createDispatchSOAPMessage() {
return service.createDispatch(portQName, SOAPMessage.class, Service.Mode.MESSAGE);
}
public Dispatch<Source> createDispatchSOAPMessageSource() {
return service.createDispatch(portQName, Source.class, Service.Mode.MESSAGE);
}
public Dispatch getDispatchJAXB() {
createService();
return createDispatchJAXB();
}
public Dispatch getDispatchSource() {
createService();
return createDispatchSource();
}
public Dispatch getDispatchSOAPMessage() {
createService();
return createDispatchSOAPMessage();
}
public Dispatch getDispatchSOAPMessageSource() {
createService();
return createDispatchSOAPMessageSource();
}
private Source makeSaxSource(String msg) {
byte[] bytes = msg.getBytes();
ByteArrayInputStream saxinputStream = new ByteArrayInputStream(bytes);
InputSource inputSource = new InputSource(saxinputStream);
return new SAXSource(inputSource);
}
protected Source makeStreamSource(String msg) {
byte[] bytes = msg.getBytes();
ByteArrayInputStream sinputStream = new ByteArrayInputStream(bytes);
return new StreamSource(sinputStream);
}
private Source makeDOMSource(String msg) {
byte[] bytes = msg.getBytes();
ByteArrayInputStream inputStream = new ByteArrayInputStream(bytes);
return new DOMSource(createDOMNode(inputStream));
}
public Node createDOMNode(InputStream inputStream) {
DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
dbf.setNamespaceAware(true);
dbf.setValidating(false);
try {
DocumentBuilder builder = dbf.newDocumentBuilder();
try {
return builder.parse(inputStream);
} catch (SAXException e) {
e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates.
} catch (IOException e) {
e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates.
}
} catch (ParserConfigurationException pce) {
IllegalArgumentException iae = new IllegalArgumentException(pce.getMessage());
iae.initCause(pce);
throw iae;
}
return null;
}
SOAPMessage getSOAPMessage(Source msg) throws Exception {
MessageFactory factory = MessageFactory.newInstance();
SOAPMessage message = factory.createMessage();
message.getSOAPPart().setContent((Source) msg);
message.saveChanges();
return message;
}
private String sourceToXMLString(Source result) {
String xmlResult = null;
try {
TransformerFactory factory = TransformerFactory.newInstance();
Transformer transformer = factory.newTransformer();
transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");
transformer.setOutputProperty(OutputKeys.METHOD, "xml");
OutputStream out = new ByteArrayOutputStream();
StreamResult streamResult = new StreamResult();
streamResult.setOutputStream(out);
transformer.transform(result, streamResult);
xmlResult = streamResult.getOutputStream().toString();
} catch (TransformerException e) {
e.printStackTrace();
}
return xmlResult;
}
}