From be8aad8306ef02d57c361892c389839e7426d675 Mon Sep 17 00:00:00 2001 From: Rob Winch Date: Fri, 13 Sep 2013 16:20:43 -0700 Subject: [PATCH] SEC-2196: Demonstrate Method Security works on Generic methods --- .../security/config/method/Sec2196Tests.java | 79 +++++++++++++++++++ 1 file changed, 79 insertions(+) create mode 100644 config/src/test/java/org/springframework/security/config/method/Sec2196Tests.java diff --git a/config/src/test/java/org/springframework/security/config/method/Sec2196Tests.java b/config/src/test/java/org/springframework/security/config/method/Sec2196Tests.java new file mode 100644 index 0000000000..5ad0c54ea1 --- /dev/null +++ b/config/src/test/java/org/springframework/security/config/method/Sec2196Tests.java @@ -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.config.method; + +import org.junit.After; +import org.junit.Test; +import org.springframework.context.ConfigurableApplicationContext; +import org.springframework.security.access.AccessDeniedException; +import org.springframework.security.access.prepost.PreAuthorize; +import org.springframework.security.authentication.TestingAuthenticationToken; +import org.springframework.security.config.util.InMemoryXmlApplicationContext; +import org.springframework.security.core.context.SecurityContextHolder; + +/** + * @author Rob Winch + * + */ +public class Sec2196Tests { + + private ConfigurableApplicationContext context; + + @Test(expected = AccessDeniedException.class) + public void genericMethodsProtected() { + loadContext("" + + ""); + + SecurityContextHolder.getContext().setAuthentication( + new TestingAuthenticationToken("test", "pass", "ROLE_USER")); + Service service = context.getBean(Service.class); + service.save(new User()); + } + + @Test + public void genericMethodsAllowed() { + loadContext("" + + ""); + + SecurityContextHolder.getContext().setAuthentication( + new TestingAuthenticationToken("test", "pass", "saveUsers")); + Service service = context.getBean(Service.class); + service.save(new User()); + } + + private void loadContext(String context) { + this.context = new InMemoryXmlApplicationContext(context); + } + + @After + public void closeAppContext() { + if (context != null) { + context.close(); + context = null; + } + SecurityContextHolder.clearContext(); + } + + public static class Service { + @PreAuthorize("hasAuthority('saveUsers')") + public T save(T dto) { + return dto; + } + } + + static class User { + } +}