Mocking方法使用Mockito进行链接(尤其是JPA TypedQuery)

Question

我有很多代码，比如下面的示例，我需要用Mockito进行测试。

return entityManager
    .createNamedQuery("queryName", Type.class)
    .setParameter("foo", "fish")
    .setParameter("bar", 42)
    .getSingleResult();

我已经注入了一个模拟EntityManager，并让它返回一个模拟TypedQuery等。问题是，我需要为TypedQuery中的每个方法指定如下规则，以便在没有NullPointerExceptions的情况下成功运行测试。

when(mockedTypedQuery.setParameter(any(String.class), any(Object.class)))
        .thenReturn(mockedTypedQuery);

有没有更有效的方法？

Answer 1

在这种情况下，我所做的一切都受到了这篇文章的启发：http://geeks.autotrader.co.uk/2014/03/mocking-fluent-interfaces-with-mockito.html

我有以下服务：

public class TestService {
    public TypedQuery typedQuery;
    public List test(){
        final TypedQuery typedQuery1 = typedQuery.setParameter("a", "a");
        return typedQuery1.setParameter("b", "b").getResultList();
    }
}

我的测试案例是：

@RunWith(MockitoJUnitRunner.class)
public class ChainMethodsMockTest {
    TypedQuery mockedTypedQuery = fluentMock(TypedQuery.class);
    @InjectMocks
    TestService testService;
    @Test
    public void testMockito(){
        final ArrayList value = new ArrayList();
        value.add("a");
        value.add("b");
        Mockito.when(mockedTypedQuery.getResultList()).thenReturn(value);
        final List test = testService.test();
        Assert.assertEquals(2, test.size());
    }
    public static <T> T fluentMock(final Class<T> type) {
        return Mockito.mock(type, Mockito.withSettings().defaultAnswer(
                new ReturnsEmptyValues() {
                    @Override
                    public Object answer(InvocationOnMock invocation) {
                        Object defaultReturnValue = super.answer(invocation);
                        if (type.equals(invocation.getMethod().getReturnType())) {
                            return invocation.getMock();
                        } else {
                            return defaultReturnValue;
                        }
                    }
                }));
    }
}

fluentMock方法，您可以在项目之间共享和重用该方法。