模拟CloudBlockBlob并使其返回流

时间:2019-03-08 13:12:52

标签: c# asp.net-core azure-storage moq

我正在尝试Moz Azure CloudBlockBlob并让其返回流,以便我可以测试BlobStorage存储库是否正确处理了输出。 但是以某种方式返回的流始终为空。

单元测试代码:

        ....
        var stream = new MemoryStream();
        var writer = new StreamWriter(stream);
        writer.Write("sample data");
        writer.Flush();
        stream.Position = 0;

        var blobMock = new Mock<CloudBlockBlob>(new Uri("http://tempuri.org/blob"));
        blobMock
            .Setup(m => m.ExistsAsync())
            .ReturnsAsync(true);
        blobMock
            .Setup(m => m.DownloadToStreamAsync(It.IsAny<MemoryStream>()))
            .Returns(Task.FromResult(stream));
        ....

存储库代码:

        ....
        var blob = GetContainer(container).GetBlockBlobReference(name);
        if (await blob.ExistsAsync())
        {
            var ms = new MemoryStream();
            await blob.DownloadToStreamAsync(ms);
            ms.Seek(0, SeekOrigin.Begin);
            return ms;
        }
        ....

因此,我返回的ms流始终是空流,而不是Moq stream方法中使用的Returns()对象。 我如何让该Blob返回我的样本流?

1 个答案:

答案 0 :(得分:2)

这是两个不同的流。在Callback中获取通过模拟参数传递的流,然后将测试流复制过来。

例如

//....
var stream = new MemoryStream();
var writer = new StreamWriter(stream);
writer.Write("sample data");
writer.Flush();
stream.Position = 0;

var blobMock = new Mock<CloudBlockBlob>(new Uri("http://tempuri.org/blob"));
blobMock
    .Setup(m => m.ExistsAsync())
    .ReturnsAsync(true);
blobMock
    .Setup(m => m.DownloadToStreamAsync(It.IsAny<Stream>()))
    .Callback((Stream target) => stream.CopyTo(target)) //<---Something like this
    .Returns(Task.CompletedTask);
//....

该模拟实际上并不返回流。假定要对流进行操作,这就是为什么需要回调来复制预期行为的原因。

记笔记

  

复制从当前流中的当前位置开始,并且在复制操作完成后不会重置目标流的位置。

因此,在这种情况下,如果打算从目标中读取内容,则可能需要重置它

//...

.Callback((Stream target) => {
    stream.CopyTo(target);
    target.Position = 0;
})

//...