SpringBoot让测试类飞起来的方法

单元测试是项目开发中必不可少的一环，在 SpringBoot 的项目中，我们用 @SpringBootTest 注解来标注一个测试类，在测试类中注入这个接口的实现类之后对每个方法进行单独测试。

比如下面这个示例测试类：

@SpringBootTest
public class Tests {  
    @Test
    public void testHello() {
        // ...
    }
}

但是随着项目的代码量越来越大，你会发现测试类的启动速度变得越来越慢，而大多数情况下只是为了测试一下某个实现类的某个方法而已，比如测试一个DAO类的persist方法。

实际上， @SpringBootTest 注解还提供了两个参数，好好利用这两个参数就可以让测试类的启动速度变得更快。

1.webEnvironment

这个属性决定了测试类要不要启动一个 web 环境，说白了就是要不要启动一个 Tomcat 容器，可选的值为：

MOCK, 启动一个模拟的 Servlet 环境，这是默认值
RANDOM_PORT，启动一个 Tomcat 容器，并监听一个随机的端口号
DEFINED_PORT，启动一个 Tomcat 容器，并监听配置文件中定义的端口（未定义则默认监听8080）
NONE，不启动 Tomcat 容器

只需要通过指定 @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE) 即可达到加速的效果。这时测试类启动时就只会初始化 Spring 上下文，不再启动 Tomcat 容器了!

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)
public class Tests {  
    @Test
    public void testHello() {
        // ...
    }
}

2.classes

classes 属性用来指定运行测试类需要装载的 class 集合，如果不指定，那么会默认装载 @SpringBootConfiguration 注解标注的类。

提到 @SpringBootConfiguration 你可能比较陌生，其实 @SpringBootApplication 的源码里就引入了这个注解：

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@SpringBootConfiguration
@EnableAutoConfiguration
@ComponentScan(
    excludeFilters = {@Filter(
    type = FilterType.CUSTOM,
    classes = {TypeExcludeFilter.class}
), @Filter(
    type = FilterType.CUSTOM,
    classes = {AutoConfigurationExcludeFilter.class}
)}
)
public @interface SpringBootApplication {
    // ...
}

也就是说，如果我们不指定classes属性，那么启动测试类时需要加载的Bean的数量和正常启动一次入口类(即有@SpringBootApplication注解的类)加载的 Bean 数量是一样的。

如果你的项目中有很多个 Bean，特别是有以下几种时：

有 CommandLineRunner 的实现类
用 @PostConstruct 注解指定了初始化方法的类

启动测试类的时候，就只会加载需要的 Bean 到上下文中，从而加快启动速度。比如：

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE, classes={HelloServiceImpl.class})
public class Tests {  
    @Test
    public void testHello() {
        // ...
    }
}

即使此时项目中还有另外一个 Bean 在它的初始化方法里写了类似 Thread.sleep(10000) 等操作也不会影响，因为这个 Bean 根本就没有被加载和初始化!