/**
* 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.cxf.systest.swa;
import java.awt.Image;
import java.io.InputStream;
import java.net.URL;
import javax.activation.DataHandler;
import javax.imageio.ImageIO;
import javax.mail.util.ByteArrayDataSource;
import javax.xml.namespace.QName;
import javax.xml.soap.AttachmentPart;
import javax.xml.soap.MessageFactory;
import javax.xml.soap.SOAPBody;
import javax.xml.soap.SOAPMessage;
import javax.xml.transform.Source;
import javax.xml.transform.stream.StreamSource;
import javax.xml.ws.Dispatch;
import javax.xml.ws.Holder;
import javax.xml.ws.Service;
import org.apache.cxf.helpers.IOUtils;
import org.apache.cxf.swa.SwAService;
import org.apache.cxf.swa.SwAServiceInterface;
import org.apache.cxf.swa.types.DataStruct;
import org.apache.cxf.swa.types.OutputResponseAll;
import org.apache.cxf.swa.types.VoidRequest;
import org.apache.cxf.testutil.common.AbstractBusClientServerTestBase;
import org.junit.BeforeClass;
import org.junit.Test;
public class ClientServerSwaTest extends AbstractBusClientServerTestBase {
@BeforeClass
public static void startServers() throws Exception {
assertTrue("server did not launch correctly", launchServer(Server.class, true));
}
@Test
public void testSwa() throws Exception {
SwAService service = new SwAService();
SwAServiceInterface port = service.getSwAServiceHttpPort();
// ((BindingProvider)port).getRequestContext().put(BindingProvider.ENDPOINT_ADDRESS_PROPERTY,
// "http://localhost:9037/swa");
Holder<String> textHolder = new Holder<String>();
Holder<DataHandler> data = new Holder<DataHandler>();
ByteArrayDataSource source = new ByteArrayDataSource("foobar".getBytes(), "application/octet-stream");
DataHandler handler = new DataHandler(source);
data.value = handler;
textHolder.value = "Hi";
port.echoData(textHolder, data);
InputStream bis = null;
bis = data.value.getDataSource().getInputStream();
byte b[] = new byte[10];
bis.read(b, 0, 10);
String string = new String(b);
assertEquals("testfoobar", string);
assertEquals("Hi", textHolder.value);
}
@Test
public void testSwaWithHeaders() throws Exception {
SwAService service = new SwAService();
SwAServiceInterface port = service.getSwAServiceHttpPort();
// ((BindingProvider)port).getRequestContext().put(BindingProvider.ENDPOINT_ADDRESS_PROPERTY,
// "http://localhost:9037/swa");
Holder<String> textHolder = new Holder<String>();
Holder<String> headerHolder = new Holder<String>();
Holder<DataHandler> data = new Holder<DataHandler>();
ByteArrayDataSource source = new ByteArrayDataSource("foobar".getBytes(), "application/octet-stream");
DataHandler handler = new DataHandler(source);
data.value = handler;
textHolder.value = "Hi";
headerHolder.value = "Header";
port.echoDataWithHeader(textHolder, data, headerHolder);
InputStream bis = null;
bis = data.value.getDataSource().getInputStream();
byte b[] = new byte[10];
bis.read(b, 0, 10);
String string = new String(b);
assertEquals("testfoobar", string);
assertEquals("Hi", textHolder.value);
assertEquals("Header", headerHolder.value);
}
@Test
public void testSwaDataStruct() throws Exception {
SwAService service = new SwAService();
SwAServiceInterface port = service.getSwAServiceHttpPort();
// ((BindingProvider)port).getRequestContext().put(BindingProvider.ENDPOINT_ADDRESS_PROPERTY,
// "http://localhost:9037/swa");
Holder<DataStruct> structHolder = new Holder<DataStruct>();
ByteArrayDataSource source = new ByteArrayDataSource("foobar".getBytes(), "application/octet-stream");
DataHandler handler = new DataHandler(source);
DataStruct struct = new DataStruct();
struct.setDataRef(handler);
structHolder.value = struct;
port.echoDataRef(structHolder);
handler = structHolder.value.getDataRef();
InputStream bis = null;
bis = handler.getDataSource().getInputStream();
byte b[] = new byte[10];
bis.read(b, 0, 10);
String string = new String(b);
assertEquals("testfoobar", string);
}
@Test
public void testSwaTypes() throws Exception {
if (Boolean.getBoolean("java.awt.headless")) {
System.out.println("Running headless. Skipping test as Images may not work.");
return;
}
SwAService service = new SwAService();
SwAServiceInterface port = service.getSwAServiceHttpPort();
URL url1 = this.getClass().getResource("resources/attach.text");
URL url2 = this.getClass().getResource("resources/attach.html");
URL url3 = this.getClass().getResource("resources/attach.xml");
URL url4 = this.getClass().getResource("resources/attach.jpeg1");
URL url5 = this.getClass().getResource("resources/attach.jpeg2");
DataHandler dh1 = new DataHandler(url1);
DataHandler dh2 = new DataHandler(url2);
DataHandler dh3 = new DataHandler(url3);
//DataHandler dh4 = new DataHandler(url4);
//DataHandler dh5 = new DataHandler(url5);
Holder<DataHandler> attach1 = new Holder<DataHandler>();
attach1.value = dh1;
Holder<DataHandler> attach2 = new Holder<DataHandler>();
attach2.value = dh2;
Holder<Source> attach3 = new Holder<Source>();
attach3.value = new StreamSource(dh3.getInputStream());
Holder<Image> attach4 = new Holder<Image>();
Holder<Image> attach5 = new Holder<Image>();
attach4.value = ImageIO.read(url4);
attach5.value = ImageIO.read(url5);
VoidRequest request = new VoidRequest();
OutputResponseAll response = port.echoAllAttachmentTypes(request, attach1, attach2, attach3, attach4,
attach5);
assertNotNull(response);
}
@Test
public void testSwaTypesWithDispatchAPI() throws Exception {
if (Boolean.getBoolean("java.awt.headless")) {
System.out.println("Running headless. Skipping test as Images may not work.");
return;
}
URL url1 = this.getClass().getResource("resources/attach.text");
URL url2 = this.getClass().getResource("resources/attach.html");
URL url3 = this.getClass().getResource("resources/attach.xml");
URL url4 = this.getClass().getResource("resources/attach.jpeg1");
URL url5 = this.getClass().getResource("resources/attach.jpeg2");
byte[] bytes = IOUtils.readBytesFromStream(url1.openStream());
byte[] bigBytes = new byte[bytes.length * 50];
for (int x = 0; x < 50; x++) {
System.arraycopy(bytes, 0, bigBytes, x * bytes.length, bytes.length);
}
DataHandler dh1 = new DataHandler(new ByteArrayDataSource(bigBytes, "text/plain"));
DataHandler dh2 = new DataHandler(url2);
DataHandler dh3 = new DataHandler(url3);
DataHandler dh4 = new DataHandler(url4);
DataHandler dh5 = new DataHandler(url5);
SwAService service = new SwAService();
Dispatch<SOAPMessage> disp = service
.createDispatch(SwAService.SwAServiceHttpPort,
SOAPMessage.class,
Service.Mode.MESSAGE);
SOAPMessage msg = MessageFactory.newInstance().createMessage();
SOAPBody body = msg.getSOAPPart().getEnvelope().getBody();
body.addBodyElement(new QName("http://cxf.apache.org/swa/types",
"VoidRequest"));
AttachmentPart att = msg.createAttachmentPart(dh1);
att.setContentId("<attach1=c187f5da-fa5d-4ab9-81db-33c2bb855201@apache.org>");
msg.addAttachmentPart(att);
att = msg.createAttachmentPart(dh2);
att.setContentId("<attach2=c187f5da-fa5d-4ab9-81db-33c2bb855202@apache.org>");
msg.addAttachmentPart(att);
att = msg.createAttachmentPart(dh3);
att.setContentId("<attach3=c187f5da-fa5d-4ab9-81db-33c2bb855203@apache.org>");
msg.addAttachmentPart(att);
att = msg.createAttachmentPart(dh4);
att.setContentId("<attach4=c187f5da-fa5d-4ab9-81db-33c2bb855204@apache.org>");
msg.addAttachmentPart(att);
att = msg.createAttachmentPart(dh5);
att.setContentId("<attach5=c187f5da-fa5d-4ab9-81db-33c2bb855205@apache.org>");
msg.addAttachmentPart(att);
//Test for CXF-
msg = disp.invoke(msg);
assertEquals(5, msg.countAttachments());
}
}