SWS-352 - Full streaming WebServiceMessage/SoapMessage

Author: poutsma

core/src/main/java/org/springframework/ws/server/endpoint/adapter/method/jaxb/AbstractJaxb2PayloadMethodProcessor.java

@@ -314,6 +314,7 @@ private class JaxbStreamingPayload implements StreamingPayload {
         private JaxbStreamingPayload(Class<?> clazz, Object jaxbElement) throws JAXBException {
             JAXBContext jaxbContext = getJaxbContext(clazz);
             this.marshaller = jaxbContext.createMarshaller();
+            this.marshaller.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);
             this.jaxbElement = jaxbElement;
             JAXBIntrospector introspector = jaxbContext.createJAXBIntrospector();
             this.name = introspector.getElementName(jaxbElement);

core/src/test/java/org/springframework/ws/server/endpoint/adapter/method/jaxb/XmlRootElementPayloadMethodProcessorTest.java

@@ -139,6 +139,29 @@ public void handleReturnValueAxiom() throws Exception {
 
     }
 
+    @Test
+    public void handleReturnValueAxiomNoPayloadCaching() throws Exception {
+        AxiomSoapMessageFactory messageFactory = new AxiomSoapMessageFactory();
+        messageFactory.setPayloadCaching(false);
+        MessageContext messageContext = new DefaultMessageContext(messageFactory);
+
+        MyRootElement rootElement = new MyRootElement();
+        rootElement.setString("Foo");
+
+        processor.handleReturnValue(messageContext, rootElementReturnType, rootElement);
+        assertTrue("context has no response", messageContext.hasResponse());
+        AxiomSoapMessage response = (AxiomSoapMessage) messageContext.getResponse();
+
+        ByteArrayOutputStream bos = new ByteArrayOutputStream();
+        response.writeTo(bos);
+        String messageResult = bos.toString("UTF-8");
+
+        assertXMLEqual("<soapenv:Envelope xmlns:soapenv='http://schemas.xmlsoap.org/soap/envelope/'><soapenv:Body>" +
+                "<root xmlns='http://springframework.org'><string>Foo</string></root>" +
+                "</soapenv:Body></soapenv:Envelope>", messageResult);
+
+    }
+
     @ResponsePayload
     public MyRootElement rootElement(@RequestPayload MyRootElement rootElement) {
         return rootElement;