/*
* 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.handler.context;
import junit.framework.TestCase;
import org.apache.axis2.jaxws.context.factory.MessageContextFactory;
import org.apache.axis2.jaxws.core.MessageContext;
import org.apache.axis2.jaxws.handler.MEPContext;
import org.apache.axis2.jaxws.handler.SoapMessageContext;
import org.apache.axis2.jaxws.message.Block;
import org.apache.axis2.jaxws.message.Message;
import org.apache.axis2.jaxws.message.Protocol;
import org.apache.axis2.jaxws.message.databinding.JAXBBlockContext;
import org.apache.axis2.jaxws.message.factory.JAXBBlockFactory;
import org.apache.axis2.jaxws.message.factory.MessageFactory;
import org.apache.axis2.jaxws.registry.FactoryRegistry;
import test.EchoString;
import test.ObjectFactory;
import javax.xml.bind.JAXBContext;
import javax.xml.soap.SOAPMessage;
public class SOAPMessageContextTests extends TestCase {
private final String INPUT_1 = "sample input";
private final String INPUT_2 = "sample input two";
private Message createMessage(String input) throws Exception {
MessageFactory factory = (MessageFactory) FactoryRegistry.getFactory(MessageFactory.class);
Message msg = factory.create(Protocol.soap11);
// Create a jaxb object
ObjectFactory objFactory = new ObjectFactory();
EchoString echo = objFactory.createEchoString();
echo.setInput(input);
// Create the necessary JAXBContext
JAXBContext jbc = JAXBContext.newInstance("test");
JAXBBlockContext blockCtx = new JAXBBlockContext(jbc);
// Create the Block
JAXBBlockFactory blockFactory = (JAXBBlockFactory) FactoryRegistry.getFactory(JAXBBlockFactory.class);
Block block = blockFactory.createFrom(echo, blockCtx, null);
msg.setBodyBlock(block);
return msg;
}
private SoapMessageContext createSampleContext() throws Exception {
Message msg = createMessage(INPUT_1);
MessageContext mc = new MessageContext();
mc.setMEPContext(new MEPContext(mc));
mc.setMessage(msg);
return MessageContextFactory.createSoapMessageContext(mc);
}
public void testGetAsSoapMessageObjectID() {
try {
SoapMessageContext smc = createSampleContext();
SOAPMessage m1 = smc.getMessage();
SOAPMessage m2 = smc.getMessage();
// not using assertEquals because I want object id equality
assertTrue("retrieval of message from SoapMessageContext twice in a row should result in same object", m1 == m2);
} catch (Exception e) {
assertNull("should not get an exception in this test", e);
}
}
public void testSoapMessageCaching() throws Exception {
Message msg1 = createMessage(INPUT_1);
Message msg2 = createMessage(INPUT_2);
MessageContext mc = new MessageContext();
mc.setMessage(msg1);
SoapMessageContext soapCtx = MessageContextFactory.createSoapMessageContext(mc);
SOAPMessage m1 = soapCtx.getMessage();
SOAPMessage m2 = soapCtx.getMessage();
assertTrue("retrieval of message from SoapMessageContext twice in a row should result in same object", m1 == m2);
// let's change underlying message as
// HandlerChainProcessor.convertToFaultMessage() does
mc.setMessage(msg2);
SOAPMessage m3 = soapCtx.getMessage();
SOAPMessage m4 = soapCtx.getMessage();
assertTrue("retrieval of message from SoapMessageContext twice in a row should result in same object", m3 == m4);
// Soap messages from before and after must be different!
assertTrue("retrieval of message from SoapMessageContext after message has changed in the underlying context", m1 != m3);
}
}