Spring Boot 单元测试运行整个程序

Question

我想用 Spring Boot 实现一个集成测试。我从spring-boot-starter-test 依赖项开始，版本2.2.5.RELEASE。

我有以下组件：

@Component
public class MyMath {

    public int add(int a, int b) {
        return a + b;
    }

}

主程序如下所示：

@SpringBootApplication
public class Main implements CommandLineRunner {

    public static void main(String[] args) {
        SpringApplication.run(Main.class, args);
    }

    @Autowired
    private MyMath myMath;

    @Override
    public void run(String... args) throws Exception {
        System.out.println(myMath.add(2, 3));
    }

}

它按预期工作 - 到目前为止，一切都很好。我想添加一个单元测试：

@RunWith(SpringRunner.class)
@SpringBootTest
public class MyMathTest {

    @Autowired
    private MyMath myMath;

    @Test
    public void testSaveAndList() {
        assertEquals(5, myMath.add(2, 3));
    }

}

这也有效，但根据日志它执行整个程序。我不想运行程序本身，只想运行MyMath.add() 函数。我该怎么做？

到目前为止，我尝试了以下方法：

• @RunWith(SpringJUnit4ClassRunner.class) 提供了相同的结果。
• 省略@SpringBootTest 结果NoSuchBeanDefinitionException。
• 重新格式化代码以使用 bean 而不是像下面这样的组件。

MyMath 无注解：

public class MyMath {

    public int add(int a, int b) {
        return a + b;
    }

}

Main 保持不变。

@Configuration
public class AppConfig {

    @Bean
    public MyMath getMyMath() {
        return new MyMath();
    }
}

还有测试：

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = AppConfig.class)
public class MyMathTest {

    @Autowired
    private MyMath myMath;

    @Test
    public void testSaveAndList() {
        assertEquals(5, myMath.add(2, 3));
    }

}

所以我不能做的是在不运行整个程序的情况下测试一个组件。有什么可以帮助我的吗？谢谢！

Answer 1

您无需重构代码。保持 MyMath 类的原样

@Component
public class MyMath {

    public int add(int a, int b) {
        return a + b;
    }
}

像这样改变你的测试类

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = MyMath.class)
public class MyMathTest {

    @Autowired
    private MyMath myMath;

    @Test
    public void testSaveAndList() {
        assertEquals(5, myMath.add(2, 3));
    }

}

如果您的 MyMath 类具有自动装配的其他依赖项，这会变得有点复杂。然后你必须使用模拟。

Answer 2

如果您的 MyMath 类如此简单，我不会使用 Spring 来初始化它。没有必要，更重要的是 - junit 应该很快。因此，与其使用 Spring Context 运行测试，不如将其更改为简单的 JUnit 并像普通对象一样创建 MyMath：

public class MyMathTest {

    private MyMath myMath = new MyMath();

    @Test
    public void testSaveAndList() {
        assertEquals(5, myMath.add(2, 3));
    }

}