Java tutorial
/* * 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.transport.http; import java.io.IOException; import java.util.ArrayList; import java.util.List; import javax.mail.MessagingException; import javax.servlet.http.HttpServletResponse; import javax.xml.namespace.QName; import junit.framework.TestCase; import org.apache.axiom.om.OMAbstractFactory; import org.apache.axiom.om.OMElement; import org.apache.axiom.om.OMFactory; import org.apache.axiom.soap.SOAPBody; import org.apache.axiom.soap.SOAPEnvelope; import org.apache.axiom.soap.SOAPFactory; import org.apache.axis2.AxisFault; import org.apache.axis2.Constants; import org.apache.axis2.context.ConfigurationContext; import org.apache.axis2.context.ConfigurationContextFactory; import org.apache.axis2.context.MessageContext; import org.apache.axis2.context.NamedValue; import org.apache.axis2.description.Parameter; import org.apache.axis2.description.TransportOutDescription; import org.apache.axis2.engine.Handler.InvocationResponse; import org.apache.axis2.transport.OutTransportInfo; import org.apache.axis2.transport.TransportSender; import org.apache.axis2.transport.http.mock.MockAxisHttpResponse; import org.apache.axis2.transport.http.mock.MockHttpServletResponse; import org.apache.axis2.transport.http.mock.MockHTTPResponse; import org.apache.commons.httpclient.HttpMethod; import org.apache.commons.httpclient.methods.GetMethod; import org.apache.http.ProtocolVersion; import org.apache.http.RequestLine; import org.apache.http.message.BasicRequestLine; public abstract class CommonsHTTPTransportSenderTest extends TestCase { protected abstract TransportSender getTransportSender(); public void testInvokeWithServletBasedOutTransportInfo() throws Exception { MockHTTPResponse httpResponse = new MockHttpServletResponse(); ServletBasedOutTransportInfo info = new ServletBasedOutTransportInfo((HttpServletResponse) httpResponse); SOAPEnvelope envelope = getEnvelope(); httpResponse = configAndRun(httpResponse, info, null, getTransportSender()); assertEquals("Not the expected Header value", "application/xml", httpResponse.getHeaders().get("Content-Type")); assertEquals("Not the expected Header value", "custom-value", httpResponse.getHeaders().get("Custom-header")); assertEquals("Not the expected body content", envelope.toString().replace("utf", "UTF"), new String(httpResponse.getByteArrayOutputStream().toByteArray())); } public void testInvokeWithAxisHttpResponseImpl() throws Exception { RequestLine line = new BasicRequestLine("", "", new ProtocolVersion("http", 1, 0)); MockHTTPResponse httpResponse = new MockAxisHttpResponse(line); SOAPEnvelope envelope = getEnvelope(); httpResponse = (MockAxisHttpResponse) configAndRun(httpResponse, (OutTransportInfo) httpResponse, null, getTransportSender()); assertEquals("Not the expected Header value", "application/xml", httpResponse.getHeaders().get("Content-Type")); assertEquals("Not the expected Header value", "custom-value", httpResponse.getHeaders().get("Custom-header")); assertEquals("Not the expected body content", envelope.toString().replace("utf", "UTF"), new String(httpResponse.getByteArrayOutputStream().toByteArray())); } public void testCleanup() throws AxisFault { TransportSender sender = getTransportSender(); MessageContext msgContext = new MessageContext(); HttpMethod httpMethod = new GetMethod(); msgContext.setProperty(HTTPConstants.HTTP_METHOD, httpMethod); assertNotNull("HttpMethod can not be null", msgContext.getProperty(HTTPConstants.HTTP_METHOD)); sender.cleanup(msgContext); assertNull("HttpMethod should be null", msgContext.getProperty(HTTPConstants.HTTP_METHOD)); } public void testInit() throws AxisFault { ConfigurationContext confContext = ConfigurationContextFactory.createEmptyConfigurationContext(); TransportOutDescription transportOut = new TransportOutDescription("http"); TransportSender sender = getTransportSender(); sender.init(confContext, transportOut); } public static MockHTTPResponse configAndRun(MockHTTPResponse outResponse, OutTransportInfo outTransportInfo, String epr, TransportSender sender) throws Exception { MockHTTPResponse response = outResponse; ConfigurationContext confContext = ConfigurationContextFactory.createEmptyConfigurationContext(); TransportOutDescription transportOut = new TransportOutDescription("http"); Parameter param = new Parameter(HTTPConstants.OMIT_SOAP_12_ACTION, false); SOAPEnvelope envelope = getEnvelope(); MessageContext msgContext = new MessageContext(); transportOut.addParameter(param); // create dummy SOAPEnvelope msgContext.setEnvelope(envelope); msgContext.setProperty(MessageContext.TRANSPORT_OUT, ((MockHTTPResponse) response).getByteArrayOutputStream()); msgContext.setProperty(Constants.OUT_TRANSPORT_INFO, outTransportInfo); msgContext.setTransportOut(transportOut); msgContext.setConfigurationContext(confContext); if (epr != null) { msgContext.setProperty(Constants.Configuration.TRANSPORT_URL, epr); } // set two Headers for testing List<NamedValue> headerList = new ArrayList<NamedValue>(); NamedValue header1 = new NamedValue("Content-Type", "application/xml"); NamedValue header2 = new NamedValue("Custom-header", "custom-value"); headerList.add(header1); headerList.add(header2); msgContext.setProperty(HTTPConstants.HTTP_HEADERS, headerList); sender.init(confContext, transportOut); InvocationResponse inResponse = sender.invoke(msgContext); assertEquals("Not the expected InvocationResponse", InvocationResponse.CONTINUE, inResponse); return response; } static SOAPEnvelope getEnvelope() throws IOException, MessagingException { SOAPFactory soapFac = OMAbstractFactory.getSOAP11Factory(); OMFactory omFac = OMAbstractFactory.getOMFactory(); SOAPEnvelope enp = soapFac.createSOAPEnvelope(); SOAPBody sopaBody = soapFac.createSOAPBody(); OMElement content = omFac.createOMElement(new QName("message")); OMElement data1 = omFac.createOMElement(new QName("part")); data1.setText("sample data"); content.addChild(data1); sopaBody.addChild(content); enp.addChild(sopaBody); return enp; } }