Spring Boot 单元测试二三事

本文翻译自： https://reflectoring.io/unit-...

原文做者：Tom Hombergs

译文原地址：https://weyunx.com/2019/02/04...

写好单元测试是一门技术活，不过好在咱们如今有不少框架来帮助咱们学习。

本文就为您介绍这些框架，同时详细介绍编写优秀的 Sping Boot 单元测试所必需的技术细节，

咱们将了解如何以可测试的方式建立 Spring bean，而后讨论 Mockito 和 AssertJ 的使用，这两个库在默认状况下都集成在 Spring Boot 里。

须要注意的是本文只讨论单元测试，组装测试、web 层测试和持久层测试会在后面的文章里讨论。

依赖

在本文中，咱们将使用 JUnit Jupiter (JUnit 5), Mockito, and AssertJ，同时还会引入 Lombok 来省去一些繁复的工做。

compileOnly('org.projectlombok:lombok')
testCompile('org.springframework.boot:spring-boot-starter-test')
testCompile 'org.junit.jupiter:junit-jupiter-engine:5.2.0'
testCompile('org.mockito:mockito-junit-jupiter:2.23.0')

spring-boot-starter-test 默认引入了 Mockito and AssertJ，对于 Lombok 则须要咱们本身手工引入。

不要使用 Spring 进行单元测试

看一下下面的「单元」测试，是用来测试 RegisterUseCase 类的一个方法：

@ExtendWith(SpringExtension.class)
@SpringBootTest
class RegisterUseCaseTest {

  @Autowired
  private RegisterUseCase registerUseCase;

  @Test
  void savedUserHasRegistrationDate() {
    User user = new User("zaphod", "zaphod@mail.com");
    User savedUser = registerUseCase.registerUser(user);
    assertThat(savedUser.getRegistrationDate()).isNotNull();
  }

}

咱们去执行这个测试类，花了大概 4.5 秒的时间，缘由仅仅是由于计算机要为它去运行一个空的 Spring 项目。

可是，一个好的单元测试应该是毫秒级的，不然这会影响「test / code / test」的工做方式，这也就是测试驱动开发的思想 (TDD)。即便咱们不作 TDD，在编写测试上花了太多时间也会影响咱们的开发思路。

其实，上面的测试方法实际执行只花费了几毫秒，剩下的 4.5 秒所有花费在了 @SpringBootRun 上，由于 Spring Boot 须要启动整个 Spring Boot 应用。

也就是说，咱们启动整个应用，耗费了大量资源，仅仅是去为了测试一个方法，当咱们的应用将来愈来愈大的时候，那将耗费更久的时间去启动。

因此，为何不要用 Spring Boot 来作单元测试呢？接下来，本文会讨论如何不用 Spring Boot 来进行单元测试。

建立测试类

一般，咱们能够有以下方法来让咱们的 Spring beans 更容易进行测试。

不要注入

首先咱们先看一个错误的例子：

@Service
public class RegisterUseCase {

  @Autowired
  private UserRepository userRepository;

  public User registerUser(User user) {
    return userRepository.save(user);
  }

}

然而这个类仍是必须经过 Spring 才能执行，由于咱们没法绕过 UserRepository 这个实例。就像前面提到的，咱们必须换一种方法，不使用 @Autowired 来注入 UserRepository。

知识点：不要注入

写一个构造器

咱们看一下不使用 @Autowired 的写法：

@Service
public class RegisterUseCase {

  private final UserRepository userRepository;

  public RegisterUseCase(UserRepository userRepository) {
    this.userRepository = userRepository;
  }

  public User registerUser(User user) {
    return userRepository.save(user);
  }

}

这个版本使用构造器来引入 UserRepository 实例。在单元测试中，咱们能够像这样来构建一个实例。

Spring 会自动的使用构造器来实例化一个 RegisterUseCase 对象。须要注意的是，在 Spring 5 以前，咱们须要@Autowired 注解来让构造器生效。

一样须要注意的是 UserRepository 字段如今是 final，这样在整个应用的生命周期里，它都将是个常量，这能够避免编码错误，由于咱们若是忘记初始化字段，编译的时候就会报错。

减小繁复的代码

使用 Lombok 的 @RequiredArgsConstructor 注解，可让构造器的写法更简洁：

@Service
@RequiredArgsConstructor
public class RegisterUseCase {

  private final UserRepository userRepository;

  public User registerUser(User user) {
    user.setRegistrationDate(LocalDateTime.now());
    return userRepository.save(user);
  }

}

如今咱们的测试类就很简洁，没有冗余繁复的代码：

class RegisterUseCaseTest {

  private UserRepository userRepository = ...;

  private RegisterUseCase registerUseCase;

  @BeforeEach
  void initUseCase() {
    registerUseCase = new RegisterUseCase(userRepository);
  }

  @Test
  void savedUserHasRegistrationDate() {
    User user = new User("zaphod", "zaphod@mail.com");
    User savedUser = registerUseCase.registerUser(user);
    assertThat(savedUser.getRegistrationDate()).isNotNull();
  }

}

不过咱们还有一点遗漏，就是如何去模拟 UserRepository 实例，由于咱们不想去真正的去执行，由于它可能须要去链接数据库。

使用 Mockito

现行的标准模拟库是 Mockito，它提供了至少两种方式来模拟 UserRepository 。

直接调用

第一种方法就是直接使用 Mockito：

private UserRepository userRepository = Mockito.mock(UserRepository.class);

这个建立一个对象，看起来和 UserRepository 同样。默认的状况下，这个类什么也不会作，若是调用有返回值的方法，也只会返回 null。

咱们的测试如今会是失败，在 assertThat(savedUser.getRegistrationDate()).isNotNull() 这儿报 NullPointerException 空指针异常，由于 userRepository.save(user) 只会返回 null。

因此，咱们须要告诉 Mockito，当 userRepository.save() 被调用的时候须要有返回值，因此咱们使用静态的 when 方法：

@Test
void savedUserHasRegistrationDate() {
  User user = new User("zaphod", "zaphod@mail.com");
  when(userRepository.save(any(User.class))).then(returnsFirstArg());
  User savedUser = registerUseCase.registerUser(user);
  assertThat(savedUser.getRegistrationDate()).isNotNull();
}

这样 userRepository.save() 会返回一个对象，其实这个对象和传入参数的对象一摸同样。

Mockito 具备一整套的测试方案，能够用来模拟、匹配参数以及识别方法的调用，更多资料能够参考这里。

使用 @Mock

此外还能够用 @Mock 注解来模拟对象，它须要和 MockitoExtension 组合使用。

@ExtendWith(MockitoExtension.class)
class RegisterUseCaseTest {

  @Mock
  private UserRepository userRepository;

  private RegisterUseCase registerUseCase;

  @BeforeEach
  void initUseCase() {
    registerUseCase = new RegisterUseCase(userRepository);
  }

  @Test
  void savedUserHasRegistrationDate() {
    // ...
  }

}

@Mock 注解会指定字段将被注入到 mock 对象，@MockitoExtension 会告诉 Mockito 去扫描 @Mock 注解，由于 JUnit 不会自动去执行。

这其实和直接手工执行 Mockito.mock() 的结果同样，只是使用习惯的区别。不过使用 MockitoExtension 咱们的测试就能够绑定到测试框架里。

须要说明的是咱们能够在 registerUseCase 字段上使用 @InjectMocks 注解来替代手工构造一个 RegisterUseCase 对象，Mockito 会帮咱们自动构造对象，如：

@ExtendWith(MockitoExtension.class)
class RegisterUseCaseTest {

  @Mock
  private UserRepository userRepository;

  @InjectMocks
  private RegisterUseCase registerUseCase;

  @Test
  void savedUserHasRegistrationDate() {
    // ...
  }

}

让断言更直白

另外一个 Spring Boot 自带的测试支持库是 AssertJ，上面的例子里，在实现断言的时候已经用到了：

assertThat(savedUser.getRegistrationDate()).isNotNull();

不过咱们想让写法变得更直白好理解，好比：

assertThat(savedUser).hasRegistrationDate();

一般，咱们能够作小改动就可让代码变得更容易理解，因此咱们新建一个自定义的断言对象：

public class UserAssert extends AbstractAssert<UserAssert, User> {

  public UserAssert(User user) {
    super(user, UserAssert.class);
  }

  public static UserAssert assertThat(User actual) {
    return new UserAssert(actual);
  }

  public UserAssert hasRegistrationDate() {
    isNotNull();
    if (actual.getRegistrationDate() == null) {
      failWithMessage("Expected user to have a registration date, but it was null");
    }
    return this;
  }
}

这样，咱们调用 UserAssert 类的 assertThat 方法，而不是直接从 Assertj 库里调用。

建立自定义的断言看起来须要不少的工做量，但其实也就是几分钟的事。我相信这几分钟的工做，绝对是值得的，即便是让代码看起来更直白容易理解。测试代码咱们只会写一次，而后其余人（包括我在之后）都只是去读这段代码，而后是反反复复的去修改这段代码，直到产品消亡。

若是还有疑问，能够参考 Assertions Generator。

结论

咱们可能有种种的理由在 Spring 里进行测试，可是对于一个普通的单元测试，能够这么作，可是没有必要。随着之后应用愈来愈庞大，启动时间愈来愈长，可能还会带来问题。因此，咱们在写单元测试的时候，应该以一种更简单的方式去构建 Sprnig bean。

Spring Boot Test Starter 附带了 Mockito 和 AssertJ 做为测试依赖库，因此尽量的使用这些测试库来作更好的单元测试吧。

全部的代码能够在这里找到。

若是发现译文存在错误或其余须要改进的地方，欢迎斧正。