[jbossws-commits] JBossWS SVN: r11071 - framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context.

[jbossws-commits at lists.jboss.org]

Author: richard.opalka at jboss.com
Date: 2009-10-30 02:23:57 -0400 (Fri, 30 Oct 2009)
New Revision: 11071

Modified:
   framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context/EndpointEJB.java
   framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context/EndpointJSE.java
Log:
[JBPAPP-2825] Fixing test (no changes to the stack behaviour) 

Modified: framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context/EndpointEJB.java
===================================================================
--- framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context/EndpointEJB.java	2009-10-30 03:47:24 UTC (rev 11070)
+++ framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context/EndpointEJB.java	2009-10-30 06:23:57 UTC (rev 11071)
@@ -21,7 +21,9 @@
  */
 package org.jboss.test.ws.jaxws.samples.context;
 
+import java.io.ByteArrayOutputStream;
 import java.io.IOException;
+import java.net.URI;
 import java.security.Principal;
 
 import javax.annotation.Resource;
@@ -85,7 +87,7 @@
          return "fail";
 
       // Check standard jaxws properties
-      InputSource wsdlSource = (InputSource)msgContext.get(MessageContext.WSDL_DESCRIPTION);
+      Object wsdl = msgContext.get(MessageContext.WSDL_DESCRIPTION);
       QName service = (QName)msgContext.get(MessageContext.WSDL_SERVICE);
       QName portType = (QName)msgContext.get(MessageContext.WSDL_INTERFACE);
       QName port = (QName)msgContext.get(MessageContext.WSDL_PORT);
@@ -102,8 +104,21 @@
       
       try
       {
-         Element root = DOMUtils.parse(wsdlSource);
-         new DOMWriter(System.out).setPrettyprint(true).print(root);
+         Element root = null;
+         if (wsdl instanceof InputSource)
+         {
+            root = DOMUtils.parse((InputSource)wsdl);
+         }
+         else if (wsdl instanceof URI)
+         {
+            root = DOMUtils.parse(((URI)wsdl).toURL().openStream());
+         }
+         ByteArrayOutputStream out = new ByteArrayOutputStream();
+         new DOMWriter(out).setPrettyprint(true).print(root);
+         if (!out.toString().contains("http://schemas.xmlsoap.org/wsdl/"))
+         {
+            throw new WebServiceException("Not a wsdl");
+         }
       }
       catch (IOException ex)
       {

Modified: framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context/EndpointJSE.java
===================================================================
--- framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context/EndpointJSE.java	2009-10-30 03:47:24 UTC (rev 11070)
+++ framework/branches/jbossws-framework-3.1.2.SP5/testsuite/test/java/org/jboss/test/ws/jaxws/samples/context/EndpointJSE.java	2009-10-30 06:23:57 UTC (rev 11071)
@@ -21,7 +21,9 @@
  */
 package org.jboss.test.ws.jaxws.samples.context;
 
+import java.io.ByteArrayOutputStream;
 import java.io.IOException;
+import java.net.URI;
 import java.security.Principal;
 
 import javax.annotation.Resource;
@@ -61,7 +63,7 @@
          return "fail";
 
       // Check standard jaxws properties
-      InputSource wsdlSource = (InputSource)msgContext.get(MessageContext.WSDL_DESCRIPTION);
+      Object wsdl = msgContext.get(MessageContext.WSDL_DESCRIPTION);
       QName service = (QName)msgContext.get(MessageContext.WSDL_SERVICE);
       QName portType = (QName)msgContext.get(MessageContext.WSDL_INTERFACE);
       QName port = (QName)msgContext.get(MessageContext.WSDL_PORT);
@@ -76,10 +78,24 @@
       if (!operation.equals(new QName("http://org.jboss.ws/jaxws/context", "testMessageContextProperties")))
          throw new WebServiceException("Invalid qname: " + operation);
 
+
       try
       {
-         Element root = DOMUtils.parse(wsdlSource);
-         new DOMWriter(System.out).setPrettyprint(true).print(root);
+         Element root = null;
+         if (wsdl instanceof InputSource)
+         {
+            root = DOMUtils.parse((InputSource)wsdl);
+         }
+         else if (wsdl instanceof URI)
+         {
+            root = DOMUtils.parse(((URI)wsdl).toURL().openStream());
+         }
+         ByteArrayOutputStream out = new ByteArrayOutputStream();
+         new DOMWriter(out).setPrettyprint(true).print(root);
+         if (!out.toString().contains("http://schemas.xmlsoap.org/wsdl/"))
+         {
+            throw new WebServiceException("Not a wsdl");
+         }
       }
       catch (IOException ex)
       {