如何使用“Spring Data JPA”规范对方法进行单元测试

Question

我在玩 org.springframework.data.jpa.domain.Specifications，这只是一个基本的搜索：

 public Optional<List<Article>> rechercheArticle(String code, String libelle) {
    List<Article> result = null;

    if(StringUtils.isNotEmpty(code) && StringUtils.isNotEmpty(libelle)){
        result = articleRepository.findAll(Specifications.where(ArticleSpecifications.egaliteCode(code)).and(ArticleSpecifications.egaliteLibelle(libelle)));
    }else{
        if(StringUtils.isNotEmpty(code)){
            result= articleRepository.findAll(Specifications.where(ArticleSpecifications.egaliteCode(code)));
        }else{
            result = articleRepository.findAll(Specifications.where(ArticleSpecifications.egaliteLibelle(libelle)));
        }
    }

    if(result.isEmpty()){
        return Optional.empty();
    }else{
        return Optional.of(result);
    }
}

这实际上工作正常，但我想为此方法编写单元测试，但我不知道如何检查传递给我的 articleRepository.findAll() 的规范

目前我的单元测试看起来像：

@Test
public void rechercheArticle_okTousCriteres() throws FacturationServiceException {
    String code = "code";
    String libelle = "libelle";
    List<Article> articles = new ArrayList<>();
    Article a1 = new Article();
    articles.add(a1);
    Mockito.when(articleRepository.findAll(Mockito.any(Specifications.class))).thenReturn(articles);


    Optional<List<Article>> result = articleManager.rechercheArticle(code, libelle);

    Assert.assertTrue(result.isPresent());
    //ArgumentCaptor<Specifications> argument = ArgumentCaptor.forClass(Specifications.class);
    Mockito.verify(articleRepository).findAll(Specifications.where(ArticleSpecifications.egaliteCode(code)).and(ArticleSpecifications.egaliteLibelle(libelle)));
    //argument.getValue().toPredicate(root, query, builder);


}

有什么想法吗？

Answer 1

我遇到了几乎与您相同的问题，我将包含 Specifications 的类更改为一个对象，而不仅仅是一个具有静态方法的类。这样我可以轻松地模拟它，使用依赖注入来传递它，并测试调用了哪些方法（无需使用 PowerMockito 模拟静态方法）。

如果你想像我一样做，我建议你用集成测试来测试规范的正确性，其余的，只要调用了正确的方法。

例如：

public class CdrSpecs {

public Specification<Cdr> calledBetween(LocalDateTime start, LocalDateTime end) {
    return (root, query, cb) -> cb.between(root.get(Cdr_.callDate), start, end);
}
}

然后你对这个方法做一个集成测试，会测试这个方法是否正确：

@RunWith(SpringRunner.class)
@DataJpaTest
@Sql("/cdr-test-data.sql")
public class CdrIntegrationTest {

@Autowired
private CdrRepository cdrRepository;

private CdrSpecs specs = new CdrSpecs();

@Test
public void findByPeriod() throws Exception {
    LocalDateTime today = LocalDateTime.now();
    LocalDateTime firstDayOfMonth = today.with(TemporalAdjusters.firstDayOfMonth());
    LocalDateTime lastDayOfMonth = today.with(TemporalAdjusters.lastDayOfMonth());
    List<Cdr> cdrList = cdrRepository.findAll(specs.calledBetween(firstDayOfMonth, lastDayOfMonth));
    assertThat(cdrList).isNotEmpty().hasSize(2);
}

现在当你想对其他组件进行单元测试时，你可以这样测试，例如：

@RunWith(JUnit4.class)
public class CdrSearchServiceTest {

@Mock
private CdrSpecs specs;
@Mock
private CdrRepository repo;

private CdrSearchService searchService;

@Before
public void setUp() throws Exception {
    initMocks(this);
    searchService = new CdrSearchService(repo, specs);
}

@Test
public void testSearch() throws Exception {

    // some code here that interact with searchService

    verify(specs).calledBetween(any(LocalDateTime.class), any(LocalDateTime.class));
   // and you can verify any other method of specs that should have been called
}

当然，在 Service 内部，您仍然可以使用 Specifications 类的 where 和 and 静态方法。

希望对你有帮助。

Answer 2

如果您正在编写单元测试，那么您可能应该使用 Mockito 或 PowerMock 等模拟框架模拟对 articleRepository 类的 findAll() 方法的调用。

有一个方法verify()，您可以使用该方法检查是否为特定参数调用了模拟。

例如，如果您正在模拟 articleRepository 类的 findAll() 方法并想知道是否使用特定参数调用此方法，那么您可以执行以下操作：

Mokito.verify(mymock, Mockito.times(1)).findAll(/* Provide Arguments */);

如果没有为您提供的参数调用 mock，这将导致测试失败。

Answer 3

您的问题是您在一种方法中做了太多事情。您应该有三种不同的方法可以在 articleRepository 上工作。

然后您可以按照其他人的建议使用模拟：

• 设置您的模拟，以便您知道应该对 articleRepository 进行哪个调用
• 验证是否正在发生预期的调用

请注意：这三个方法应该是内部的；要点是：您不能通过外部的 ONE 调用来测试此方法；因为它做的不止一件事，具体取决于您提供的输入。因此，您需要为代码中的每个潜在路径创建至少一个测试方法。当您将代码分成不同的方法时，这变得更容易（从概念的角度来看）。