SpringBoot使用@RunWith(Parameterized.class)进行参数化测试同时支持依赖注入,以及CommandLineRunner在单元测试时不执行主进程

问题提出

在使用SpringBoot进行单元测试的时候,我发现了两个问题

  1. 使用参数化测试的时候,必须使用 @RunWith(Parameterized.class),而对Spring进行单元测试时,如果想使用依赖注入,即使用 @Autowired 注解,需要使用 @RunWith(SpringRunner.class) 。两个不能同时使用。
  2. 如果 SpringBootApplication 在启动时执行一个 CommandLineRunner,那么在启动单元测试的时候,会立即执行该Runner,但是我只想执行我的单元测试,并不想让整个程序都启动起来。

解决方案

我查了一下资料,发现相关的文章很少,Google了一下发现一篇文章 写JUnit时既用Parameterized又用@Autowired出现无法成功自动注入的解决办法 有提到相关的问题,但是这篇文章在关键位置排版太乱,看不清楚。于是自己尝试着处理了一下。

Parameterized和SpringRunner

只使用 @RunWith(Parameterized.class),跟普通的参数化测试一样。然后在测试类中添加以下几行代码,就可以使用依赖注入了。

    @Before
    public void setUp() throws Exception {
        TestContextManager testContextManager = new TestContextManager(getClass());
        testContextManager.prepareTestInstance(this);
    }

单元测试启动时不执行 CommandLineRunner

我的程序主要任务是监听mq中的消息。而单元测试只想测试每个Handler的处理逻辑是否正确。但是单元测试启动的时候,会先启动SpringBootApplication,于是我的监听程序就启动了。这并不是我想要的。特别是这个程序启动之后要是一直阻塞在这里的话,单元测试就无法进行了。
SpringBoot使用@RunWith(Parameterized.class)进行参数化测试同时支持依赖注入,以及CommandLineRunner在单元测试时不执行主进程

我查了一些资料,没有找到满意的解决方案。于是想了一个办法。思路是在执行 run 方法的时候判断一下当前的 Profile ,如果是 test 就直接返回,不执行后面的逻辑。(当然,这个前提是你单元测试使用的配置文件中设置了 spring.profiles.active=test,正常都会这么做)
SpringBoot使用@RunWith(Parameterized.class)进行参数化测试同时支持依赖注入,以及CommandLineRunner在单元测试时不执行主进程
以上两个问题,如果你有更好的解决方案,可以在留言区留言,我们一起交流探讨,谢谢!