diff --git a/config/spring-authorization-server-config.gradle b/config/spring-authorization-server-config.gradle index dbedfce..3a88f2a 100644 --- a/config/spring-authorization-server-config.gradle +++ b/config/spring-authorization-server-config.gradle @@ -6,6 +6,7 @@ dependencies { compile springCoreDependency compile project(':spring-authorization-server-core') + testCompile 'org.springframework.security:spring-security-test' testCompile 'junit:junit' testCompile 'org.assertj:assertj-core' testCompile 'org.mockito:mockito-core' diff --git a/config/src/test/java/org/springframework/security/config/test/SpringTestContext.java b/config/src/test/java/org/springframework/security/config/test/SpringTestContext.java new file mode 100644 index 0000000..cbac116 --- /dev/null +++ b/config/src/test/java/org/springframework/security/config/test/SpringTestContext.java @@ -0,0 +1,153 @@ +/* + * Copyright 2002-2017 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 + * + * https://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.test; + +import org.springframework.beans.factory.annotation.AutowiredAnnotationBeanPostProcessor; +import org.springframework.mock.web.MockServletConfig; +import org.springframework.mock.web.MockServletContext; +import org.springframework.security.config.util.InMemoryXmlWebApplicationContext; +import org.springframework.test.context.web.GenericXmlWebContextLoader; +import org.springframework.test.web.servlet.MockMvc; +import org.springframework.test.web.servlet.request.RequestPostProcessor; +import org.springframework.test.web.servlet.setup.ConfigurableMockMvcBuilder; +import org.springframework.test.web.servlet.setup.MockMvcBuilders; +import org.springframework.test.web.servlet.setup.MockMvcConfigurer; +import org.springframework.web.context.ConfigurableWebApplicationContext; +import org.springframework.web.context.WebApplicationContext; +import org.springframework.web.context.support.AnnotationConfigWebApplicationContext; +import org.springframework.web.context.support.XmlWebApplicationContext; +import org.springframework.web.filter.OncePerRequestFilter; + +import javax.servlet.Filter; +import javax.servlet.FilterChain; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.Closeable; +import java.util.ArrayList; +import java.util.List; + +import static org.springframework.security.config.BeanIds.SPRING_SECURITY_FILTER_CHAIN; +import static org.springframework.security.test.web.servlet.setup.SecurityMockMvcConfigurers.springSecurity; + +/** + * TODO + * This class is a straight copy from Spring Security. + * It should be removed when merging this codebase into Spring Security. + * + * @author Rob Winch + * @since 5.0 + */ +public class SpringTestContext implements Closeable { + private Object test; + + private ConfigurableWebApplicationContext context; + + private List filters = new ArrayList<>(); + + public void setTest(Object test) { + this.test = test; + } + + @Override + public void close() { + try { + this.context.close(); + } catch(Exception e) {} + } + + public SpringTestContext context(ConfigurableWebApplicationContext context) { + this.context = context; + return this; + } + + public SpringTestContext register(Class... classes) { + AnnotationConfigWebApplicationContext applicationContext = new AnnotationConfigWebApplicationContext(); + applicationContext.register(classes); + this.context = applicationContext; + return this; + } + + public SpringTestContext testConfigLocations(String... configLocations) { + GenericXmlWebContextLoader loader = new GenericXmlWebContextLoader(); + String[] locations = loader.processLocations(this.test.getClass(), + configLocations); + return configLocations(locations); + } + + public SpringTestContext configLocations(String... configLocations) { + XmlWebApplicationContext context = new XmlWebApplicationContext(); + context.setConfigLocations(configLocations); + this.context = context; + return this; + } + + public SpringTestContext context(String configuration) { + InMemoryXmlWebApplicationContext context = new InMemoryXmlWebApplicationContext(configuration); + this.context = context; + return this; + } + + public SpringTestContext mockMvcAfterSpringSecurityOk() { + return addFilter(new OncePerRequestFilter() { + @Override + protected void doFilterInternal(HttpServletRequest request, + HttpServletResponse response, FilterChain filterChain) { + response.setStatus(HttpServletResponse.SC_OK); + } + }); + } + + private SpringTestContext addFilter(Filter filter) { + this.filters.add(filter); + return this; + } + + public ConfigurableWebApplicationContext getContext() { + if (!this.context.isRunning()) { + this.context.setServletContext(new MockServletContext()); + this.context.setServletConfig(new MockServletConfig()); + this.context.refresh(); + } + return this.context; + } + + public void autowire() { + this.context.setServletContext(new MockServletContext()); + this.context.setServletConfig(new MockServletConfig()); + this.context.refresh(); + + if (this.context.containsBean(SPRING_SECURITY_FILTER_CHAIN)) { + MockMvc mockMvc = MockMvcBuilders.webAppContextSetup(this.context) + .apply(springSecurity()) + .apply(new AddFilter()).build(); + this.context.getBeanFactory() + .registerResolvableDependency(MockMvc.class, mockMvc); + } + + AutowiredAnnotationBeanPostProcessor bpp = new AutowiredAnnotationBeanPostProcessor(); + bpp.setBeanFactory(this.context.getBeanFactory()); + bpp.processInjection(this.test); + } + + private class AddFilter implements MockMvcConfigurer { + public RequestPostProcessor beforeMockMvcCreated( + ConfigurableMockMvcBuilder builder, WebApplicationContext context) { + builder.addFilters(SpringTestContext.this.filters.toArray(new Filter[0])); + return null; + } + } +} diff --git a/config/src/test/java/org/springframework/security/config/test/SpringTestRule.java b/config/src/test/java/org/springframework/security/config/test/SpringTestRule.java new file mode 100644 index 0000000..56ca708 --- /dev/null +++ b/config/src/test/java/org/springframework/security/config/test/SpringTestRule.java @@ -0,0 +1,47 @@ +/* + * Copyright 2002-2017 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 + * + * https://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.test; + +import org.junit.rules.MethodRule; +import org.junit.runners.model.FrameworkMethod; +import org.junit.runners.model.Statement; +import org.springframework.security.test.context.TestSecurityContextHolder; + +/** + * TODO + * This class is a straight copy from Spring Security. + * It should be removed when merging this codebase into Spring Security. + * + * @author Rob Winch + * @since 5.0 + */ +public class SpringTestRule extends SpringTestContext implements MethodRule { + @Override + public Statement apply(Statement base, FrameworkMethod method, Object target) { + return new Statement() { + public void evaluate() throws Throwable { + setTest(target); + try { + base.evaluate(); + } finally { + TestSecurityContextHolder.clearContext(); + close(); + } + } + }; + } +} diff --git a/config/src/test/java/org/springframework/security/config/util/InMemoryXmlApplicationContext.java b/config/src/test/java/org/springframework/security/config/util/InMemoryXmlApplicationContext.java new file mode 100644 index 0000000..11e5f06 --- /dev/null +++ b/config/src/test/java/org/springframework/security/config/util/InMemoryXmlApplicationContext.java @@ -0,0 +1,81 @@ +/* + * Copyright 2009-2020 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 + * + * https://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.util; + +import org.springframework.beans.factory.support.DefaultListableBeanFactory; +import org.springframework.context.ApplicationContext; +import org.springframework.context.support.AbstractXmlApplicationContext; +import org.springframework.core.io.Resource; +import org.springframework.security.util.InMemoryResource; + +/** + * TODO + * This class is a straight copy from Spring Security. + * It should be removed when merging this codebase into Spring Security. + * + * @author Luke Taylor + * @author Eddú Meléndez + */ +public class InMemoryXmlApplicationContext extends AbstractXmlApplicationContext { + static final String BEANS_OPENING = "\n" + xml + BEANS_CLOSE; + inMemoryXml = new InMemoryResource(fullXml); + setAllowBeanDefinitionOverriding(true); + setParent(parent); + refresh(); + } + + @Override + protected DefaultListableBeanFactory createBeanFactory() { + return new DefaultListableBeanFactory(getInternalParentBeanFactory()) { + @Override + protected boolean allowAliasOverriding() { + return true; + } + }; + } + + protected Resource[] getConfigResources() { + return new Resource[] { inMemoryXml }; + } +} diff --git a/config/src/test/java/org/springframework/security/config/util/InMemoryXmlWebApplicationContext.java b/config/src/test/java/org/springframework/security/config/util/InMemoryXmlWebApplicationContext.java new file mode 100644 index 0000000..3910fab --- /dev/null +++ b/config/src/test/java/org/springframework/security/config/util/InMemoryXmlWebApplicationContext.java @@ -0,0 +1,61 @@ +/* + * Copyright 2012-2016 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 + * + * https://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.util; + +import org.springframework.beans.BeansException; +import org.springframework.beans.factory.support.DefaultListableBeanFactory; +import org.springframework.beans.factory.xml.XmlBeanDefinitionReader; +import org.springframework.context.ApplicationContext; +import org.springframework.core.io.Resource; +import org.springframework.security.util.InMemoryResource; +import org.springframework.web.context.support.AbstractRefreshableWebApplicationContext; + +import static org.springframework.security.config.util.InMemoryXmlApplicationContext.BEANS_CLOSE; +import static org.springframework.security.config.util.InMemoryXmlApplicationContext.BEANS_OPENING; +import static org.springframework.security.config.util.InMemoryXmlApplicationContext.SPRING_SECURITY_VERSION; + +/** + * TODO + * This class is a straight copy from Spring Security. + * It should be removed when merging this codebase into Spring Security. + * + * @author Joe Grandja + */ +public class InMemoryXmlWebApplicationContext extends AbstractRefreshableWebApplicationContext { + private Resource inMemoryXml; + + public InMemoryXmlWebApplicationContext(String xml) { + this(xml, SPRING_SECURITY_VERSION, null); + } + + public InMemoryXmlWebApplicationContext(String xml, ApplicationContext parent) { + this(xml, SPRING_SECURITY_VERSION, parent); + } + + public InMemoryXmlWebApplicationContext(String xml, String secVersion, ApplicationContext parent) { + String fullXml = BEANS_OPENING + secVersion + ".xsd'>\n" + xml + BEANS_CLOSE; + inMemoryXml = new InMemoryResource(fullXml); + setAllowBeanDefinitionOverriding(true); + setParent(parent); + } + + @Override + protected void loadBeanDefinitions(DefaultListableBeanFactory beanFactory) throws BeansException { + XmlBeanDefinitionReader reader = new XmlBeanDefinitionReader(beanFactory); + reader.loadBeanDefinitions(new Resource[] { inMemoryXml }); + } + +}