瀏覽代碼

SpringTestContext allow setting Context

Fixes: gh-5015
Rob Winch 7 年之前
父節點
當前提交
1ad57adccc
共有 1 個文件被更改,包括 6 次插入0 次删除
  1. 6 0
      config/src/test/java/org/springframework/security/config/test/SpringTestContext.java

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