7 changed files with 395 additions and 0 deletions
@ -0,0 +1,79 @@
@@ -0,0 +1,79 @@
|
||||
/* |
||||
* Copyright 2002-2013 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not |
||||
* use this file except in compliance with the License. You may obtain a copy of |
||||
* the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
||||
* License for the specific language governing permissions and limitations under |
||||
* the License. |
||||
*/ |
||||
package org.springframework.security.test.context.showcase; |
||||
|
||||
import org.springframework.security.core.GrantedAuthority; |
||||
import org.springframework.security.core.authority.AuthorityUtils; |
||||
import org.springframework.security.core.userdetails.UserDetails; |
||||
|
||||
import java.util.Collection; |
||||
|
||||
/** |
||||
* @author Rob Winch |
||||
*/ |
||||
public class CustomUserDetails implements UserDetails { |
||||
private final String name; |
||||
private final String username; |
||||
private final Collection<? extends GrantedAuthority> authorities; |
||||
|
||||
public CustomUserDetails(String name, String username) { |
||||
this.name = name; |
||||
this.username = username; |
||||
this.authorities = AuthorityUtils.createAuthorityList("ROLE_USER"); |
||||
} |
||||
|
||||
@Override |
||||
public Collection<? extends GrantedAuthority> getAuthorities() { |
||||
return authorities; |
||||
} |
||||
|
||||
@Override |
||||
public String getPassword() { |
||||
return null; |
||||
} |
||||
|
||||
@Override |
||||
public String getUsername() { |
||||
return username; |
||||
} |
||||
|
||||
@Override |
||||
public boolean isAccountNonExpired() { |
||||
return true; |
||||
} |
||||
|
||||
@Override |
||||
public boolean isAccountNonLocked() { |
||||
return true; |
||||
} |
||||
|
||||
@Override |
||||
public boolean isCredentialsNonExpired() { |
||||
return true; |
||||
} |
||||
|
||||
@Override |
||||
public boolean isEnabled() { |
||||
return true; |
||||
} |
||||
|
||||
@Override |
||||
public String toString() { |
||||
return "CustomUserDetails{" + |
||||
"username='" + username + '\'' + |
||||
'}'; |
||||
} |
||||
} |
||||
@ -0,0 +1,46 @@
@@ -0,0 +1,46 @@
|
||||
/* |
||||
* Copyright 2002-2014 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package org.springframework.security.test.context.showcase; |
||||
|
||||
import org.springframework.security.test.context.support.WithSecurityContext; |
||||
|
||||
/** |
||||
* @author Rob Winch |
||||
*/ |
||||
@WithSecurityContext(factory = WithMockCustomUserSecurityContextFactory.class) |
||||
public @interface WithMockCustomUser { |
||||
/** |
||||
* The username to be used. The default is rob |
||||
* @return |
||||
*/ |
||||
String username() default "rob"; |
||||
|
||||
/** |
||||
* The roles to use. The default is "USER". A {@link org.springframework.security.core.GrantedAuthority} will |
||||
* be created for each value within roles. Each value in roles will |
||||
* automatically be prefixed with "ROLE_". For example, the default will |
||||
* result in "ROLE_USER" being used. |
||||
* |
||||
* @return |
||||
*/ |
||||
String[] roles() default { "USER" }; |
||||
|
||||
/** |
||||
* The name of the user |
||||
* @return |
||||
*/ |
||||
String name() default "Rob Winch"; |
||||
} |
||||
@ -0,0 +1,44 @@
@@ -0,0 +1,44 @@
|
||||
/* |
||||
* Copyright 2002-2013 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not |
||||
* use this file except in compliance with the License. You may obtain a copy of |
||||
* the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
||||
* License for the specific language governing permissions and limitations under |
||||
* the License. |
||||
*/ |
||||
package org.springframework.security.test.context.showcase; |
||||
|
||||
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; |
||||
import org.springframework.security.core.Authentication; |
||||
import org.springframework.security.core.GrantedAuthority; |
||||
import org.springframework.security.core.authority.AuthorityUtils; |
||||
import org.springframework.security.core.authority.SimpleGrantedAuthority; |
||||
import org.springframework.security.core.context.SecurityContext; |
||||
import org.springframework.security.core.context.SecurityContextHolder; |
||||
import org.springframework.security.test.context.support.WithSecurityContextFactory; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
/** |
||||
* @author Rob Winch |
||||
*/ |
||||
public class WithMockCustomUserSecurityContextFactory implements WithSecurityContextFactory<WithMockCustomUser> { |
||||
@Override |
||||
public SecurityContext createSecurityContext(WithMockCustomUser customUser) { |
||||
SecurityContext context = SecurityContextHolder.createEmptyContext(); |
||||
|
||||
CustomUserDetails principal = new CustomUserDetails(customUser.name(), customUser.username()); |
||||
Authentication auth = |
||||
new UsernamePasswordAuthenticationToken(principal, "password", principal.getAuthorities()); |
||||
context.setAuthentication(auth); |
||||
return context; |
||||
} |
||||
} |
||||
@ -0,0 +1,83 @@
@@ -0,0 +1,83 @@
|
||||
/* |
||||
* Copyright 2002-2013 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not |
||||
* use this file except in compliance with the License. You may obtain a copy of |
||||
* the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
||||
* License for the specific language governing permissions and limitations under |
||||
* the License. |
||||
*/ |
||||
package org.springframework.security.test.context.showcase; |
||||
|
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.context.annotation.ComponentScan; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.security.authentication.AuthenticationCredentialsNotFoundException; |
||||
import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder; |
||||
import org.springframework.security.config.annotation.method.configuration.EnableGlobalMethodSecurity; |
||||
import org.springframework.security.test.context.DefaultSecurityTestExecutionListeners; |
||||
import org.springframework.security.test.context.showcase.service.HelloMessageService; |
||||
import org.springframework.security.test.context.showcase.service.MessageService; |
||||
import org.springframework.security.test.context.support.WithMockUser; |
||||
import org.springframework.test.context.ContextConfiguration; |
||||
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; |
||||
|
||||
import static org.fest.assertions.Assertions.assertThat; |
||||
|
||||
/** |
||||
* @author Rob Winch |
||||
*/ |
||||
|
||||
@RunWith(SpringJUnit4ClassRunner.class) |
||||
@ContextConfiguration(classes = WithMockUserTests.Config.class) |
||||
@DefaultSecurityTestExecutionListeners |
||||
public class WithMockUserTests { |
||||
@Autowired |
||||
private MessageService messageService; |
||||
|
||||
@Test(expected = AuthenticationCredentialsNotFoundException.class) |
||||
public void getMessageUnauthenticated() { |
||||
messageService.getMessage(); |
||||
} |
||||
|
||||
@Test |
||||
@WithMockUser |
||||
public void getMessageWithMockUser() { |
||||
String message = messageService.getMessage(); |
||||
assertThat(message).contains("user"); |
||||
} |
||||
|
||||
@Test |
||||
@WithMockUser("customUsername") |
||||
public void getMessageWithMockUserCustomUsername() { |
||||
String message = messageService.getMessage(); |
||||
assertThat(message).contains("customUsername"); |
||||
} |
||||
|
||||
@Test |
||||
@WithMockUser(username="admin",roles={"USER","ADMIN"}) |
||||
public void getMessageWithMockUserCustomUser() { |
||||
String message = messageService.getMessage(); |
||||
assertThat(message).contains("admin").contains("ROLE_USER").contains("ROLE_ADMIN"); |
||||
} |
||||
|
||||
@Configuration |
||||
@EnableGlobalMethodSecurity(prePostEnabled = true) |
||||
@ComponentScan(basePackageClasses = HelloMessageService.class) |
||||
static class Config { |
||||
@Autowired |
||||
public void configureGlobal(AuthenticationManagerBuilder auth) throws Exception { |
||||
auth |
||||
.inMemoryAuthentication() |
||||
.withUser("user").password("password").roles("USER"); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,99 @@
@@ -0,0 +1,99 @@
|
||||
/* |
||||
* Copyright 2002-2013 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not |
||||
* use this file except in compliance with the License. You may obtain a copy of |
||||
* the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
||||
* License for the specific language governing permissions and limitations under |
||||
* the License. |
||||
*/ |
||||
package org.springframework.security.test.context.showcase; |
||||
|
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.context.annotation.Bean; |
||||
import org.springframework.context.annotation.ComponentScan; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.security.authentication.AuthenticationCredentialsNotFoundException; |
||||
import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder; |
||||
import org.springframework.security.config.annotation.method.configuration.EnableGlobalMethodSecurity; |
||||
import org.springframework.security.core.context.SecurityContextHolder; |
||||
import org.springframework.security.core.userdetails.UserDetails; |
||||
import org.springframework.security.core.userdetails.UserDetailsService; |
||||
import org.springframework.security.core.userdetails.UsernameNotFoundException; |
||||
import org.springframework.security.test.context.DefaultSecurityTestExecutionListeners; |
||||
import org.springframework.security.test.context.showcase.service.HelloMessageService; |
||||
import org.springframework.security.test.context.showcase.service.MessageService; |
||||
import org.springframework.security.test.context.support.WithUserDetails; |
||||
import org.springframework.test.context.ContextConfiguration; |
||||
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; |
||||
|
||||
import static org.fest.assertions.Assertions.assertThat; |
||||
|
||||
/** |
||||
* @author Rob Winch |
||||
*/ |
||||
|
||||
@RunWith(SpringJUnit4ClassRunner.class) |
||||
@ContextConfiguration(classes = WithUserDetailsTests.Config.class) |
||||
@DefaultSecurityTestExecutionListeners |
||||
public class WithUserDetailsTests { |
||||
@Autowired |
||||
private MessageService messageService; |
||||
|
||||
@Test(expected = AuthenticationCredentialsNotFoundException.class) |
||||
public void getMessageUnauthenticated() { |
||||
messageService.getMessage(); |
||||
} |
||||
|
||||
@Test |
||||
@WithUserDetails |
||||
public void getMessageWithUserDetails() { |
||||
String message = messageService.getMessage(); |
||||
assertThat(message).contains("user"); |
||||
assertThat(getPrincipal()).isInstanceOf(CustomUserDetails.class); |
||||
} |
||||
|
||||
@Test |
||||
@WithUserDetails("customUsername") |
||||
public void getMessageWithUserDetailsCustomUsername() { |
||||
String message = messageService.getMessage(); |
||||
assertThat(message).contains("customUsername"); |
||||
assertThat(getPrincipal()).isInstanceOf(CustomUserDetails.class); |
||||
} |
||||
|
||||
@Configuration |
||||
@EnableGlobalMethodSecurity(prePostEnabled = true) |
||||
@ComponentScan(basePackageClasses = HelloMessageService.class) |
||||
static class Config { |
||||
@Autowired |
||||
public void configureGlobal(AuthenticationManagerBuilder auth) throws Exception { |
||||
auth |
||||
.userDetailsService(userDetailsService()); |
||||
} |
||||
|
||||
@Bean |
||||
public UserDetailsService userDetailsService() { |
||||
return new CustomUserDetailsService(); |
||||
} |
||||
} |
||||
|
||||
private Object getPrincipal() { |
||||
return SecurityContextHolder.getContext().getAuthentication().getPrincipal(); |
||||
} |
||||
|
||||
static class CustomUserDetailsService implements UserDetailsService { |
||||
|
||||
@Override |
||||
public UserDetails loadUserByUsername(final String username) throws UsernameNotFoundException { |
||||
return new CustomUserDetails("name", username); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,34 @@
@@ -0,0 +1,34 @@
|
||||
/* |
||||
* Copyright 2002-2013 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not |
||||
* use this file except in compliance with the License. You may obtain a copy of |
||||
* the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
||||
* License for the specific language governing permissions and limitations under |
||||
* the License. |
||||
*/ |
||||
package org.springframework.security.test.context.showcase.service; |
||||
|
||||
import org.springframework.security.access.prepost.PreAuthorize; |
||||
import org.springframework.security.core.Authentication; |
||||
import org.springframework.security.core.context.SecurityContextHolder; |
||||
import org.springframework.stereotype.Component; |
||||
|
||||
/** |
||||
* @author Rob Winch |
||||
*/ |
||||
@Component |
||||
public class HelloMessageService implements MessageService { |
||||
|
||||
@PreAuthorize("authenticated") |
||||
public String getMessage() { |
||||
Authentication authentication = SecurityContextHolder.getContext().getAuthentication(); |
||||
return "Hello " + authentication; |
||||
} |
||||
} |
||||
@ -0,0 +1,10 @@
@@ -0,0 +1,10 @@
|
||||
package org.springframework.security.test.context.showcase.service; |
||||
|
||||
import org.springframework.security.access.prepost.PreAuthorize; |
||||
|
||||
/** |
||||
* @author Rob Winch |
||||
*/ |
||||
public interface MessageService { |
||||
String getMessage(); |
||||
} |
||||
Loading…
Reference in new issue