如何使用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)));
}

如何最好地模拟EntityManager将Id从0L更改为1L的行为？可能性最小的可读性障碍.

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

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