如何模拟引用私有对象实例的 .NET 代码



我有以下代码,我正在尝试为其编写单元测试:

// Grab all of the content.
var rawContent = new StringBuilder();
foreach (var fileInfo in response.Files)
{
    using (var stream = fileInfo.VirtualFile.Open())
    {
        using (var streamReader = new StreamReader(stream))
        {
            rawContent.Append(streamReader.ReadToEnd());
        }
    }
}

现在,我已经模拟了Open()返回的 Stream 实例。这很容易。流对象也是一个MemoryStream,顺便说一句。

但是我不确定如何设置ReadToEnd()..以返回我拥有的一些字符串内容。这是因为StreamReader实例是此方法专用的。

我想提供一个可以设置的属性。

例如。

public string AStreamReader { get; set; }

然后这样做..

// Grab all of the content.
var rawContent = new StringBuilder();
foreach (var fileInfo in response.Files)
{
    using (var stream = fileInfo.VirtualFile.Open())
    {
        using (var streamReader = AStreamReader ?? new StreamReader(stream))
        {
            rawContent.Append(streamReader.ReadToEnd());
        }
    }
}

然后只是通过AStreamReader属性传递模拟StreamReader

但我真的不想这样做。我觉得我泄露了一些内在的逻辑,不必要,只是为了嘲笑!真正的用户永远不会设置它,所以这感觉像是一件非常肮脏的事情。

也许我错过了一些东西,但是如果您能够模拟输入Stream,为什么需要模拟StreamReader

以下方法不适合您吗?

    private string ReadFromResponseFiles(Stream[] inputs)
    {
        var rawContent = new StringBuilder();
        foreach (var stream in inputs)
        {
            using (stream)
            {
                using (var streamReader = new StreamReader(stream))
                {
                    rawContent.Append(streamReader.ReadToEnd());
                }
            }
        }
        return rawContent.ToString();
    }
    [Test]
    public void Can_read_files()
    {
        var file1 = new MemoryStream(Encoding.UTF8.GetBytes("Foo"));
        var file2 = new MemoryStream(Encoding.UTF8.GetBytes("Bar"));
        var result = ReadFromResponseFiles(new[] { file1, file2 });
        Assert.AreEqual("FooBar", result);
        file1.Dispose();
        file2.Dispose();
    }
}

将修饰符设置为受保护,并将其子类化。这就是我们处理大多数测试的方式,在这种情况下。

最新更新