5

对于要迁移到 Mockito的 JPA 查询，我有一个特别讨厌的 JMock checks() 块：

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

我在 mocking builders 上找到了这个问题，这在某种程度上解决了模拟的 CriteriaBuilder 部分；但我的主要问题是使用模拟对象作为另一个模拟的 .thenReturn() 值 - Mockito 似乎不允许这样做。例如，对于该行：

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

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

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

这会引发语法错误：

OngoingStubbing 类型中的 thenReturn(Query) 方法不适用于参数 (CriteriaQuery)

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

4

2 回答 2

3

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

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

当我真正想要的是通过一个类时：

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

错误是 Mockito 用神秘的方式告诉我我搞砸了我的期望。

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

于 2012-02-08T11:12:08.290 回答
0

像你一样 jMock CriteriaQuery

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

你也需要模拟 CriteriaQuery

final CriteriaQuery<String> fakeCriteriaQuery = mock(CriteriaQuery.class);
于 2012-02-08T10:14:47.403 回答