1
2
3
4
5
6
7
8
9
10
11
12
13
14
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 }