如何在单元测试期间覆盖被测试类中的某些属性

时间:2015-06-09 23:15:47

标签: c# unit-testing design-patterns

我有一个课我想用单元测试来测试。它有一些逻辑可以在本地xml文件中查找某些值,如果找不到该值,它将读取一些外部源(SqlServer DB)。但是在单元测试期间,我根本不希望这个组件与Sql Server交互。在单元测试期间,我想将外部源读取器的SqlServer实现替换为某些Noop读取器。实现这一目标的好方法是什么?重要说明:我无法定义接受读者类型实例的构造函数,因为它是面向客户端的API,我想限制它们可以对我的类做什么。 我目前在单元测试中使用的方法很少:

  • 使用反射将private / protected属性的值设置为我的模拟读者实现
  • 定义将创建具体类的工厂。在Unity容器中注册工厂&被测试的类将从DI容器中获取工厂对象并根据我的需要实例化读者
  • 子类class-under-test并在那里设置属性。

但它们似乎都不够干净。有没有更好的方法来实现这一目标? 以下是演示测试类的示例的示例代码:

namespace UnitTestProject1
{
    using System.Collections.Generic;
    using System.Data.SqlClient;

    public class SomeDataReader
    {
        private Dictionary<string, string> store = new Dictionary<string, string>();

        // I need to override what this property does
        private readonly IExternalStoreReader ExternalStore = new SqlExternalStoreReader(null, false, new List<string>() {"blah"});


        public string Read(string arg1, int arg2, bool arg3)
        {
            if (!store.ContainsKey(arg1))
            {
                return ExternalStore.ReadSource().ToString();
            }

            return null;
        }
    }

    internal interface IExternalStoreReader
    {
        object ReadSource();
    }

    // This 
    internal class SqlExternalStoreReader : IExternalStoreReader
    {
        public SqlExternalStoreReader(object arg1, bool arg2, List<string> arg3)
        {

        }

        public object ReadSource()
        {
            using (var db = new SqlConnection("."))
            {
                return new object();    
            }
        }
    }

    internal class NoOpExternalStoreReader : IExternalStoreReader
    {
        public object ReadSource()
        {
            return null;
        }
    }
}

[TestClass]
    public class UnitTest1
    {
        [TestMethod]
        public void TestMethod1()
        {
            var objectToTest = new SomeDataReader();
            objectToTest.Read("", -5, false); // This will try to read DB, I don't want that.
        }
    }

3 个答案:

答案 0 :(得分:3)

在这种情况下,您可以使用InternalsVisibleTo属性,该属性会将所有内部成员公开为friend assemblies

您可以先创建一个单独的internal构造函数重载,它可以接受IExternalStoreReader的不同实例:

public class SomeDataReader
{
    // publicly visible
    public SomeDataReader()
        : this(new SqlExternalStoreReader(...))
    { }

    // internally visible
    internal SomeDataReader(IExternalStoreReader storeReader)
    {
        ExternalStore = storeReader;
    }

    ...
}

然后通过将InternalsVisibleTo属性添加到AssemblyInfo.cs来允许单元测试程序集访问内部成员:

[assembly: InternalsVisibleTo("YourUnitTestingAssembly")]

如果您真的担心有人试图访问您的内部成员,您还可以使用strong naming InternalsVisibleTo属性来确保没有人尝试模拟您的单元测试程序集。

答案 1 :(得分:2)

简单的答案是:更改代码。因为你有一个私有的只读字段可以创建自己的值,所以如果没有真正的hacky,你就无法改变这种行为。所以相反,不要这样做。将代码更改为:

public class SomeDataReader
{
    private Dictionary<string, string> store = new Dictionary<string, string>();

    private readonly IExternalStoreReader externalStore;

    public SomeDataReader(IExternalStoreReader externalStore)
    {
        this.externalStore = externalStore;
    }

换句话说,将IExternalStoreReader实例注入类中。这样,您可以为单元测试创​​建noop版本,为生产代码创建真实版本。

答案 2 :(得分:1)

如果您自己编译完整代码,可以在单元测试项目中创建一个带有存根实现的虚假SqlExternalStoreReader。 此存根实现允许您访问所有字段并将调用转发到模拟框架/单元测试