如何使用 Mockito 和 jUnit 模拟持久化和实体

2022-01-14 00:00:00 mockito java junit hamcrest

我正在尝试寻找一种方法来使用 Mockito 测试我的实体;

I'm trying to find a way to test my entity using Mockito;

这是简单的测试方法:

@Mock
private EntityManager em;

@Test
public void persistArticleWithValidArticleSetsArticleId() {
    Article article = new Article();
    em.persist(article);
    assertThat(article.getId(), is(not(0L)));
}

如何最好地模拟 EntityManager 将 Id 从 0L 更改为即 1L 的行为?可能在可读性方面的障碍最少.

How do I best mock the behaviour that the EntityManager changes the Id from 0L to i.e. 1L? Possibly with the least obstructions in readability.

一些额外的信息;在测试范围之外,EntityManager 由应用程序容器生成

Some extra information; Outside test-scope the EntityManager is produced by an application-container

推荐答案

public class AssignIdToArticleAnswer implements Answer<Void> {

    private final Long id;

    public AssignIdToArticleAnswer(Long id) {
        this.id = id;
    }

    @Override
    public Void answer(InvocationOnMock invocation) throws Throwable {
        Article article = (Article) invocation.getArguments()[0];
        article.setId(id);
        return null;
    }
}

然后

doAnswer(new AssignIdToArticleAnswer(1L)).when(em).persist(any(Article.class));

相关文章