用Mockito模拟JPA标准生成器

Question

对于一个JPA查询，我有一个特别讨厌的JMock checking（）块，我想迁移到Mockito：

Mockery jMock = new Mockery();
final EntityManager fakeEntityManager = jMock.mock(EntityManager.class);
final CriteriaBuilder fakeCriteriaBuilder = jMock.mock(CriteriaBuilder.class);
final CriteriaQuery<String> fakeCriteriaQuery = jMock.mock(CriteriaQuery.class);
jMock.checking(new Expectations() {{
    oneOf(fakeEntityManager).getCriteriaBuilder(); will(returnValue(fakeCriteriaBuilder));
    oneOf(fakeCriteriaBuilder).createQuery(String.class); will(returnValue(fakeCriteriaQuery));
    oneOf(fakeCriteriaQuery).from(Archiveusergrouplicences.class);
    oneOf(fakeCriteriaQuery).select(with(any(Selection.class)));
    oneOf(fakeCriteriaBuilder).isNotNull(with(any(Expression.class)));
    oneOf(fakeCriteriaQuery).where(with(any(Expression.class)));
    oneOf(fakeEntityManager).createQuery(fakeCriteriaQuery);
    // Return an empty resultset
}});

正在测试的代码如下所示：

CriteriaBuilder builder = entityManager.getCriteriaBuilder();
CriteriaQuery<String> criteria = builder.createQuery(String.class);
Root<Archiveusergrouplicences> institution = criteria.from(Archiveusergrouplicences.class);
criteria.select(institution.get(Archiveusergrouplicences_.usergroupid));    
criteria.where(builder.isNotNull(institution.get(Archiveusergrouplicences_.usergroupid)));
List<String> result = entityManager.createQuery(criteria).getResultList();

我在嘲笑建设者时发现了这个问题，这在一定程度上解决了mock中的CriteriaBuilder部分；但我的主要问题是使用模拟对象作为另一个mock的.thenReturn（）值——Mockito似乎不允许这样做。例如，对于线路：

CriteriaQuery<String> criteria = builder.createQuery(String.class);

我想返回mock CriteriaQuery对象，如下所示：

CriteriaQuery<String> fakeCriteriaQuery = mock(CriteriaQuery.class, RETURNS_DEEP_STUBS);
when(entityManager.createQuery(anyString())).thenReturn(fakeCriteriaQuery);

这引发了一个语法错误：

类型OngoingStubbing中的方法thenReturn（Query）不是适用于参数（CriteriaQuery）

我该如何测试这个代码，或者改进它以使其更易于测试？

Answer 1

事实证明，只要设置了正确的参数，就可以从其他mock返回mock！我试图断言：

when(entityManager.createQuery(anyString())).thenReturn(fakeCriteriaQuery);

当我真正想要的是通过一门课：

when(fakeCriteriaBuilder.createQuery(String.class)).thenReturn(fakeCriteriaQuery);

这个错误是Mockito神秘地告诉我，我把预期搞砸了。

然而，我可能会考虑重写这个测试，而不是翻译最初写的内容。正如一些人指出的那样；通常最好避免以这种方式嘲弄库，并且检查的条件相当模糊。

Answer 2

像你一样jMock CriteriaQuery

final CriteriaQuery<String> fakeCriteriaQuery = jMock.mock(CriteriaQuery.class);

你需要mockito CriteriaQuery太

final CriteriaQuery<String> fakeCriteriaQuery = mock(CriteriaQuery.class);