Browse Source

SpringTestContext allow setting Context

Fixes: gh-5015
Rob Winch 7 years ago
parent
commit
1ad57adccc

+ 6 - 0
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);