[jbossws-commits] JBossWS SVN: r12970 - framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider.

[jbossws-commits at lists.jboss.org]

Author: alessio.soldano at jboss.com
Date: 2010-09-14 13:08:51 -0400 (Tue, 14 Sep 2010)
New Revision: 12970

Modified:
   framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider/ProviderBeanMessage.java
   framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider/ProviderMessageTestCase.java
Log:
[JBWS-3101] Adding testcase


Modified: framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider/ProviderBeanMessage.java
===================================================================
--- framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider/ProviderBeanMessage.java	2010-09-14 11:45:27 UTC (rev 12969)
+++ framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider/ProviderBeanMessage.java	2010-09-14 17:08:51 UTC (rev 12970)
@@ -21,12 +21,15 @@
  */
 package org.jboss.test.ws.jaxws.samples.provider;
 
+import javax.xml.soap.SOAPHeader;
 import javax.xml.soap.SOAPMessage;
 import javax.xml.ws.Provider;
 import javax.xml.ws.Service;
 import javax.xml.ws.ServiceMode;
 import javax.xml.ws.WebServiceProvider;
 
+import org.w3c.dom.NodeList;
+
 /**
  * Test a Provider<SOAPMessage>
  *
@@ -43,6 +46,22 @@
 {
    public SOAPMessage invoke(SOAPMessage request)
    {
+      try
+      {
+         SOAPHeader headers = request.getSOAPHeader();
+         if (headers != null)
+         {
+            NodeList nl = headers.getElementsByTagNameNS("http://org.jboss.ws/foo", "returnNullResponse");
+            if (nl != null && nl.getLength() > 0)
+            {
+               return null;
+            }
+         }
+      }
+      catch (Exception e)
+      {
+         throw new RuntimeException(e);
+      }
       return request;
    }
 }

Modified: framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider/ProviderMessageTestCase.java
===================================================================
--- framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider/ProviderMessageTestCase.java	2010-09-14 11:45:27 UTC (rev 12969)
+++ framework/trunk/testsuite/test/java/org/jboss/test/ws/jaxws/samples/provider/ProviderMessageTestCase.java	2010-09-14 17:08:51 UTC (rev 12970)
@@ -29,12 +29,14 @@
 import javax.xml.bind.JAXBException;
 import javax.xml.namespace.QName;
 import javax.xml.soap.MessageFactory;
+import javax.xml.soap.SOAPBody;
 import javax.xml.soap.SOAPConnection;
 import javax.xml.soap.SOAPConnectionFactory;
 import javax.xml.soap.SOAPEnvelope;
 import javax.xml.soap.SOAPException;
 import javax.xml.soap.SOAPHeader;
 import javax.xml.soap.SOAPMessage;
+import javax.xml.soap.SOAPPart;
 import javax.xml.ws.Dispatch;
 import javax.xml.ws.Service;
 import javax.xml.ws.Service.Mode;
@@ -61,6 +63,16 @@
       "    <ns1:somePayload xmlns:ns1='http://org.jboss.ws/provider'/>" +
       "  </soap:Body>" +
       "</soap:Envelope>";
+   
+   private String msgStringForNullResponse =
+      "<soap:Envelope xmlns:soap='http://schemas.xmlsoap.org/soap/envelope/'>" +
+      "  <soap:Header>" +
+      "    <nsh:returnNullResponse xmlns:nsh='http://org.jboss.ws/foo'/>" +
+      "  </soap:Header>" +
+      "  <soap:Body>" +
+      "    <ns1:somePayload xmlns:ns1='http://org.jboss.ws/provider'/>" +
+      "  </soap:Body>" +
+      "</soap:Envelope>";
 
    public static Test suite()
    {
@@ -106,6 +118,23 @@
       assertEquals(reqEnv, resEnv);
    }
 
+   public void testProviderMessageNullResponse() throws Exception
+   {
+      MessageFactory msgFactory = MessageFactory.newInstance();
+      SOAPMessage reqMsg = msgFactory.createMessage(null, new ByteArrayInputStream(msgStringForNullResponse.getBytes()));
+
+      URL epURL = new URL("http://" + getServerHost() + ":8080/jaxws-samples-provider-message");
+      SOAPConnection con = SOAPConnectionFactory.newInstance().createConnection();
+      SOAPMessage resMsg = con.call(reqMsg, epURL);
+      SOAPPart soapPart = resMsg.getSOAPPart();
+      //verify there's either nothing in the reply or at least the response body is empty
+      if (soapPart != null && soapPart.getEnvelope() != null && soapPart.getEnvelope().getBody() != null)
+      {
+         SOAPBody soapBody = soapPart.getEnvelope().getBody();
+         assertFalse(soapBody.getChildElements().hasNext());
+      }
+   }
+   
    private SOAPMessage getRequestMessage() throws SOAPException, IOException
    {
       MessageFactory msgFactory = MessageFactory.newInstance();