소스 검색

Minor refactoring of aspects tests.

Luke Taylor 14 년 전
부모
커밋
7e44580c75
1개의 변경된 파일13개의 추가작업 그리고 12개의 파일을 삭제
  1. 13 12
      samples/aspectj/src/test/java/sample/aspectj/AspectJInterceptorTests.java

+ 13 - 12
samples/aspectj/src/test/java/sample/aspectj/AspectJInterceptorTests.java

@@ -16,6 +16,8 @@ import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 @RunWith(SpringJUnit4ClassRunner.class)
 @ContextConfiguration(locations = "classpath:aspectj-context.xml")
 public class AspectJInterceptorTests {
+    private Authentication admin = new UsernamePasswordAuthenticationToken("test", "xxx", AuthorityUtils.createAuthorityList("ROLE_ADMIN"));
+    private Authentication user = new UsernamePasswordAuthenticationToken("test", "xxx", AuthorityUtils.createAuthorityList("ROLE_USER"));
 
     @Autowired
     private Service service;
@@ -35,17 +37,13 @@ public class AspectJInterceptorTests {
 
     @Test(expected = AccessDeniedException.class)
     public void testSecuredMethodWrongRole() throws Exception {
-        Authentication token = new UsernamePasswordAuthenticationToken("test", "xxx", AuthorityUtils
-                .createAuthorityList("ROLE_ADMIN"));
-        SecurityContextHolder.getContext().setAuthentication(token);
+        SecurityContextHolder.getContext().setAuthentication(admin);
         service.secureMethod();
     }
 
     @Test
     public void testSecuredMethodEverythingOk() throws Exception {
-        Authentication token = new UsernamePasswordAuthenticationToken("test", "xxx", AuthorityUtils
-                .createAuthorityList("ROLE_USER"));
-        SecurityContextHolder.getContext().setAuthentication(token);
+        SecurityContextHolder.getContext().setAuthentication(user);
         service.secureMethod();
     }
 
@@ -56,18 +54,21 @@ public class AspectJInterceptorTests {
 
     @Test(expected = AccessDeniedException.class)
     public void testSecuredClassWrongRole() throws Exception {
-        Authentication token = new UsernamePasswordAuthenticationToken("test", "xxx", AuthorityUtils
-                .createAuthorityList("ROLE_ADMIN"));
-        SecurityContextHolder.getContext().setAuthentication(token);
+        SecurityContextHolder.getContext().setAuthentication(admin);
         securedService.secureMethod();
     }
 
+    @Test(expected = AccessDeniedException.class)
+    public void testSecuredClassWrongRoleOnNewedInstance() throws Exception {
+        SecurityContextHolder.getContext().setAuthentication(admin);
+        new SecuredService().secureMethod();
+    }
+
     @Test
     public void testSecuredClassEverythingOk() throws Exception {
-        Authentication token = new UsernamePasswordAuthenticationToken("test", "xxx", AuthorityUtils
-                .createAuthorityList("ROLE_USER"));
-        SecurityContextHolder.getContext().setAuthentication(token);
+        SecurityContextHolder.getContext().setAuthentication(user);
         securedService.secureMethod();
+        new SecuredService().secureMethod();
     }
 
     @After