| /* |
| * 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.rpc; |
| |
| import org.apache.axiom.om.OMAbstractFactory; |
| import org.apache.axiom.om.OMElement; |
| import org.apache.axiom.om.OMFactory; |
| import org.apache.axiom.om.OMNamespace; |
| import org.apache.axiom.om.impl.builder.StAXOMBuilder; |
| import org.apache.axiom.om.util.StAXUtils; |
| import org.apache.axiom.soap.SOAPEnvelope; |
| import org.apache.axiom.soap.SOAPFactory; |
| import org.apache.axis2.AxisFault; |
| import org.apache.axis2.client.OperationClient; |
| import org.apache.axis2.client.ServiceClient; |
| import org.apache.axis2.context.MessageContext; |
| import org.apache.axis2.databinding.utils.BeanUtil; |
| import org.apache.axis2.engine.DefaultObjectSupplier; |
| import org.apache.axis2.integration.RPCLocalTestCase; |
| import org.apache.axis2.rpc.client.RPCServiceClient; |
| import org.apache.axis2.wsdl.WSDLConstants; |
| |
| import javax.xml.stream.FactoryConfigurationError; |
| import javax.xml.stream.XMLStreamException; |
| import javax.xml.stream.XMLStreamReader; |
| import java.io.ByteArrayInputStream; |
| import java.util.ArrayList; |
| |
| public class MultirefTest extends RPCLocalTestCase { |
| |
| protected boolean finish = false; |
| public static final String NAMESPACE = "http://rpc.axis2.apache.org"; |
| |
| protected void setUp() throws Exception { |
| super.setUp(); |
| deployClassAsService("EchoXMLService", RPCServiceClass.class); |
| } |
| |
| public void testMulitref1() throws AxisFault { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "echoString"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("echoString", omNs); |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| OMElement ref = fac.createOMElement("reference", null); |
| ref.addAttribute(fac.createOMAttribute("id", null, "1")); |
| ref.setText("hello Axis2"); |
| envelope.getBody().addChild(ref); |
| |
| MessageContext reqMessageContext = new MessageContext(); |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| MessageContext responseMessageContx = |
| opClinet.getMessageContext(WSDLConstants.MESSAGE_LABEL_IN_VALUE); |
| |
| SOAPEnvelope env = responseMessageContx.getEnvelope(); |
| |
| |
| assertEquals(env.getBody().getFirstElement().getFirstElement().getText(), "hello Axis2"); |
| } |
| |
| public void testadd() throws AxisFault { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "add"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("add", omNs); |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| |
| OMElement value2 = fac.createOMElement("arg1", null); |
| value2.addAttribute(fac.createOMAttribute("href", null, "#2")); |
| method.addChild(value2); |
| |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| OMElement ref = fac.createOMElement("reference", null); |
| ref.addAttribute(fac.createOMAttribute("id", null, "1")); |
| ref.setText("10"); |
| envelope.getBody().addChild(ref); |
| |
| OMElement ref2 = fac.createOMElement("reference", null); |
| ref2.addAttribute(fac.createOMAttribute("id", null, "2")); |
| ref2.setText("10"); |
| envelope.getBody().addChild(ref2); |
| |
| MessageContext reqMessageContext = new MessageContext(); |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| MessageContext responseMessageContx = |
| opClinet.getMessageContext(WSDLConstants.MESSAGE_LABEL_IN_VALUE); |
| |
| SOAPEnvelope env = responseMessageContx.getEnvelope(); |
| |
| |
| assertEquals(env.getBody().getFirstElement().getFirstElement().getText(), "20"); |
| } |
| |
| public void testaddSameRef() throws AxisFault { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "add"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("add", omNs); |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| |
| OMElement value2 = fac.createOMElement("arg1", null); |
| value2.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value2); |
| |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| OMElement ref = fac.createOMElement("reference", null); |
| ref.addAttribute(fac.createOMAttribute("id", null, "1")); |
| ref.setText("10"); |
| envelope.getBody().addChild(ref); |
| |
| MessageContext reqMessageContext = new MessageContext(); |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| MessageContext responseMessageContx = |
| opClinet.getMessageContext(WSDLConstants.MESSAGE_LABEL_IN_VALUE); |
| SOAPEnvelope env = responseMessageContx.getEnvelope(); |
| |
| assertEquals(env.getBody().getFirstElement().getFirstElement().getText(), "20"); |
| } |
| |
| public void testaddError() { |
| try { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "add"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("add", omNs); |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| |
| OMElement value2 = fac.createOMElement("arg1", null); |
| value2.addAttribute(fac.createOMAttribute("href", null, "#2")); |
| method.addChild(value2); |
| |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| OMElement ref = fac.createOMElement("reference", null); |
| ref.addAttribute(fac.createOMAttribute("id", null, "1")); |
| ref.setText("10"); |
| envelope.getBody().addChild(ref); |
| |
| OMElement ref2 = fac.createOMElement("reference", null); |
| ref2.addAttribute(fac.createOMAttribute("id", null, "3")); |
| ref2.setText("10"); |
| envelope.getBody().addChild(ref2); |
| |
| MessageContext reqMessageContext = new MessageContext();; |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| fail("This should fail with : " + "org.apache.axis2.AxisFault: Invalid reference :2"); |
| } catch (AxisFault axisFault) { |
| String val = axisFault.getFaultDetailElement().toString(); |
| System.out.println("val = " + val); |
| int index = val.indexOf("org.apache.axis2.AxisFault: Invalid reference :2"); |
| if (index < 0) { |
| fail("This should fail with : " + |
| "org.apache.axis2.AxisFault: Invalid reference :2"); |
| } |
| } |
| } |
| |
| |
| public void testMulitrefBean() throws Exception { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "editBean"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("editBean", omNs); |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| OMElement value2 = fac.createOMElement("arg1", null); |
| value2.setText("159"); |
| method.addChild(value2); |
| |
| |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| |
| String ref1 = |
| "<reference id=\"1\"><name>Deepal</name><value href=\"#2\"/><address href=\"#3\"/></reference>"; |
| OMElement om1 = getOMElement(ref1, fac); |
| envelope.getBody().addChild(om1); |
| String ref2 = "<reference id=\"2\">false</reference>"; |
| OMElement om2 = getOMElement(ref2, fac); |
| envelope.getBody().addChild(om2); |
| String ref3 = "<reference id=\"3\"><town href=\"#4\"/><number>1010</number></reference>"; |
| OMElement om3 = getOMElement(ref3, fac); |
| envelope.getBody().addChild(om3); |
| String ref4 = "<reference id=\"4\">Colombo3</reference>"; |
| OMElement om4 = getOMElement(ref4, fac); |
| envelope.getBody().addChild(om4); |
| |
| MessageContext reqMessageContext = new MessageContext(); |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| MessageContext responseMessageContx = |
| opClinet.getMessageContext(WSDLConstants.MESSAGE_LABEL_IN_VALUE); |
| |
| SOAPEnvelope env = responseMessageContx.getEnvelope(); |
| |
| OMElement response = env.getBody().getFirstElement(); |
| MyBean resBean = (MyBean)BeanUtil.deserialize(MyBean.class, response.getFirstElement(), |
| new DefaultObjectSupplier(), null); |
| assertNotNull(resBean); |
| assertEquals(resBean.getAge(), 159); |
| } |
| |
| |
| public void testbeanOM() throws Exception { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "beanOM"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("beanOM", omNs); |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| OMElement value2 = fac.createOMElement("arg1", null); |
| value2.setText("159"); |
| method.addChild(value2); |
| |
| |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| |
| String ref1 = |
| "<reference id=\"1\"><name>Deepal</name><value href=\"#2\"/><address href=\"#3\"/></reference>"; |
| OMElement om1 = getOMElement(ref1, fac); |
| envelope.getBody().addChild(om1); |
| String ref2 = "<reference id=\"2\">false</reference>"; |
| OMElement om2 = getOMElement(ref2, fac); |
| envelope.getBody().addChild(om2); |
| String ref3 = "<reference id=\"3\"><town href=\"#4\"/><number>1010</number></reference>"; |
| OMElement om3 = getOMElement(ref3, fac); |
| envelope.getBody().addChild(om3); |
| String ref4 = "<reference id=\"4\">Colombo3</reference>"; |
| OMElement om4 = getOMElement(ref4, fac); |
| envelope.getBody().addChild(om4); |
| |
| MessageContext reqMessageContext = new MessageContext();; |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| MessageContext responseMessageContx = |
| opClinet.getMessageContext(WSDLConstants.MESSAGE_LABEL_IN_VALUE); |
| |
| SOAPEnvelope env = responseMessageContx.getEnvelope(); |
| |
| OMElement response = env.getBody().getFirstElement(); |
| MyBean resBean = (MyBean)BeanUtil.deserialize(MyBean.class, response.getFirstElement(), |
| new DefaultObjectSupplier(), null); |
| assertNotNull(resBean); |
| assertEquals(resBean.getAge(), 159); |
| } |
| |
| |
| public void testomrefs() throws Exception { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "omrefs"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("omrefs", omNs); |
| |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| |
| OMElement value2 = fac.createOMElement("arg1", null); |
| value2.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value2); |
| |
| |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| |
| String ref1 = |
| "<reference id=\"1\"><name>Deepal</name><value href=\"#2\"/><address href=\"#3\"/></reference>"; |
| OMElement om1 = getOMElement(ref1, fac); |
| envelope.getBody().addChild(om1); |
| String ref2 = "<reference id=\"2\">false</reference>"; |
| OMElement om2 = getOMElement(ref2, fac); |
| envelope.getBody().addChild(om2); |
| String ref3 = "<reference id=\"3\"><town href=\"#4\"/><number>1010</number></reference>"; |
| OMElement om3 = getOMElement(ref3, fac); |
| envelope.getBody().addChild(om3); |
| String ref4 = "<reference id=\"4\">Colombo3</reference>"; |
| OMElement om4 = getOMElement(ref4, fac); |
| envelope.getBody().addChild(om4); |
| |
| MessageContext reqMessageContext = new MessageContext();; |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| MessageContext responseMessageContx = |
| opClinet.getMessageContext(WSDLConstants.MESSAGE_LABEL_IN_VALUE); |
| |
| SOAPEnvelope env = responseMessageContx.getEnvelope(); |
| |
| OMElement response = env.getBody().getFirstElement(); |
| |
| ArrayList args = new ArrayList(); |
| args.add(boolean.class); |
| |
| Object [] resBean = |
| BeanUtil.deserialize(response, args.toArray(), new DefaultObjectSupplier()); |
| assertNotNull(resBean); |
| assertEquals(((Boolean)resBean[0]).booleanValue(), true); |
| } |
| |
| private OMElement getOMElement(String str, OMFactory fac) throws Exception { |
| StAXOMBuilder staxOMBuilder; |
| XMLStreamReader xmlReader = StAXUtils.createXMLStreamReader(new |
| ByteArrayInputStream(str.getBytes())); |
| staxOMBuilder = new StAXOMBuilder(fac, xmlReader); |
| return staxOMBuilder.getDocumentElement(); |
| } |
| |
| |
| public void testechoEmployee() throws Exception { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "echoEmployee"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("echoEmployee", omNs); |
| |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| |
| |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| |
| String str = "<reference id=\"1\">\n" + |
| " <name>John</name>\n" + |
| " <age>50</age>\n" + |
| " <emplyer href=\"#1\"/>\n" + |
| " <address href=\"#2\"/>\n" + |
| "</reference>"; |
| envelope.getBody().addChild(getOMElement(str, fac)); |
| str = "<reference id=\"2\">\n" + |
| "<town>Colombo3</town><number>1010</number>\n" + |
| "</reference>"; |
| envelope.getBody().addChild(getOMElement(str, fac)); |
| |
| MessageContext reqMessageContext = new MessageContext(); |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| MessageContext responseMessageContx = |
| opClinet.getMessageContext(WSDLConstants.MESSAGE_LABEL_IN_VALUE); |
| |
| SOAPEnvelope env = responseMessageContx.getEnvelope(); |
| |
| Employee emp = (Employee)BeanUtil.deserialize(Employee.class, env.getBody() |
| .getFirstElement().getFirstElement(), new DefaultObjectSupplier(), null); |
| assertNotNull(emp); |
| } |
| |
| |
| public void testMulitrefArray() throws AxisFault { |
| RPCServiceClient sender = getRPCClient("EchoXMLService", "handleArrayList"); |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(NAMESPACE, "my"); |
| OMElement method = fac.createOMElement("handleArrayList", omNs); |
| |
| OMElement value = fac.createOMElement("arg0", null); |
| value.addAttribute(fac.createOMAttribute("href", null, "#1")); |
| method.addChild(value); |
| |
| OMElement value2 = fac.createOMElement("arg1", null); |
| value2.setText("10"); |
| method.addChild(value2); |
| |
| |
| SOAPFactory factory = OMAbstractFactory.getSOAP11Factory(); |
| SOAPEnvelope envelope = factory.getDefaultEnvelope(); |
| envelope.getBody().addChild(method); |
| |
| |
| String str = "<reference id=\"1\">\n" + |
| " <item0>abc</item0>\n" + |
| " <item0>def</item0>\n" + |
| " <item0>ghi</item0>\n" + |
| " <item0>klm</item0>\n" + |
| "</reference>"; |
| StAXOMBuilder staxOMBuilder; |
| try { |
| XMLStreamReader xmlReader = StAXUtils.createXMLStreamReader(new |
| ByteArrayInputStream(str.getBytes())); |
| staxOMBuilder = new |
| StAXOMBuilder(fac, xmlReader); |
| } catch (XMLStreamException e) { |
| throw AxisFault.makeFault(e); |
| } catch (FactoryConfigurationError factoryConfigurationError) { |
| throw AxisFault.makeFault(factoryConfigurationError); |
| } |
| envelope.getBody().addChild(staxOMBuilder.getDocumentElement()); |
| |
| MessageContext reqMessageContext = new MessageContext(); |
| OperationClient opClinet = sender.createClient(ServiceClient.ANON_OUT_IN_OP); |
| |
| reqMessageContext.setEnvelope(envelope); |
| |
| opClinet.addMessageContext(reqMessageContext); |
| opClinet.execute(true); |
| |
| MessageContext responseMessageContx = |
| opClinet.getMessageContext(WSDLConstants.MESSAGE_LABEL_IN_VALUE); |
| |
| SOAPEnvelope env = responseMessageContx.getEnvelope(); |
| assertEquals(env.getBody().getFirstElement().getFirstElement().getText(), "abcdefghiklm10"); |
| } |
| |
| } |
