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

unit-testing java spring spring-data spring-data-jpa

2022-09-03 02:38:39

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

 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);
    }
}

这实际上工作正常，但我想为此方法编写单元测试，我无法弄清楚如何检查传递给我的文章Repository.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);


}

有什么想法吗？

答案 1

我遇到了几乎和你一样的问题，我把包含 Specification 的类改成了一个对象，而不仅仅是一个使用静态方法的类。通过这种方式，我可以轻松模拟它，使用依赖注入来传递它，并测试调用了哪些方法（不使用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
}

当然，在服务中，您仍然可以使用 Specification 类的 where 和和静态方法。

我希望这可以帮助你。

答案 2

如果你正在编写单元测试，那么你应该使用一个模拟框架来模拟对Class方法的调用，比如或。findAll()articleRepositoryMockitoPowerMock

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

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

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

如果未为您提供的参数调用 mock，则测试将失败。