Java tutorial
/* * Copyright (c) 2010, WSO2 Inc. (http://www.wso2.org) All Rights Reserved. * * Licensed 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.wso2.carbon.automation.test.utils.axis2client; 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.util.AXIOMUtil; import org.apache.axis2.AxisFault; import org.apache.axis2.addressing.EndpointReference; import org.apache.axis2.client.Options; import org.apache.axis2.client.ServiceClient; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import javax.xml.stream.XMLStreamException; import java.io.BufferedReader; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.net.MalformedURLException; import java.net.URL; import java.nio.charset.Charset; import java.util.List; import static org.testng.Assert.assertFalse; import static org.testng.Assert.fail; public class AxisServiceClientUtils { public final static boolean TWO_WAY = true; public final static boolean ONE_WAY = false; private static final Log log = LogFactory.getLog(AxisServiceClientUtils.class); public static OMElement sendRequest(String payloadStr, EndpointReference targetEPR) throws XMLStreamException, AxisFault { OMElement payload = AXIOMUtil.stringToOM(payloadStr); Options options = new Options(); options.setTo(targetEPR); //options.setAction("urn:" + operation); //since soapAction = "" //Blocking invocation ServiceClient sender = new ServiceClient(); sender.setOptions(options); OMElement result = sender.sendReceive(payload); //log.info(result.toString()); return result; } public static void sendRequestOneWay(String payloadStr, EndpointReference targetEPR) throws XMLStreamException, AxisFault { OMElement payload = AXIOMUtil.stringToOM(payloadStr); Options options = new Options(); options.setTo(targetEPR); //options.setAction("urn:" + operation); //since soapAction = "" //Blocking invocation ServiceClient sender = new ServiceClient(); sender.setOptions(options); sender.fireAndForget(payload); } public static boolean isServiceAvailable(String serviceUrl) throws IOException { URL wsdlURL; InputStream wsdlIS; try { wsdlURL = new URL(serviceUrl + "?wsdl"); } catch (MalformedURLException e) { return false; } try { wsdlIS = wsdlURL.openStream(); } catch (IOException e) { return false;// do nothing, wait for the service } BufferedReader in = null; if (wsdlIS != null) { try { in = new BufferedReader(new InputStreamReader(wsdlIS, Charset.forName("UTF-8"))); String inputLine; while ((inputLine = in.readLine()) != null) { if (inputLine.contains("definitions")) { return true; } } } catch (IOException e) { return false; } finally { if (in != null) { in.close(); } } } return false; } public static void waitForServiceDeployment(String serviceUrl) throws IOException { int serviceTimeOut = 0; while (!isServiceAvailable(serviceUrl)) { if (serviceTimeOut == 0) { } else if (serviceTimeOut > 100) { //Check for the service for 100 seconds // if Service not available assertfalse; fail(serviceUrl + " service is not found"); break; } try { Thread.sleep(500); serviceTimeOut++; } catch (InterruptedException ignored) { } } } public static void sendRequest(String eprUrl, String operation, String payload, int numberOfInstances, List<String> expectedStrings, boolean twoWay) throws Exception { waitForServiceDeployment(eprUrl); assertFalse(!AxisServiceClientUtils.isServiceAvailable(eprUrl)); for (int i = 0; i < numberOfInstances; i++) { try { EndpointReference epr = new EndpointReference(eprUrl + "/" + operation); if (twoWay) { OMElement result = AxisServiceClientUtils.sendRequest(payload, epr); if (expectedStrings != null) { for (String expectedString : expectedStrings) { assertFalse(!result.toString().contains(expectedString)); } } } else { AxisServiceClientUtils.sendRequestOneWay(payload, epr); } } catch (XMLStreamException e) { log.error(e); throw new XMLStreamException("cannot read xml stream " + e); } catch (AxisFault axisFault) { log.error(axisFault.getMessage()); throw new AxisFault("cannot read xml stream " + axisFault.getMessage()); } } } public static void sendRequest(String eprUrl, String operation, String payload, int numberOfInstances, String expectedException, boolean twoWay) throws XMLStreamException, IOException { assertFalse(!AxisServiceClientUtils.isServiceAvailable(eprUrl)); for (int i = 0; i < numberOfInstances; i++) { try { EndpointReference epr = new EndpointReference(operation); if (twoWay) { OMElement result = AxisServiceClientUtils.sendRequest(payload, epr); fail("Exception expected!!! : " + result.toString()); } else { AxisServiceClientUtils.sendRequestOneWay(payload, epr); } } catch (XMLStreamException e) { if (!e.getClass().getSimpleName().equals(expectedException)) { throw new XMLStreamException(e); } } catch (AxisFault axisFault) { if (!axisFault.getClass().getSimpleName().equals(expectedException)) { throw new AxisFault(axisFault.getMessage()); } } if (expectedException != null) { fail("Exception expected. But not found!!"); } } } public static void waitForServiceUnDeployment(String serviceUrl) throws Exception { int serviceTimeOut = 0; while (isServiceAvailable(serviceUrl)) { if (serviceTimeOut == 0) { } else if (serviceTimeOut > 60) { //Check for the service for 100 seconds throw new Exception("Service undeployment fail"); } try { Thread.sleep(500); serviceTimeOut++; } catch (InterruptedException ignored) { } } } private static OMElement createPayLoad() { OMFactory fac = OMAbstractFactory.getOMFactory(); OMNamespace omNs = fac.createOMNamespace("http://test.com", "test"); OMElement method = fac.createOMElement("add", omNs); OMElement valueOfa = fac.createOMElement("a", omNs); OMElement valueOfb = fac.createOMElement("b", omNs); valueOfa.addChild(fac.createOMText(valueOfa, "200")); valueOfb.addChild(fac.createOMText(valueOfb, "220")); method.addChild(valueOfa); method.addChild(valueOfb); return method; } }