如何使AutoFixture随机选择多个TypeRelays?

时间:2016-10-19 20:42:39

标签: c# .net unit-testing mocking autofixture

我有一个抽象基类,比如Base,以及一些派生类,比方说D1,D2,D3。

当Base的请求进入时,如何设置AutoFixture随机选择D1,D2,D3中的一个?

为Fixture.Customizations添加D1,D2,D3中的每一个的TypeRelay似乎总是选择添加的第一个。

1 个答案:

答案 0 :(得分:1)

能够将以下代码放在一起。如果由于某种原因,其中一个请求失败,它将回退到指定的默认值,因此应该是一个失败证明。它没有完全使用TypeRelay,但效果基本相同。

public class RandomCustomization<T> : ICustomization
{
    private readonly Type _defaultType;
    private readonly Type[] _types;
    public RandomCustomization(Type defaultType, params Type[] types)
    {
        _defaultType = defaultType;
        _types = types;
    }
    public void Customize(IFixture fixture)
    {
        fixture.Customize<T>(v => v.FromFactory(new RandomFactory(_defaultType, _types)));
    }
}

public class RandomFactory : ISpecimenBuilder
{
    private readonly Type _defaultType;
    private readonly Type[] _types;
    private readonly Random _ran = new Random();

    public RandomFactory(Type defaultType, params Type[] types)
    {
        _defaultType = defaultType;
        _types = defaultType.Concat(types).ToArray();
    }
    public object Create(object request, ISpecimenContext context)
    {
        var which = _types[_ran.Next() % _types.Length];
        var toret = context.Resolve(which);
        if (toret == null || toret is OmitSpecimen)
        {
            toret = context.Resolve(_defaultType);
        }
        return toret;
    }
}