SpringFramework之mvc controller的单元测试

克虏伯

发布于 2019-04-15 10:13:53

4950

发布于 2019-04-15 10:13:53

文章被收录于专栏：软件开发-青出于蓝

项目里面经常会将controller的扫描配置与其它的分开以便于管理开发，但是controller的bean是在webApplicationContext中，与web容器结合起来，怎么单元测试时该怎么做呢?

１、手动创建webApplicationContext

如下List-1.1所示，手动创建XmlWebApplicationContext，并指定xml路径

List-1.1

import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.mock.web.MockRequestDispatcher;
import org.springframework.mock.web.MockServletContext;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.XmlWebApplicationContext;

@Configuration
public class WebApplicationContextConfiguration implements ApplicationContextAware {
    private ApplicationContext applicationContext;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
    }

    @Bean
    public WebApplicationContext getBean() {
        MockServletContext servletContext = new MockServletContext();
        servletContext.setMajorVersion(2);
        servletContext.setMinorVersion(4);
        servletContext.setContextPath("");
        XmlWebApplicationContext webApplicationContext = new XmlWebApplicationContext();
        webApplicationContext.setConfigLocation("classpath*:spring-mvc.xml");
        servletContext.registerNamedDispatcher("default", new MockRequestDispatcher("default"));
        webApplicationContext.setServletContext(servletContext);
        webApplicationContext.setParent(applicationContext);
        webApplicationContext.refresh();
        return webApplicationContext;
    }
}

２、建测试基类

如下List-2.1所示

List-2.1

import org.junit.runner.RunWith;
import org.springframework.test.context.ActiveProfiles;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"classpath*:spring-context.xml", "classpath*:spring-db.xml"})
public class TestBase {

}

3、样例

如下List-3.1所示，拿到webApplicationContext，就可以做controller的单元测试了。

List-3.1

import org.junit.Assert;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.mvc.condition.PatternsRequestCondition;
import org.springframework.web.servlet.mvc.method.RequestMappingInfo;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;

public class TestController extends TestBase {
    private static final Logger LOGGER = LoggerFactory.getLogger(TestController.class);
    @Autowired(required = true)
    private WebApplicationContext webApplicationContext;

    @Test
    public void test() {
        Assert.assertNotNull(webApplicationContext);
    }

(adsbygoogle = window.adsbygoogle || []).push({});

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

单元测试