java - Mockito 模拟对象在方法内部的方法调用中被分配了一个 null 值

Question

我有一个类需要用mockito 进行测试。下面是类和 Mockito 测试。

dbBuilder.parse(file) 始终返回 null，因为它调用多个类，而这些类又调用 jar 文件中的多个方法。尽管我 mock 了所有这些，但它总是返回 null。我无法跟踪空值来自哪里。我尝试抑制这些方法，但仍然没有用。

由于此方法调用返回 null，因此文档值为 null。因此，文档调用 getElementsByTagName 方法，并且运行 mockito 测试失败并出现 NullPointerException。这个方法中的代码后面有几行代码我需要测试。

如何解决这个问题？

class DocumentClass{
    public void docMethod(){
        DocumentBuilder dbBuilder = new DocumentBuilder();
        Document doc = new Document();
        FileStream file = new FileStream(new File(some path));
        doc = dbBuilder.parse(file);
        NodeList nodes = doc.getElementsByTagName("documents");
    }
}

@RunWith(PowerMockRunner.class) 
@PrepareForTest({Document.class,DocumentBuilder.class,FileStream.class})
public class TestDocument{
    @Test
    public documentTest(){
        DocumentBuilder dbBuilder = PowerMockito.mock(DocumentBuilder.class);
        Document doc = PowerMockito.mock(Document.class);
        FileStream file = PowerMockito.mock(FileStream.class);
        PowerMockito.whenNew(FileStream.class).withAnyArguments().thenReturn(file);
        PowerMockito.doReturn(doc).when(dbBuilder).parse(file);
        DocumentClass docClass = PowerMockito.mock(DocumentClass.class);
        docClass.docMethod();
   }
}

Answer 1

编辑:我知道您已经提到过您之前尝试过whenNew，但我会更详细地提供答案，以便检查此操作所需的所有详细信息。

在我看来，您需要模拟 DocumentBuilder 类的构造函数调用。执行此操作时，您需要在PrepareForTest注释中包含调用构造函数的类，因此请确保其中存在DocumentClass。请参阅documentation (主要是快速摘要)模拟构造函数调用时要准备测试的内容。

然后，您还需要为 DocumentBuilder 上的方法调用提供期望调用。

另外，我认为您不需要调用 Document 类的构造函数，在您只需将引用替换为 dbBuilder 中的引用之前，您无需对其执行任何操作。

所以我认为你的方法可以是这样的:

class DocumentClass{
    public void docMethod(){
        final FileStream file = new FileStream(new File(some path));

        final DocumentBuilder dbBuilder = new DocumentBuilder();
        final Document doc = dbBuilder.parse(file);

        final NodeList nodes = doc.getElementsByTagName("documents");
    }
}

所以我认为你的测试应该看起来像这样:(我还没有尝试过这段代码，所以我对任何拼写错误表示歉意)

@RunWith(PowerMockRunner.class)
@PrepareForTest({DocumentClass.class, Document.class, DocumentBuilder.class, FileStream.class})
public class TestDocument {
    @Test
    public documentTest() {
        final DocumentBuilder dbBuilder = PowerMockito.mock(DocumentBuilder.class);
        final Document doc = PowerMockito.mock(Document.class);
        final FileStream file = PowerMockito.mock(FileStream.class);
        final NodeList nodes = PowerMockito.mock(NodeList.class);

        PowerMockito.whenNew(FileStream.class).withAnyArguments().thenReturn(file);
        PowerMockito.whenNew(DocumentBuilder).thenReturn(dbBuilder);
        PowerMockito.doReturn(doc).when(dbBuilder).parse(file);
        PowerMockito.doReturn(node).when(doc).getElementsByTagName("documents");

        PowerMock.replayAll();

        DocumentClass docClass = new DocumentClass();
        docClass.docMethod();

        PowerMock.verifyAll();
    }
}