/* | |
* 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.xmlhttp.clientTests.dispatch.datasource; | |
import junit.framework.Test; | |
import junit.framework.TestSuite; | |
import org.apache.axis2.jaxws.framework.AbstractTestCase; | |
import org.apache.axis2.jaxws.provider.DataSourceImpl; | |
import org.apache.axiom.attachments.utils.IOUtils; | |
import org.apache.axiom.attachments.impl.BufferUtils; | |
import org.apache.axiom.om.util.UUIDGenerator; | |
import javax.activation.DataSource; | |
import javax.activation.FileDataSource; | |
import javax.activation.DataHandler; | |
import javax.imageio.ImageIO; | |
import javax.imageio.stream.FileImageInputStream; | |
import javax.imageio.stream.ImageInputStream; | |
import javax.xml.namespace.QName; | |
import javax.xml.ws.Dispatch; | |
import javax.xml.ws.Service; | |
import javax.xml.ws.handler.MessageContext; | |
import javax.xml.ws.http.HTTPBinding; | |
import java.awt.*; | |
import java.io.File; | |
import java.io.ByteArrayOutputStream; | |
import java.io.InputStream; | |
import java.io.IOException; | |
import java.util.Arrays; | |
import java.util.Map; | |
import java.util.HashMap; | |
public class DispatchXMessageDataSourceTests extends AbstractTestCase { | |
public String HOSTPORT = "http://localhost:6060"; | |
private String ENDPOINT_URL = HOSTPORT + "/axis2/services/XMessageDataSourceProvider.XMessageDataSourceProviderPort"; | |
private QName SERVICE_NAME = new QName("http://ws.apache.org/axis2", "XMessageDataSourceProvider"); | |
private QName PORT_NAME = new QName("http://ws.apache.org/axis2", "XMessageDataSourceProviderPort"); | |
private DataSource imageDS; | |
private FileDataSource txtDS; | |
private DataSource attachmentDS; | |
public static Test suite() { | |
return getTestSetup(new TestSuite(DispatchXMessageDataSourceTests.class)); | |
} | |
public void setUp() throws Exception { | |
String imageResourceDir = System.getProperty("basedir",".")+"/"+"test-resources"+File.separator+"image"; | |
File file = new File(imageResourceDir+File.separator+"test.jpg"); | |
ImageInputStream fiis = new FileImageInputStream(file); | |
Image image = ImageIO.read(fiis); | |
imageDS = new DataSourceImpl("image/jpeg","test.jpg",image); | |
String textResourceDir = System.getProperty("basedir",".")+"/"+"test/org/apache/axis2/jaxws/xmlhttp"; | |
File file2 = new File(textResourceDir+File.separator+"README.txt"); | |
txtDS = new FileDataSource(file2); | |
String resourceDir = System.getProperty("basedir",".")+"/"+"test-resources"; | |
File file3 = new File(resourceDir+File.separator+"log4j.properties"); | |
attachmentDS = new FileDataSource(file3); | |
} | |
public Dispatch<DataSource> getDispatch() { | |
Service service = Service.create(SERVICE_NAME); | |
service.addPort(PORT_NAME, HTTPBinding.HTTP_BINDING,ENDPOINT_URL); | |
Dispatch<DataSource> dispatch = service.createDispatch(PORT_NAME, DataSource.class, Service.Mode.MESSAGE); | |
return dispatch; | |
} | |
public void testDataSourceWithTXT() throws Exception { | |
Dispatch<DataSource> dispatch = getDispatch(); | |
DataSource request = txtDS; | |
DataSource response = dispatch.invoke(request); | |
assertTrue(response != null); | |
assertEquals(response.getContentType(),"text/plain"); | |
String req = new String(getStreamAsByteArray(request.getInputStream())); | |
String res = new String(getStreamAsByteArray(response.getInputStream())); | |
assertEquals(req, res); | |
} | |
public void testDataSourceWithImage() throws Exception { | |
Dispatch<DataSource> dispatch = getDispatch(); | |
DataSource request = imageDS; | |
DataSource response = dispatch.invoke(request); | |
assertTrue(response != null); | |
assertEquals(response.getContentType(),"image/jpeg"); | |
assertTrue(Arrays.equals(getStreamAsByteArray(request.getInputStream()), | |
getStreamAsByteArray(response.getInputStream()))); | |
} | |
public void testDataSourceWithTXTPlusAttachment() throws Exception { | |
Dispatch<DataSource> dispatch = getDispatch(); | |
Map attachments = new HashMap(); | |
Map requestContext = dispatch.getRequestContext(); | |
// requestContext.put(org.apache.axis2.transport.http.HTTPConstants.SO_TIMEOUT , new | |
// Integer(999999)); | |
// requestContext.put(org.apache.axis2.transport.http.HTTPConstants.CONNECTION_TIMEOUT, new | |
// Integer(999999)); | |
requestContext.put(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS, | |
attachments); | |
attachments.put(UUIDGenerator.getUUID(), new DataHandler(attachmentDS)); | |
DataSource request = txtDS; | |
DataSource response = dispatch.invoke(request); | |
assertTrue(response != null); | |
assertEquals(response.getContentType(),"text/plain"); | |
String req = new String(getStreamAsByteArray(request.getInputStream())); | |
String res = new String(getStreamAsByteArray(response.getInputStream())); | |
assertEquals(req, res); | |
Map attachments2 = (Map) dispatch.getResponseContext().get(MessageContext.INBOUND_MESSAGE_ATTACHMENTS); | |
assertTrue(attachments2 != null); | |
assertEquals(attachments2.size(), 1); | |
} | |
public void testDataSourceWithImagePlusAttachment() throws Exception { | |
Dispatch<DataSource> dispatch = getDispatch(); | |
Map attachments = new HashMap(); | |
Map requestContext = dispatch.getRequestContext(); | |
requestContext.put(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS, | |
attachments); | |
attachments.put(UUIDGenerator.getUUID(), new DataHandler(attachmentDS)); | |
DataSource request = imageDS; | |
DataSource response = dispatch.invoke(request); | |
assertTrue(response != null); | |
assertEquals(response.getContentType(),"image/jpeg"); | |
assertTrue(Arrays.equals(getStreamAsByteArray(request.getInputStream()), | |
getStreamAsByteArray(response.getInputStream()))); | |
Map attachments2 = (Map) dispatch.getResponseContext().get(MessageContext.INBOUND_MESSAGE_ATTACHMENTS); | |
assertTrue(attachments2 != null); | |
assertEquals(attachments2.size(), 1); | |
} | |
public void testDataSourceWithTXTPlusTwoAttachments() throws Exception { | |
Dispatch<DataSource> dispatch = getDispatch(); | |
Map attachments = new HashMap(); | |
Map requestContext = dispatch.getRequestContext(); | |
requestContext.put(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS, | |
attachments); | |
attachments.put(UUIDGenerator.getUUID(), new DataHandler(attachmentDS)); | |
attachments.put(UUIDGenerator.getUUID(), new DataHandler(imageDS)); | |
DataSource request = txtDS; | |
DataSource response = dispatch.invoke(request); | |
assertTrue(response != null); | |
assertEquals(response.getContentType(),"text/plain"); | |
String req = new String(getStreamAsByteArray(request.getInputStream())); | |
String res = new String(getStreamAsByteArray(response.getInputStream())); | |
assertEquals(req, res); | |
Map attachments2 = (Map) dispatch.getResponseContext().get(MessageContext.INBOUND_MESSAGE_ATTACHMENTS); | |
assertTrue(attachments2 != null); | |
assertEquals(attachments2.size(), 2); | |
} | |
private byte[] getStreamAsByteArray(InputStream is) throws IOException { | |
ByteArrayOutputStream baos = new ByteArrayOutputStream(); | |
BufferUtils.inputStream2OutputStream(is, baos); | |
baos.flush(); | |
return baos.toByteArray(); | |
} | |
} |