Question

假设我有一个模拟对象，我不想存根任何方法，但我想要存根一个它返回的对象的方法。例如，

when(mockObject.method1()).thenReturn(returnValue)

它是如何正常完成的，但我正在寻找，

when(mockObject.method1().method2()).thenReturn(returnValue)

这可能吗？如果我这样做，我会得到一个NullPointerException。目前我有第一个返回模拟对象的方法，然后使用返回的模拟对象，存根第二个方法。但是，这些临时模拟对象对我来说是无用的，并且在将许多方法链接在一起之后，会导致很多无用的模拟对象。

编辑：实际上，链接可能有效，但我的对象正在引起NPE。此代码（第一行）导致NPE：

when(graphDb.index().getNodeAutoIndexer()).thenReturn(nodeAutoIndexer);
when(graphDb.index().getRelationshipAutoIndexer()).thenReturn(relAutoIndexer);

但是这段代码有效：

IndexManager indexManager = mock(IndexManager.class);
when(graphDb.index()).thenReturn(indexManager);
when(indexManager.getNodeAutoIndexer()).thenReturn(nodeAutoIndexer);
when(graphDb.index().getRelationshipAutoIndexer()).thenReturn(relAutoIndexer);

因此，链接不适用于getNodeAutoIndexer（），它返回一个AutoIndexer对象，同时它对getRelationshipAutoIndexer（）有效，返回一个RelationshipAutoIndexer。两个返回值都被模拟如下：

nodeAutoIndexer = (AutoIndexer<Node>) mock(AutoIndexer.class);
relAutoIndexer = mock(RelationshipAutoIndexer.class);

那么可能导致问题的原因是什么？

Answer 1

完全没有问题。

让我们检查这4行代码：

IndexManager indexManager = mock(IndexManager.class);
when(graphDb.index()).thenReturn(indexManager);
when(indexManager.getNodeAutoIndexer()).thenReturn(nodeAutoIndexer);
when(graphDb.index().getRelationshipAutoIndexer()).thenReturn(relAutoIndexer);

第一行创建一个模拟indexManager。

第二个告诉模拟graphDb在调用索引方法时返回indexManager（在第一行创建的模拟）。

第三个是在调用其getNodeAutoIndexer方法时，使用模拟indexManager（在第一行创建）返回nodeAutoIndexer。

最后一行调用graphDb.index（），它返回模拟indexManager（您告诉它在第二行执行此操作），并询问此indexManager（您在第一行创建的模拟）返回relAutoIndexer时调用它的getRelationshipAutoIndexer方法。

最后一行的作用只是因为你告诉模拟graphDb在调用索引方法时要返回什么。如果你以前没有这样做过，那么mock graphDb.index（）方法会返回null，你会得到一个NPE。

Mockito - 存储由模拟对象方法返回的对象的方法

1 个答案: