/* | |
* Licensed to the Apache Software Foundation (ASF) under one | |
* or more contributor license agreements. See the NOTICE file | |
* distributed with this work for additional information | |
* regarding copyright ownership. The ASF licenses this file | |
* to you 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. | |
*/ | |
package org.apache.axis2.jaxws.dispatch; | |
import junit.framework.Test; | |
import junit.framework.TestSuite; | |
import org.apache.axis2.jaxws.TestLogger; | |
import org.apache.axis2.jaxws.framework.AbstractTestCase; | |
import org.test.dispatch.jaxbsource.Invoke; | |
import org.test.dispatch.jaxbsource.ObjectFactory; | |
import javax.xml.bind.JAXBContext; | |
import javax.xml.bind.util.JAXBSource; | |
import javax.xml.namespace.QName; | |
import javax.xml.transform.Result; | |
import javax.xml.transform.Source; | |
import javax.xml.transform.Transformer; | |
import javax.xml.transform.TransformerFactory; | |
import javax.xml.transform.stream.StreamResult; | |
import javax.xml.ws.Dispatch; | |
import javax.xml.ws.Service; | |
import java.io.StringWriter; | |
/* | |
* This is a test case for Invoking Dispatch with a JAXBSource. | |
* test uses JAXB Objects from org.test.dispatch.jaxbsource package, create a request of JAXBSource type | |
* and invokes the service endpoint and reads the response of type Source. Assert failure if response not received. | |
*/ | |
public class JAXBSourceDispatchTests extends AbstractTestCase { | |
/** | |
* Invoke a sync Dispatch<JAXBSource> in PAYLOAD mode | |
*/ | |
private String url = "http://localhost:6060/axis2/services/SourceProviderService"; | |
private QName serviceName = new QName("http://ws.apache.org/axis2", "SourceProviderService"); | |
private QName portName =new QName("http://ws.apache.org/axis2", "SimpleProviderServiceSOAP11port0"); | |
public static Test suite() { | |
return getTestSetup(new TestSuite(JAXBSourceDispatchTests.class)); | |
} | |
public void testJAXBSourceSyncPayloadMode() throws Exception { | |
TestLogger.logger.debug("---------------------------------------"); | |
TestLogger.logger.debug("test: " + getName()); | |
try{ | |
// Initialize the JAX-WS client artifacts | |
Service svc = Service.create(serviceName); | |
svc.addPort(portName, null, url); | |
Dispatch<Source> dispatch = svc.createDispatch(portName, Source.class, Service.Mode.PAYLOAD); | |
//Create JAXBContext and JAXBSource here. | |
ObjectFactory factory = new ObjectFactory(); | |
Invoke invokeObj = factory.createInvoke(); | |
invokeObj.setInvokeStr("Some Request"); | |
JAXBContext ctx = JAXBContext.newInstance("org.test.dispatch.jaxbsource"); | |
JAXBSource jbSrc = new JAXBSource(ctx.createMarshaller(), invokeObj); | |
// Invoke the Dispatch | |
TestLogger.logger.debug(">> Invoking sync Dispatch"); | |
//Invoke Server endpoint and read response | |
Source response = dispatch.invoke(jbSrc); | |
assertNotNull("dispatch invoke returned null", response); | |
//Print the response as string. | |
StringWriter writer = new StringWriter(); | |
Transformer t = TransformerFactory.newInstance().newTransformer(); | |
Result result = new StreamResult(writer); | |
t.transform(response, result); | |
TestLogger.logger.debug("Response On Client: \n" + writer.getBuffer().toString()); | |
// Invoke a second time | |
jbSrc = new JAXBSource(ctx.createMarshaller(), invokeObj); | |
// Invoke the Dispatch | |
TestLogger.logger.debug(">> Invoking sync Dispatch"); | |
//Invoke Server endpoint and read response | |
response = dispatch.invoke(jbSrc); | |
assertNotNull("dispatch invoke returned null", response); | |
//Print the response as string. | |
writer = new StringWriter(); | |
t = TransformerFactory.newInstance().newTransformer(); | |
result = new StreamResult(writer); | |
t.transform(response, result); | |
TestLogger.logger.debug("Response On Client: \n" + writer.getBuffer().toString()); | |
TestLogger.logger.debug("---------------------------------------"); | |
}catch(Exception e){ | |
e.printStackTrace(); | |
} | |
} | |
} |