1   /*
2    * Copyright 2005-2010 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.security.xwss.callback;
18  
19  import org.springframework.security.core.Authentication;
20  import org.springframework.security.core.context.SecurityContextHolder;
21  import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
22  
23  import com.sun.xml.wss.impl.callback.PasswordCallback;
24  import com.sun.xml.wss.impl.callback.UsernameCallback;
25  import org.junit.After;
26  import org.junit.Assert;
27  import org.junit.Before;
28  import org.junit.Test;
29  
30  public class SpringUsernamePasswordCallbackHandlerTest {
31  
32      private SpringUsernamePasswordCallbackHandler handler;
33  
34      @Before
35      public void setUp() throws Exception {
36          handler = new SpringUsernamePasswordCallbackHandler();
37          Authentication authentication = new UsernamePasswordAuthenticationToken("Bert", "Ernie");
38          SecurityContextHolder.getContext().setAuthentication(authentication);
39      }
40  
41      @After
42      public void tearDown() throws Exception {
43          SecurityContextHolder.clearContext();
44      }
45  
46      @Test
47      public void testUsernameCallback() throws Exception {
48          UsernameCallback usernameCallback = new UsernameCallback();
49          handler.handleInternal(usernameCallback);
50          Assert.assertEquals("Invalid username", "Bert", usernameCallback.getUsername());
51      }
52  
53      @Test
54      public void testPasswordCallback() throws Exception {
55          PasswordCallback passwordCallback = new PasswordCallback();
56          handler.handleInternal(passwordCallback);
57          Assert.assertEquals("Invalid username", "Ernie", passwordCallback.getPassword());
58      }
59  }