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

Question

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

这是简单的测试方法：

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

如何最好地模拟实体管理器将 ID 从 0L 更改为 1L 的行为？可能在可读性方面障碍最少。

编辑：一些额外的信息;在测试范围之外，实体管理器由应用程序容器生成

Answer 1

你可以

为此使用Mockito Answer。

doAnswer(new Answer<Object>(){
     @Override
     public Object answer(InvocationOnMock invocation){
        Article article = (Article) invocation.getArguments()[0];
        article.setId(1L);
        return null;
     }
  }).when(em).persist(any(Article.class));

这告诉 Mockito ，当调用 persist 方法时，第一个参数应该调用其setId方法。

但是如果你这样做，我不明白测试的目的是什么。 你实际上只是在测试Mockito Answer机制是否有效，而不是Article或EntityManager的代码是否正常工作。

Answer 2

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

Answer 3

与

上面类似的答案，但带有 lambda

   doAnswer((InvocationOnMock invocation) -> {
        Article article = (Article) invocation.getArguments()[0];
        article.setId(1L);
        return null;
    }).when(em).persist(any(Article.class));