1   /*
2    * Copyright 2006 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.soap.server.endpoint.mapping;
18  
19  import junit.framework.TestCase;
20  import org.easymock.MockControl;
21  import org.springframework.ws.MockWebServiceMessageFactory;
22  import org.springframework.ws.context.DefaultMessageContext;
23  import org.springframework.ws.context.MessageContext;
24  import org.springframework.ws.server.EndpointInvocationChain;
25  import org.springframework.ws.server.EndpointMapping;
26  import org.springframework.ws.soap.server.SoapEndpointInvocationChain;
27  
28  public class DelegatingSoapEndpointMappingTest extends TestCase {
29  
30      private DelegatingSoapEndpointMapping endpointMapping;
31  
32      private MockControl control;
33  
34      private EndpointMapping mock;
35  
36      protected void setUp() throws Exception {
37          endpointMapping = new DelegatingSoapEndpointMapping();
38          control = MockControl.createControl(EndpointMapping.class);
39          mock = (EndpointMapping) control.getMock();
40          endpointMapping.setDelegate(mock);
41      }
42  
43      public void testGetEndpointMapping() throws Exception {
44          String role = "http://www.springframework.org/spring-ws/role";
45          endpointMapping.setActorOrRole(role);
46          MessageContext context = new DefaultMessageContext(new MockWebServiceMessageFactory());
47          EndpointInvocationChain delegateChain = new EndpointInvocationChain(new Object());
48          control.expectAndReturn(mock.getEndpoint(context), delegateChain);
49          control.replay();
50          SoapEndpointInvocationChain resultChain = (SoapEndpointInvocationChain) endpointMapping.getEndpoint(context);
51          assertNotNull("No chain returned", resultChain);
52          assertEquals("Invalid ampount of roles returned", 1, resultChain.getActorsOrRoles().length);
53          assertEquals("Invalid role returned", role, resultChain.getActorsOrRoles()[0]);
54          control.verify();
55      }
56  }