java - 模拟 openInputStream 读取方法

Question

我正在从 openInputStream 读取数据 block ，并想模拟它的读取方法。

 final byte[] testFileData = { 0x74, 0x65, 0x73, 0x74, 0x20, 0x64, 0x61, 0x74, 0x61, 0x0d, 0x0a};

 CloudBlob cloudBlob = this.getCloudBlobContainer().getBlobReferenceFromServer(blobName);

    ByteArrayOutputStream blobStream = new ByteArrayOutputStream();

    try (final InputStream inputStream =  cloudBlob.openInputStream()) {

        //Read 4MB chunks of data 
        byte[] bufferToRead = new byte[4 * 1024 *1024];
        int bytesRead = inputStream.read(bufferToRead );

        while (bytesRead > 0) {
            //Add only the total number of bytes read to Bytearrayoutputstream
            blobStream.write(bufferToRead, 0, bytesRead);
            bytesRead = inputStream.read(bufferToRead);                
        }
    } `

我确实模拟了InputStream，但发现模拟它的读取方法很困难，因为它接受缓冲区作为引用，并在读取后将字节数组复制到其中。

 @Mock
private BlobInputStream inputStream;


       // Mock
        when(cloudBlobContainer.getBlobReferenceFromServer(anyString())).thenReturn(cloudBlob);
        when(cloudBlob.openInputStream()).thenReturn(inputStream);  

Answer 1

InputStream 很难模拟，尤其是因为它的三个 read 重写以及稍后添加的 readNBytes 和 readAllBytes 方法Java 版本。如果使用 Mockito 完成，您需要让所有这些方法实现与相同的数据进行交互，否则您将得到一个脆弱的测试，一旦您的实现调用不同的 InputStream 方法，该测试就可能会中断。如果你必须使用测试替身，你最好写一个“假的”，但当 Java 有 ByteArrayInputStream 时，没有理由这样做。内置:您可以构造您的 byte[] 缓冲区(或编写一个辅助方法来根据您的测试需要构造一个缓冲区)并将其替换为测试中的 InputStream。 这对于任何提出这个问题只是询问一般情况下模拟 InputStream 的人来说应该足够了。

final byte[] testFileData = {
    0x74, 0x65, 0x73, 0x74, 0x20, 0x64, 0x61, 0x74, 0x61, 0x0d, 0x0a};
ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(testFileData);

不幸的是，这并没有回答有关如何专门为 Azure 的 BlobInputStream 提供测试夹具的问题。 ，这特别棘手，因为 BlobInputStream has a four-arg constructor with a lot of Azure internals 。好消息是，从 1.9.5 开始，Mockito 提供了 delegatesTo method in AdditionalAnswers记录(我自己强调)为:

An answer that directly forwards the calls to the delegate. The delegate may or may not be of the same type as the mock. If the type is different, a matching method needs to be found on delegate type otherwise an exception is thrown.

这意味着您可以通过创建真正的 ByteArrayInputStream 并将可重写方法委托(delegate)给它来模拟 BlobInputStream。不幸的是，delegatesTo位于AdditionalAnswers上，而不是Answers枚举上(并且它需要一个实例参数，而您无法在注释中提供)，因此您需要手动构建您的模拟:

BlobInputStream mockInputStream = Mockito.mock(
    BlobInputStream.class,
    AdditionalAnswers.delegatesTo(byteArrayInputStream));

when(cloudBlobContainer.getBlobReferenceFromServer(anyString()))
    .thenReturn(cloudBlob);
when(cloudBlob.openInputStream()).thenReturn(mockInputStream);

但是，如果可能的话，这将是一个很好的机会，可以将您的代码分为处理 Azure 输入流的部分与处理任何 InputStream 的应用程序代码 。例如，如果您的代码采用 BlobInputStream 并运行解析或纠错代码，您可以提取一个方法 handleInputStream(InputStream) 并传入您自己的 ByteArrayInputStream 来对其进行大量测试。这最大限度地减少了模拟 BlobInputStream 的需要，或者如果您选择仅使用真实后端将 BlobInputStream 处理作为集成测试来测试，则可能完全消除它。另请参阅Mocking Files in Java - Mock Contents - Mockito ，其中 Brice 讨论了单元测试和集成测试之间的类似划分，而不是模拟 java.io.File 实例。