From 1ad57adcccf098ca30042d968693e030ac762147 Mon Sep 17 00:00:00 2001 From: Rob Winch Date: Mon, 12 Feb 2018 14:44:24 -0600 Subject: [PATCH] SpringTestContext allow setting Context Fixes: gh-5015 --- .../security/config/test/SpringTestContext.java | 6 ++++++ 1 file changed, 6 insertions(+) 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 index 4ba03e4ec7..482db06c0f 100644 --- a/config/src/test/java/org/springframework/security/config/test/SpringTestContext.java +++ b/config/src/test/java/org/springframework/security/config/test/SpringTestContext.java @@ -17,6 +17,7 @@ package org.springframework.security.config.test; import org.springframework.beans.factory.annotation.AutowiredAnnotationBeanPostProcessor; +import org.springframework.context.ApplicationContext; import org.springframework.context.ConfigurableApplicationContext; import org.springframework.mock.web.MockServletConfig; import org.springframework.mock.web.MockServletContext; @@ -68,6 +69,11 @@ public class SpringTestContext implements Closeable { } 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);