通过 id 删除实体的 JUnit 测试失败

Question

我正在尝试测试通过 id 删除实体的方法。我已经为具有给定 id 的实体不存在的情况编写了测试，但如果实体存在，我显然做错了什么。这是服务中的删除方法。我正在使用JpaRepository 的deleteById() 方法。

  public void deleteById(Integer id) {
        Optional<Agency> optionalAgency = repo.findAll().stream()
                .filter(agency -> agency.getId().equals(id))
                .findFirst();

        if (optionalAgency.isPresent()) {
            repo.deleteById(optionalAgency.get().getId());
        } else {
            throw new AgencyNotFoundException("Agency not found");
        }
    }

这是我的测试：

    @Mock
    private AgencyRepository agencyRepository;

    @Mock
    private AgencyMapper mapper;

    @InjectMocks
    private AgencyService agencyService;

    @Test
    void delete() {
        Agency agency = new Agency();

        when(agencyRepository.findById(1)).thenReturn(Optional.of(agency));

        agencyService.deleteById(1);

        verify(agencyRepository).deleteById(1);

    }

我应该做出什么断言来验证删除是否成功？我尝试过的方式不起作用，测试的结果是抛出异常。我猜测抛出异常是因为agencyRepository.findById((int) 1L); 基本上不存在了，但我想也许有更好的方法来验证删除，而无需搜索已删除的对象。

Answer 1

不是答案，但你的方法可以写成这样：

 public void deleteById(Integer id) {
     Agency agency = repo.findById(id).orElseThrow(() -> throw new AgencyNotFoundException("Agency not found"));
     repo.deleteById(agency.getId());
 }

因为：

• Optional<Agency> optionalAgency = repo.findAll().stream().filter(agency -> agency.getId().equals(id)).findFirst(); 效率不高。如果你有成千上万的机构怎么办？您是否会获取所有内容并过滤所有内容，以便通过 id 查找。您的存储库中已经有了该方法。
• if (optionalAgency.isPresent()) {} Optional.isPresent 不是好的做法。阅读here 和this answer

Answer 2

repo 是一个模拟，因此在其上调用deleteById 实际上不会删除任何内容。相反，您可以验证调用AgencyService 的deleteById 实际上调用了AgencyRepository 的deleteById。

编辑：
存储库的代码使用findAll，而不是findById，因此您需要确保模拟它：

@Test
void delete() {
    Agency agency = new Agency();
    agenct.setId((int) 1L);

    when(agencyRepository.findAll()).thenReturn(Collections.singletonList(agency));

    agencyService.deleteById((int) 1L);

    verify(agencyRepository).delteById((int) 1L);
}