1   /*
2    * Copyright 2007 the original author or authors.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *      http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  
17  package org.springframework.ws.server.endpoint.adapter;
18  
19  import java.util.Properties;
20  import javax.xml.parsers.DocumentBuilder;
21  import javax.xml.parsers.DocumentBuilderFactory;
22  import javax.xml.transform.Source;
23  import javax.xml.transform.dom.DOMSource;
24  
25  import junit.framework.TestCase;
26  import static org.easymock.EasyMock.*;
27  import org.springframework.ws.WebServiceMessage;
28  import org.springframework.ws.WebServiceMessageFactory;
29  import org.springframework.ws.context.DefaultMessageContext;
30  import org.springframework.ws.context.MessageContext;
31  import org.springframework.ws.server.endpoint.MethodEndpoint;
32  import org.springframework.ws.server.endpoint.annotation.XPathParam;
33  import org.springframework.xml.transform.StringResult;
34  import org.springframework.xml.transform.StringSource;
35  import org.w3c.dom.Document;
36  import org.w3c.dom.Element;
37  import org.w3c.dom.Node;
38  import org.w3c.dom.NodeList;
39  import org.w3c.dom.Text;
40  
41  public class XPathParamAnnotationMethodEndpointAdapterTest extends TestCase {
42  
43      private static final String CONTENTS = "<root><child><text>text</text><number>42.0</number></child></root>";
44  
45      private XPathParamAnnotationMethodEndpointAdapter adapter;
46  
47      private boolean supportedTypesInvoked = false;
48  
49      private boolean supportedSourceInvoked;
50  
51      private boolean namespacesInvoked;
52  
53      protected void setUp() throws Exception {
54          adapter = new XPathParamAnnotationMethodEndpointAdapter();
55          adapter.afterPropertiesSet();
56      }
57  
58      public void testUnsupportedInvalidParam() throws NoSuchMethodException {
59          MethodEndpoint endpoint = new MethodEndpoint(this, "unsupportedInvalidParamType", new Class[]{Integer.TYPE});
60          assertFalse("Method supported", adapter.supports(endpoint));
61      }
62  
63      public void testUnsupportedInvalidReturnType() throws NoSuchMethodException {
64          MethodEndpoint endpoint = new MethodEndpoint(this, "unsupportedInvalidReturnType", new Class[]{String.class});
65          assertFalse("Method supported", adapter.supports(endpoint));
66      }
67  
68      public void testUnsupportedInvalidParams() throws NoSuchMethodException {
69          MethodEndpoint endpoint =
70                  new MethodEndpoint(this, "unsupportedInvalidParams", new Class[]{String.class, String.class});
71          assertFalse("Method supported", adapter.supports(endpoint));
72      }
73  
74      public void testSupportedTypes() throws NoSuchMethodException {
75          MethodEndpoint endpoint = new MethodEndpoint(this, "supportedTypes",
76                  new Class[]{Boolean.TYPE, Double.TYPE, Node.class, NodeList.class, String.class});
77          assertTrue("Not all types supported", adapter.supports(endpoint));
78      }
79  
80      public void testSupportsStringSource() throws NoSuchMethodException {
81          MethodEndpoint endpoint = new MethodEndpoint(this, "supportedStringSource", new Class[]{String.class});
82          assertTrue("StringSource method not supported", adapter.supports(endpoint));
83      }
84  
85      public void testSupportsSource() throws NoSuchMethodException {
86          MethodEndpoint endpoint = new MethodEndpoint(this, "supportedSource", new Class[]{String.class});
87          assertTrue("Source method not supported", adapter.supports(endpoint));
88      }
89  
90      public void testSupportsVoid() throws NoSuchMethodException {
91          MethodEndpoint endpoint = new MethodEndpoint(this, "supportedVoid", new Class[]{String.class});
92          assertTrue("void method not supported", adapter.supports(endpoint));
93      }
94  
95      public void testInvokeTypes() throws Exception {
96          WebServiceMessage messageMock = createMock(WebServiceMessage.class);
97          expect(messageMock.getPayloadSource()).andReturn(new StringSource(CONTENTS));
98          WebServiceMessageFactory factoryMock = createMock(WebServiceMessageFactory.class);
99          replay(messageMock, factoryMock);
100 
101         MessageContext messageContext = new DefaultMessageContext(messageMock, factoryMock);
102         MethodEndpoint endpoint = new MethodEndpoint(this, "supportedTypes",
103                 new Class[]{Boolean.TYPE, Double.TYPE, Node.class, NodeList.class, String.class});
104         adapter.invoke(messageContext, endpoint);
105         assertTrue("Method not invoked", supportedTypesInvoked);
106 
107         verify(messageMock, factoryMock);
108     }
109 
110     public void testInvokeSource() throws Exception {
111         WebServiceMessage requestMock = createMock(WebServiceMessage.class);
112         WebServiceMessage responseMock = createMock(WebServiceMessage.class);
113         expect(requestMock.getPayloadSource()).andReturn(new StringSource(CONTENTS));
114         expect(responseMock.getPayloadResult()).andReturn(new StringResult());
115         WebServiceMessageFactory factoryMock = createMock(WebServiceMessageFactory.class);
116         expect(factoryMock.createWebServiceMessage()).andReturn(responseMock);
117         replay(requestMock, responseMock, factoryMock);
118 
119         MessageContext messageContext = new DefaultMessageContext(requestMock, factoryMock);
120         MethodEndpoint endpoint = new MethodEndpoint(this, "supportedSource", new Class[]{String.class});
121         adapter.invoke(messageContext, endpoint);
122         assertTrue("Method not invoked", supportedSourceInvoked);
123 
124         verify(requestMock, responseMock, factoryMock);
125     }
126 
127     public void testInvokeVoidDom() throws Exception {
128         DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();
129         DocumentBuilder documentBuilder = documentBuilderFactory.newDocumentBuilder();
130         Document document = documentBuilder.newDocument();
131         String rootNamespace = "http://rootnamespace";
132         Element rootElement = document.createElementNS(rootNamespace, "root");
133         document.appendChild(rootElement);
134         String childNamespace = "http://childnamespace";
135         Element first = document.createElementNS(childNamespace, "child");
136         rootElement.appendChild(first);
137         Text text = document.createTextNode("value");
138         first.appendChild(text);
139         Element second = document.createElementNS(rootNamespace, "other-child");
140         rootElement.appendChild(second);
141         text = document.createTextNode("other-value");
142         second.appendChild(text);
143 
144         WebServiceMessage requestMock = createMock(WebServiceMessage.class);
145         expect(requestMock.getPayloadSource()).andReturn(new DOMSource(first));
146         WebServiceMessageFactory factoryMock = createMock(WebServiceMessageFactory.class);
147 
148         replay(requestMock, factoryMock);
149 
150         Properties namespaces = new Properties();
151         namespaces.setProperty("root", rootNamespace);
152         namespaces.setProperty("child", childNamespace);
153         adapter.setNamespaces(namespaces);
154 
155         MessageContext messageContext = new DefaultMessageContext(requestMock, factoryMock);
156         MethodEndpoint endpoint = new MethodEndpoint(this, "namespaces", new Class[]{Node.class});
157         adapter.invoke(messageContext, endpoint);
158         assertTrue("Method not invoked", namespacesInvoked);
159     }
160 
161     public void supportedVoid(@XPathParam("/")String param1) {
162     }
163 
164     public Source supportedSource(@XPathParam("/")String param1) {
165         supportedSourceInvoked = true;
166         return new StringSource("<response/>");
167     }
168 
169     public StringSource supportedStringSource(@XPathParam("/")String param1) {
170         return null;
171     }
172 
173     public void supportedTypes(@XPathParam("/root/child")boolean param1,
174                                @XPathParam("/root/child/number")double param2,
175                                @XPathParam("/root/child")Node param3,
176                                @XPathParam("/root/*")NodeList param4,
177                                @XPathParam("/root/child/text")String param5) {
178         supportedTypesInvoked = true;
179         assertTrue("Invalid boolean value", param1);
180         assertEquals("Invalid double value", 42D, param2, 0.00001D);
181         assertEquals("Invalid Node value", "child", param3.getLocalName());
182         assertEquals("Invalid NodeList value", 1, param4.getLength());
183         assertEquals("Invalid Node value", "child", param4.item(0).getLocalName());
184         assertEquals("Invalid Node value", "text", param5);
185     }
186 
187     public void unsupportedInvalidParams(@XPathParam("/")String param1, String param2) {
188 
189     }
190 
191     public String unsupportedInvalidReturnType(@XPathParam("/")String param1) {
192         return null;
193     }
194 
195     public void unsupportedInvalidParamType(@XPathParam("/")int param1) {
196     }
197 
198     public void namespaces(@XPathParam(".")Node param) {
199         namespacesInvoked = true;
200         assertEquals("Invalid parameter", "child", param.getLocalName());
201     }
202 }