如何为在源控制器中自动装配的junit测试用例注入servletcontext

Question

I have a controller class in which, servletContext is autowired.

@Controller
public class BaseController {
    @Autowired
    private ServletContext servletContext;

protected ServletContext getServletContext() {
        return servletContext;
    }

    public Object getBean(String beanName) {

        ApplicationContext appContext =    WebApplicationContextUtils.getWebApplicationContext(this.servletContext);
        return appContext.getBean(beanName);
    }
}

The getBean() in BaseController is used by many methods in subclasses extending it.

When I tried to execute jUnit test cases for the methods in subclasses (which were invoking getBean() ), I was getting the below error.

java.lang.IllegalArgumentException: ServletContext must not be null

I am unable to load web.xml while running junit.

Can someone please help in explaining , on how to set servlet context from JUnit ? If MockServletContext is the solution, please provide a code snippet on how to inject MockServletContext , to the autowired servletContext in the controller.

Note : There is no setServletContext() in BaseController

Answer 1

您可以使用PowerMock的WhiteBox设置servletContext。

WhiteBox.setInternalState(controller, "servletContext", yourServletContext);